-
Notifications
You must be signed in to change notification settings - Fork 0
/
RaspythonBot.py
executable file
·630 lines (545 loc) · 28.6 KB
/
RaspythonBot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
#!/usr/bin/python3
# -*- coding: utf-8 -*-
# RaspythonBot (c) 2017 - Raspython Group
#
# Bot de moderación creado por y para el grupo de Telegram @Raspython
# t.me/raspython
#
#
##########################################################################
##########################################################################
# Librería PTB
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackQueryHandler
from telegram import MessageEntity, ParseMode, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.ext.dispatcher import run_async
# Librería para el log
import logging
# Librerías del sistema
import os
# Librería regex para filtrar textos
import re
# Librería números enteros aleatorios
from random import randint
# Librería de tiempo, para pausas y demás
from time import sleep
import datetime
# Librería json
import json
# Librería para el wrapping de librerías (decoración)
from functools import wraps
# Importamos las constantes desde el fichero correspondiente
# Las importamos una a una por claridad y para saber cuales añadimos nuevas
from constantes import TOKEN, LOG_FILE, GROUPS_ID, HASHTAG_FILE, ADMINS, START, END, HASHTAG, ANTIFLOOD, ANTIFLOOD_FILE
from constantes import ANTIFLOOD_MINUTES, ANTIFLOOD_TAG, ANTIFLOOD_TAG_FILE
from constantes import MSG_FLOOD, MSG_AFLOOD_FILE, AFLOOD_MSGS, AFLOOD_TIME, AFLOOD_MINUTES_DEL
# Variables globales:
# Semáforo para ficheros. Usado para controlar la grabación de ficheros en modo asíncrono
semaforo = {
HASHTAG: False,
ANTIFLOOD: False,
ANTIFLOOD_TAG: False,
MSG_FLOOD: False
}
# Inicializamos el subsistema de log para nuestro bot. Todos los errores y las informaciones irán a este fichero.
logging.basicConfig(filename=LOG_FILE, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.INFO)
logger = logging.getLogger(__name__)
# Control del semáforo para grabación de ficheros
def switch_semaforo(estado, fichero):
global semaforo
# Intento de inicio de grabación
if estado == START:
# Esperamos a tener el semáforo libre para pillar hueco
while semaforo[fichero]:
# Esperamos un tiempo aleatorio de 0 a 10
sleep(randint(1, 100) / 100)
# Cuando lo conseguimos, pillamos
semaforo[fichero] = True
# Si queremos cerrar el semáforo...
else:
semaforo[fichero] = False
# Decorador para comprobar si un usuario está autorizado a usar una función (Comandos Admin)
def authorized(func):
@wraps(func)
def wrapped(bot, update, *args, **kwargs):
user_id = update.effective_user.id
chat_id = update.message.chat_id
# Sólo funcionará si se envía desde privado. En público no funcionan los comandos de Admin
if chat_id not in GROUPS_ID.values():
# Si el usuario no es admin, registramos el intento, le respondemos que no puede y anulamos
if user_id not in ADMINS:
logger.info("Intento de ejecución de un comando privilegiado por el Usuario: {}.".format(user_id))
update.message.reply_text("Lo siento, no tienes permiso para ejecutar este comando.")
return
# Si sí que es admin, entonces la ejecutamos
return func(bot, update, *args, **kwargs)
# Si lo intenta ejecutar en el grupo, no lo permitimos.
else:
return
return wrapped
# Decorador para comprobar si un comando se ha ejecutado repetidamente para no permitirlo (AntiFlood)
def antiflood(func):
@wraps(func)
def wrapped(bot, update, *args, **kwargs):
user_id = update.effective_user.id
chat_id = update.message.chat_id
name_func = func.__name__
# Si se ejecuta desde el grupo comprobaremos el flood
if chat_id in GROUPS_ID.values():
funciones_times = read_json(ANTIFLOOD_FILE)
# Si el fichero existe y no está vacío y el canal existe
if funciones_times and str(chat_id) in funciones_times:
# Si la función ya está registrada en el fichero
if name_func in funciones_times[str(chat_id)]:
# Si la función ha sido ejecutada en los últimos minutos, no ejecutamos la función
hora = datetime.datetime.strptime(funciones_times[str(chat_id)][name_func], "%d-%m-%Y %H:%M")
if hora > datetime.datetime.now() - datetime.timedelta(minutes=ANTIFLOOD_MINUTES):
return
# Si aún no se ha registrado el canal, lo registramos
else:
funciones_times[str(chat_id)] = {}
# Y registramos la función ejecutada y la hora
funciones_times[str(chat_id)][name_func] = datetime.datetime.now().strftime("%d-%m-%Y %H:%M")
graba_json(ANTIFLOOD, ANTIFLOOD_FILE, funciones_times)
return func(bot, update, *args, **kwargs)
# Si se envía desde privado, la ejecutamos siempre. Si se quiere hacer flood a si mismo... :-p
else:
return func(bot, update, *args, **kwargs)
return wrapped
# Función para comprobar si la impresión de un hashtag es muy repetida en tiempo (Antiflood)
def antiflood_tags(chat_id, tag):
# Si la ejecución del trag se hace en los grupos
if chat_id in GROUPS_ID.values():
tag_times = read_json(ANTIFLOOD_TAG_FILE)
# Si el fichero existe y no está vacío y el canal existe
if tag_times and str(chat_id) in tag_times:
# Si el tag ya está registrado en el fichero
if tag in tag_times[str(chat_id)]:
# Si el tag ha sido ejecutada en los últimos minutos, no ejecutamos la impresión
hora = datetime.datetime.strptime(tag_times[str(chat_id)][tag], "%d-%m-%Y %H:%M")
if hora > datetime.datetime.now() - datetime.timedelta(minutes=ANTIFLOOD_MINUTES):
return False
# Si ese tag o grupo aún no estaba registrado, lo registramos
else:
tag_times[str(chat_id)] = {}
# Grabamos la hora de ejecución del tag correspondiente
tag_times[str(chat_id)][tag] = datetime.datetime.now().strftime("%d-%m-%Y %H:%M")
graba_json(ANTIFLOOD_TAG, ANTIFLOOD_TAG_FILE, tag_times)
return True
# Si se envía desde privado, la ejecutamos siempre. Si se quiere hacer flood a si mismo... :-p
else:
return True
# Función de ayuda. Aquí iremos poniendo todos los comandos y opciones de los que vayamos dotando al bot.
# Intenta enviarla por privado, si no puede, la envía al grupo con enlace al bot + AntiFlood
@antiflood
def bot_help(bot, update):
# Obtenemos el ID del chat desde donde nos hablan...
chat_id = update.message.chat_id
# Y el privado del usuario
user_id = update.message.from_user.id
# Intentamos enviarle la ayuda por privado...
try:
bot.send_message(chat_id=user_id, text="""
ℹ️ COMANDOS Y FUNCIONES DISPONIBLES ℹ️
▪️/canales
Muestra la relación de canales de Raspito's Family
▪️ /normas
Muestra las reglas comunes a los canales de Raspito's Family
▪️/tags
Muestra la relación de tags con información disponibles
""")
# Si el usuario es un Admin, le mandamos la extensión de ayuda a administradores
if user_id in ADMINS:
bot.send_message(chat_id=user_id, text="🔐 ***ADMIN DETECTED***‼\n"
"Como admin, tienes derecho a añadir, editar y borrar tags.\n"
"Envía /tags para obtener más ayuda sobre cómo hacerlo.",
parse_mode=ParseMode.MARKDOWN)
# Si no hemos podido enviarlo por privado...
except:
update.message.reply_text("Para acceder a la ayuda, pincha en <a href='http://t.me/RaspythonBot'>"
"ESTE ENLACE</a> e iníciame...",
parse_mode=ParseMode.HTML)
# Función de arranque del Bot. Será lo que se muestre cuando un usuario inicie el Bot en privado.
def start(bot, update):
# Obtenemos el ID del chat desde donde nos hablan...
chat_id = update.message.chat_id
# La ayuda sólo la mostraremos si se ejecuta en privado. Si se hace en uno de los canales la ignoraremos.
if chat_id not in GROUPS_ID.values():
update.message.reply_text("🤖 Bienvenido al RaspythonBot 🤖\n"
"Bot creado y pensado para dar soporte al grupo @Raspython\n"
"t.me/Raspython\n"
"Pulsa /help para obtener ayuda sobre lo que puedes hacer")
# Función que muestra la relación de canales de la Raspito's Family
@antiflood
def canales(bot, update):
# Enviamos la lista de canales de la Raspito's Family con control Antiflood
update.message.reply_text("🔸 Relación de los canales de la Raspito's Family 👨👩👧👦\n\n"
"▫️@GrupoRaspberryPI\n"
"Canal precursor de la Raspito's Family. Dialoga sobre los aspectos generales "
"sobre la Raspberry PI, como donde comprarla, que accesorios son recomendables, "
"dudas sobre configuración, etc. En definitiva, todo lo relacionado sobre nuestras "
"PI que no abarquen más específicamente el resto de canales de la familia.\n\n"
"▫️@RaspberryPiMediacenters\n"
"Canal enfocado al uso y disfrute de nuestras Raspberry Pi como centro multimedia.\n\n"
"▫ @RaspberryPiEmuladores\n"
"Canal dedicado a la temática sobre emulación en nuestras Raspberry Pi. Configura tu "
"pequeña Pi como un centro de juego arcade. Disfruta con la emulación!.\n\n"
"▫ @Raspython\n"
"Grupo para debatir, dialogar y preguntar sobre todo lo relacionado con la programación "
"en Python y Raspberry. Este Bot es precisamente un proyecto desarrollado en este canal. "
"Únete y disfruta.\n\n▫ @RaspberryPiOfftopic\n"
"¿Te apetece hablar con los compañeros de la Raspito's family sobre cualquier otro "
"tema?, pues este es tu canal. Entra charla, comparte y disfruta con el resto de "
"compañeros sin límite de temática. (Contenido sexual explícito, violento y de temáticas "
"similares está prohibido)")
# Función que tiene como objeto mostrar las reglas del grupo
@antiflood
def normas(bot, update):
# Enviamos las reglas del grupo con control Antiflood
update.message.reply_text("📜 Las reglas de este grupo son muy sencillas...\n\n"
"⛔️Nada de sexo explícito, ni política, ni religión, ni nada no relacionado "
"con la temática del canal.\n"
"🚫 Tampoco están permitidos los enlaces a otros grupos o páginas webs que cumplan "
"los criterios anteriormente descritos.\n"
"❓ En caso de dudas, consultar con cualquier administrador del grupo.\n\n"
"🤙🏼 Como último requisito, pasarlo bien, disfrutar, compartir y que fluya el buen rollo")
# En esta función trataremos los comandos no reconocidos. De momento lo dejamos en "pass" para ignorarlos
def comando_invalido(bot, update):
pass
# Función en la que filtraremos los datos enviados por el usuario y que trataremos según corresponda
# De momento sólo responde a un pequeño easteregg en honor a "Raspito". Tiene control de flood
@run_async
def trata_texto(bot, update):
chat_id = update.message.chat_id
# user_id = update.message.from_user.id
# Si se está hablando desde uno de los canales del grupo, pasamos el antiflood
if chat_id in GROUPS_ID.values():
msg_antiflood(bot, update)
texto = update.message.text
if re.search(r"(?i)qu[eé] es raspito\b", texto):
update.message.reply_text("...son los padres")
def msg_antiflood(bot, update):
chat_id = update.message.chat_id
user_id = str(update.message.from_user.id)
name = update.message.from_user.name
message_id = update.message.message_id
time = datetime.datetime.now()
user_time = None
# El diccionario flood time tendrá la siguiente estructura:
# ID de usuario : [ Hora de último bloque, Contador mensajes, Warnings, Advertido ]
flood_tab = read_json(MSG_AFLOOD_FILE)
# Si ha habido error en lectura
if flood_tab is None:
logger.info("Error en la lectura del fichero de Aflood de mensajes")
return
# Si el usuario ya está registrado
if user_id in flood_tab:
# Convertimos la hora:
user_time = datetime.datetime.strptime(flood_tab[user_id][0], "%d-%m-%Y %H:%M:%S")
# Comprobamos la hora de su último bloque de mensajes
if user_time > time - datetime.timedelta(seconds=AFLOOD_TIME):
# Si entra dentro del rango de flood, le sumamos uno al contador de mensajes
flood_tab[user_id][1] += 1
elif user_time < time + datetime.timedelta(hours=24):
# Si no, y no han pasado 24 h. le registramos el nuevo bloque de mensajes manteniendo
# los warning ya existentes
flood_tab[user_id] = [time.strftime("%d-%m-%Y %H:%M:%S"), 1, flood_tab[user_id][2],
flood_tab[user_id][3]]
else:
# Si han pasado más de 24 horas, le bajamos un grado de nivel de warning
if flood_tab[user_id][2] >= 2:
warnings = 1
else:
warnings = 0
flood_tab[user_id] = [time.strftime("%d-%m-%Y %H:%M:%S"), 1, warnings, 0]
else:
# Si el usuario no existe, pasamos a registrarlo
flood_tab[user_id] = [time.strftime("%d-%m-%Y %H:%M:%S"), 1, 0, 0]
# Si ha pasado el tiempo de castigo de borrado de mensajes, le bajamos un nivel. A la próxima fuera!
if flood_tab[user_id][2] == 2 and user_time + datetime.timedelta(minutes=AFLOOD_MINUTES_DEL) < time:
flood_tab[user_id][2] = flood_tab[user_id][2] - 1
# Ahora comprobamos cómo está el usuario respecto a contadores y warnings
# Si el contador es igual o mayor que el número máximo de mensajes, aumentamos warnings
if flood_tab[user_id][1] >= AFLOOD_MSGS:
# Si ya ha sido advertido alguna vez del nivel 2:
if flood_tab[user_id][3]:
flood_tab[user_id] = [time.strftime("%d-%m-%Y %H:%M:%S"), 0, 3, flood_tab[user_id][3]]
else:
flood_tab[user_id] = [time.strftime("%d-%m-%Y %H:%M:%S"), 0,
flood_tab[user_id][2] + 1, flood_tab[user_id][3]]
# Y ahora hacemos según tenga el warning
# Si es el primer warning
if flood_tab[user_id][2] == 1 and flood_tab[user_id][1] == 0:
update.message.reply_text("⚠️***PRIMERA ADVERTENCIA*** ⚠️\nEl flood no está permitido. "
"No puedes enviar más de {} mensajes en {} segundos"
.format(AFLOOD_MSGS, AFLOOD_TIME), parse_mode=ParseMode.MARKDOWN)
elif flood_tab[user_id][2] == 2 and flood_tab[user_id][1] == 0:
# Si ya es la segunda vez que está en este nivel en 24h, a la calle
update.message.reply_text("⛔️***SEGUNDA ADVERTENCIA***‼️\nEl flood no está permitido. "
"Los mensajes que envíe durante los próximos {} minutos serán, eliminados.\n‼️"
"***No hay más advertencias, a la siguiente, FUERA***‼️"
.format(AFLOOD_MSGS, AFLOOD_MINUTES_DEL), parse_mode=ParseMode.MARKDOWN)
flood_tab[user_id][3] += 1
elif flood_tab[user_id][2] >= 2:
# Si ya estamos a nivel warning 2, le borramos el mensaje
del_right = bot.delete_message(chat_id=chat_id, message_id=message_id)
if not del_right:
logger.info("Problema al borrar el mensaje {} del usuario {}".format(message_id, user_id))
# Si alcanza warning 3, o es la segunda vez que alcanza warning 2, a la calle!
if flood_tab[user_id][2] > 2 or flood_tab[user_id][2] == 2 and flood_tab[user_id][3] > 1:
bot.send_message(chat_id=chat_id, text="EL USUARIO {}{} HA SIDO EXPULSADO POR FLOOD CONTINUO"
.format(user_id, name))
bot.send_message(chat_id=209200079, text="EL USUARIO {}{} HA SIDO EXPULSADO POR FLOOD CONTINUO"
.format(user_id, name))
bot.kick_chat_member(chat_id=chat_id, user_id=user_id)
graba_json(MSG_FLOOD, MSG_AFLOOD_FILE, flood_tab)
# Función en la que tratamos los hashtags de los mensajes
@run_async
def parsing_hashtag(bot, update):
chat_id = update.message.chat_id
# user_id = update.message.from_user.id
hashtags = update.message.parse_entities('hashtag')
# Pasamos los hashtags a minúsculas y eliminamos duplicados
hashtags = set([x.lower() for x in list(hashtags.values())])
# Leemos la relación de tags de nuestro fichero
# Ahora mismo sólo soportamos un HASHTAG_FILE, pero lo suyo es tener tantos HASHTAG_FILE como canales diferentes
# para que cada grupo pueda administrar sus propios tags. Eso obliga a cambiar las funciones de creación de tags.
bot_tags = read_json(HASHTAG_FILE)
# Si no a habido problemas y tenemos tags en el fichero, los procesamos
if bot_tags:
for h in sorted(hashtags):
if h in bot_tags:
if antiflood_tags(chat_id, h):
update.message.reply_text("***#{}***\n{}".format(h[1:].capitalize(), bot_tags[h]),
parse_mode=ParseMode.MARKDOWN, quote=False,
disable_web_page_preview=True)
# Función que abre el JSON de los Hashtag y devuelve su contenido. En caso de error devolvemos nulo.
def read_json(file):
# Si el fichero no existe, devolvemos un diccionario vacío
if not os.path.isfile(file):
return {}
# Intentamos abrirlo y retornamos el contenido
try:
with open(file, "r") as f:
return json.load(f)
# Si no podemos, retornamos nulo
except:
return None
# Función para añadir tags al fichero
def editing_tags(bot, update):
# Separamos los comandos
comando = update.message.text_markdown
chat_id = update.message.chat_id
# Si sólo ha mandado el comando, mostramos la ayuda
if not len(comando.split()) == 1:
# Si el comando es el de "añadir"
if comando.split()[1].lower() == "add":
# Si faltan parámetros
if len(comando.split()) < 3 or not len(comando.split("@@")) == 2:
tagtoadd = ""
else:
# Separamos tag del texto
tagtoadd = " ".join(comando.split()[2:]).split("@@")
# Si no lo ha mandado correctamente formateado
if not len(tagtoadd) == 2 or not len(tagtoadd[0].split()) == 1:
update.message.reply_text("⚠️ Recuerda, `/tags add #etiqueta@@texto`\n\n",
parse_mode=ParseMode.MARKDOWN)
return
# Separamos tag del texto
tag = comando.split("@@")[0].split()[2]
text = comando.split("@@")[1]
# Comprobamos que es un tag
if tag[0] == "#":
# grabamos(tag,texto)
if graba_tag(tag.lower(), text):
update.message.reply_text("Actualización de hastags correcta 👍🏼")
return
else:
update.message.reply_text("❌ Problema en la grabación de los hashtags")
logger.info("Problema en la grabación de los hashtags en editing_tags")
return
else:
update.message.reply_text("⚠️No has enviado un hashtag. Te faltó la #")
return
elif comando.split()[1].lower() == "list" and len(comando.split()) == 2:
print_list(bot, update)
return
elif comando.split()[1].lower() == "del" and len(comando.split()) == 3:
tag = comando.split()[2]
if tag[0] == "#":
keyboard = [[InlineKeyboardButton("Si", callback_data='Si_delete'),
InlineKeyboardButton("No", callback_data='No_delete')]]
reply_markup = InlineKeyboardMarkup(keyboard)
bot.send_message(chat_id=chat_id, text='¿Quieres eliminar el tag {}?:'
.format(tag), reply_markup=reply_markup)
else:
update.message.reply_text("⚠️No has enviado un hashtag. Te faltó la #")
return
# Si hemos llegado aquí, es que el comando estaba mal formado.
update.message.reply_text("📌 El uso de /tags es el siguiente:\n\n"
"▫️Para añadir o editar un hashtag existente usa:\n`/tags add #etiqueta@@texto`\n"
"▫ Para eliminar un hashtag usa:\n`/tags del #etiqueta`\n"
"▫ Para listar los hashtag establecidos usa:\n`/tags list`\n",
parse_mode=ParseMode.MARKDOWN)
return
# Función que realiza la grabación del JSON con control asíncrono
# Recibe en tipo la variable del semáforo, el fichero y los datos a guardar
def graba_json(tipo, file, datos):
# Solicitamos permiso al semáforo
switch_semaforo(START, tipo)
try:
with open(file, "w") as f:
json.dump(datos, f)
# Liberamos el semáforo
switch_semaforo(END, tipo)
return True
except:
logger.info("Error de grabación al grabar {}.".format(file))
# Liberamos el semáforo
switch_semaforo(END, tipo)
return False
# Función que realiza la grabación del JSON de los tags con control asíncrono
# Recibo el tag a guardar y el texto del mismo
def graba_tag(tag, texto):
# Solicitamos permiso al semáforo
switch_semaforo(START, HASHTAG)
# Leemos las tags del fichero
bot_tags = read_json(HASHTAG_FILE)
# Comprobamos si la lectura ha tenido éxito
if bot_tags is None:
logger.info("Error de lectura del hashtagfile en graba_tag.")
# Liberamos el semáforo
switch_semaforo(END, HASHTAG)
return False
# Actualizamos o creamos el tag con el texto
bot_tags[tag] = texto
try:
with open(HASHTAG_FILE, "w") as f:
json.dump(bot_tags, f)
# Liberamos el semáforo
switch_semaforo(END, HASHTAG)
return True
except:
logger.info("Error de lectura del hashtagfile en graba_tag.")
# Liberamos el semáforo
switch_semaforo(END, HASHTAG)
return False
# Función que recorre el fichero de tags y los manda formateados en columnas
@antiflood
@run_async
def list_tags(bot, update):
# Si lo manda uno de los admins por privado
if update.message.chat_id in ADMINS:
# Le mando a la función avanzada
editing_tags(bot, update)
else:
print_list(bot, update)
def print_list(bot, update):
# Leemos la relación de tags de nuestro fichero
bot_tags = read_json(HASHTAG_FILE)
# Si no a habido problemas y tenemos tags en el fichero, los procesamos
if bot_tags:
update.message.reply_text("👇🏼 A continuación la relación de tags registrados 👇🏼", quote=False)
i = 0
tag_text = ""
for h in sorted(bot_tags):
i += 1
tag_text += "#️⃣" + " " + "`" + h[1:].upper() + "`" + "\n"
if i > 20:
update.message.reply_text(tag_text, parse_mode=ParseMode.MARKDOWN, quote=False)
i = 0
tag_text = ""
if i:
update.message.reply_text(tag_text, parse_mode=ParseMode.MARKDOWN, quote=False)
else:
update.message.reply_text("😣 Actualmente no hay tags registrados")
def delete_tag(bot, update):
query = update.callback_query
tag = "#" + update.callback_query.message.text.split("#")[1][:-2]
if query.data in "Si_delete":
# Solicitamos permiso al semáforo
switch_semaforo(START, HASHTAG)
# Leemos las tags del fichero
bot_tags = read_json(HASHTAG_FILE)
# Comprobamos si la lectura ha tenido éxito
if bot_tags is None:
logger.info("Error de lectura del hashtagfile en delete_tag.")
# Liberamos el semáforo
switch_semaforo(END, HASHTAG)
bot.edit_message_text(text="❌ Problema de lectura del fichero hashtagfile al borrar el tag",
chat_id=query.message.chat_id,
message_id=query.message.message_id)
return
# Buscamos primero el tag, y si no está, retornamos error
if tag.lower() not in bot_tags:
switch_semaforo(END, HASHTAG)
bot.edit_message_text(text="⚠ El tag no se encuentra en la lista. Abortamos...",
chat_id=query.message.chat_id,
message_id=query.message.message_id)
return
# Si está, entonces lo borramos
del bot_tags[tag]
# Y grabamos
try:
with open(HASHTAG_FILE, "w") as f:
json.dump(bot_tags, f)
# Liberamos el semáforo
switch_semaforo(END, HASHTAG)
bot.edit_message_text(text="♻️El tag ha sido eliminado correctamente de la lista! 🗑",
chat_id=query.message.chat_id,
message_id=query.message.message_id)
return
except:
logger.info("Error de lectura del hashtagfile en graba_tag.")
# Liberamos el semáforo
switch_semaforo(END, HASHTAG)
bot.edit_message_text(text="❌ Problema al guardar el fichero hashtagfile al borrar el tag",
chat_id=query.message.chat_id,
message_id=query.message.message_id)
return
else:
bot.edit_message_text(text="✋🏼 Anulado el borrado del tag... ✋🏼",
chat_id=query.message.chat_id,
message_id=query.message.message_id)
def bienvenida(bot, update):
chat_id = update.message.chat_id
if chat_id == GROUPS_ID["Raspython"]:
update.message.reply_text("Bienvenido al grupo de Raspython... DISFRUTA Y PARTICIPA! 👋🏼\n"
"Ábreme un privado <a href='http://t.me/RaspythonBot'>"
"PINCHANDO AQUÍ</a> e iníciame para obtener información...",
parse_mode=ParseMode.HTML)
def idchannel(bot, update):
chat_id = update.message.chat_id
user_id = update.message.from_user.id
name = update.message.from_user.name
update.message.reply_text("{}, tienes el UserID: {} y me escribes desde {}".format(name, user_id, chat_id))
def main():
# Inicializamos nuestro bot
raspython_bot = Updater(token=TOKEN)
# Generamos el dispatcher para recoger los comandos y textos del chat
dispatcher = raspython_bot.dispatcher
# Definición de los comandos
dispatcher.add_handler(CommandHandler("start", start))
dispatcher.add_handler(CommandHandler("help", bot_help))
dispatcher.add_handler(CommandHandler("canales", canales))
dispatcher.add_handler(CommandHandler("normas", normas))
dispatcher.add_handler(CommandHandler("tags", list_tags))
dispatcher.add_handler(CommandHandler("idchannel", idchannel))
# Capturador de nuevos miembros
dispatcher.add_handler(MessageHandler(Filters.status_update.new_chat_members, bienvenida))
# Capturador de comandos no reconocidos
dispatcher.add_handler(MessageHandler(Filters.command, comando_invalido))
# Capturador de textos. Capturaremos aquello que no sean comandos explícitos
# Cuando contienen hashtags
dispatcher.add_handler(MessageHandler(Filters.entity(MessageEntity.HASHTAG), parsing_hashtag))
# Cuando es texto
dispatcher.add_handler(MessageHandler(Filters.text, trata_texto))
# Capturador del menú de borrado de tags (Si y No)
raspython_bot.dispatcher.add_handler(CallbackQueryHandler(delete_tag, pattern="Si_delete|No_delete"))
# Arrancamos el bot, empezamos a capturar peticiones
# Con clean ignoramos los mensajes pendientes...
raspython_bot.start_polling(clean=True)
# Y lo dejamos esperando
raspython_bot.idle()
if __name__ == "__main__":
main()