-
Notifications
You must be signed in to change notification settings - Fork 0
/
reborn.py
247 lines (184 loc) · 7.89 KB
/
reborn.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
# encoding=utf-8
import api
import importlib
import logging
import re
import config
import threading
import traceback
import sys
from time import gmtime
import datetime as dt
from calendar import timegm
from random import random,randint
logging.basicConfig(format='%(asctime)s.%(msecs)d %(levelname)-8s [%(filename)s:%(lineno)d] %(message)s',
datefmt='%d-%m-%Y:%H:%M:%S',
level=logging.INFO, handlers=[logging.FileHandler("log.log", "a", "utf-8")])
def msg_type(msg):
""" Retorna se é texto, foto, áudio, vídeo """
if "text" in msg:
return "text"
elif "photo" in msg:
return "photo"
elif "voice" in msg:
return "voice"
elif "video" in msg:
return "video"
elif "document" in msg:
return "document"
elif "audio" in msg:
return "audio"
elif "sticker" in msg:
return "sticker"
elif "video_note" in msg:
return "video note"
else:
return "outra coisa"
def msg_origin(msg):
""" Mensagem privada ou mensagem de grupo"""
return msg["chat"]["type"]
def log(msg):
""" Loga pra arquivo tudo o que acontecer. """
if type(msg) is str:
logging.info(msg)
print(msg)
else:
origin = msg_origin(msg)
message_type = msg_type(msg)
log_str = "[" + dt.datetime.fromtimestamp(int(msg["date"])).strftime('%Y-%m-%d %H:%M:%S') + "] "
log_str += msg["from"]["first_name"] + "(" + str(msg["from"]["id"]) + ")" + " enviou " + message_type + " "
# Suportando supergroups e outros no log
if origin == "group" or origin == "supergroup" or origin == "channel":
log_str += "em \"" + msg["chat"]["title"] + "\" (" + str(msg["chat"]["id"]) + ")"
elif origin == "private":
log_str += "em PRIVADO"
if message_type == "text":
log_str += ": " + msg["text"]
#if message_type == "audio":
# print(msg)
logging.info(log_str)
print(log_str)
def is_sudoer(id):
return id in config.config["sudoers"]
def is_authorized(msg):
if msg["from"]["id"] in config.config["authorized_users"]:
return True
elif msg["from"]["id"] in config.config["automatically_authorized_users"]:
return True # Para usuários autorizados automaticamente com base no pertencimento
# a grupos. Por ora, só podem ser removidos manualmente
else:
# Como eu não sei a ID de todos os indivíduos romanos, resolvi fazer uma query para ver se o caboclo
# pertence ao chat...
if isAuthorizedGroupMember(msg["from"]["id"]) is True:
# Adiciona o indivíduo de vez, economizando essa consulta burra e redundante
config.config["automatically_authorized_users"].append(msg["from"]["id"])
config.save_config()
log(msg["from"]["first_name"] + "(" + str(msg["from"]["id"]) + ")" + \
" foi adicionado à lista de usuários autorizados. ")
return True
else:
return False
# Verifica se o caboclo faz parte de algum dos chats autorizados
def isAuthorizedGroupMember(user_id):
for gid in config.config["authorized_chats"]:
if api.isGroupMember(gid, user_id):
return True
return False # Solução estranha, porém funcional e razoavelmente elegante
def msg_matches(msg_text):
for query, plugin in config.plugins.items():
pattern = re.compile(query)
###print(query)
match = pattern.search(msg_text)
if match:
if query != "^(.*)$":
log("MATCH! Plugin: " + plugin)
return plugin, match
# Hardcodando o plugin multipurpose, para evitar
# que ele continue causando problemas ao comando !plugin
return "multipurpose", ""
def on_msg_received(msg):
""" Callback pra quando uma mensagem é recebida. """
if is_authorized(msg):
log(msg)
tipo = msg_type(msg)
if tipo == "text":
plugin_match, matches = msg_matches(msg["text"])
if plugin_match is not None and matches is not None:
loaded = importlib.import_module("plugins." + plugin_match)
try:
loaded.on_msg_received(msg, matches)
except:
# Hardcodando aviso de erro. Não serve pra prevenir.
exc_type, exc_value, exc_traceback = sys.exc_info()
printable = repr(traceback.format_exception(exc_type, exc_value, exc_traceback))
for sudoer in config.config["sudoers"]:
api.send_message(sudoer, "ME CAPOTARO AQUI PORRA \n\n" + printable)
raise
elif tipo != "text" and tipo != "outra coisa":
# Garante as estatísticas para o que não for texto
stt = importlib.import_module("plugins.stats")
stt.do_statistics(msg)
# Eu provavelmente deveria fazer um tratamento de erros aqui, mas o
# plugin em questão é meio que parte do core, então tanto faz...
else:
log("Mensagem não autorizada de " + msg["from"]["first_name"] + " (" + str(msg["from"]["id"]) + ")")
def on_old_msg_received(msg):
""" Callback pra quando uma mensagem ANTIGA é recebida. """
if is_authorized(msg):
log(msg)
tipo = msg_type(msg)
# Tudo vira estatística...
if tipo != "outra coisa":
# Mensagens antigas são postas nas estatísticas
stt = importlib.import_module("plugins.stats")
stt.do_statistics(msg)
else:
log("Mensagem não autorizada de " + msg["from"]["first_name"] + " (" + str(msg["from"]["id"]) + ")")
def on_msg_edited(msg):
""" Callback que define o que acontecerá quando uma mensagem for editada. """
pass
def on_callback_query(msg):
""" Callback que define o que acontecerá quando um dado de um InlineKeyboardButton for recebido. """
for plugin in config.config["callback_query_plugins"]:
loaded = importlib.import_module("plugins." + plugin)
loaded.on_callback_query(msg)
def start_longpoll():
""" Inicia longpolling do get_updates. """
most_recent = 0
while True:
updates = api.get_updates(offset=most_recent)
if updates is not None:
for update in updates:
if "message" in update and timegm(gmtime()) - update["message"]["date"] < 10:
on_msg_received(update["message"])
elif "edited_message" in update and timegm(gmtime()) - update["edited_message"]["date"] < 10:
on_msg_edited(update["edited_message"])
elif "callback_query" in update:
on_callback_query(update["callback_query"])
elif "message" in update and timegm(gmtime()) - update["message"]["date"] > 10:
# Mensagens antigas poderão ser processadas...
# A priori, isto servirá para manter estatísticas mesmo
# quanto o bot não conseguir se manter online
on_old_msg_received(update["message"])
else:
log("Mensagem desconhecida; ignorando.")
most_recent = update["update_id"] + 1
# De improviso, implementa um pedido de WS
fenestrao = "-1001049208929"
agora = dt.datetime.now()
probab = randint(0,9999)
#if probab == 9999:
# send_message(fenestrao, "Já assistiu Diamond is Ubreakable, @Daniele_Harai?")
def start_plugins():
for match, plugin in config.plugins.items():
loaded = importlib.import_module("plugins." + plugin)
if hasattr(loaded, "run"):
thread = threading.Thread(name=plugin, target=loaded.run)
thread.start()
def main():
""" Entry point né porra. """
log("Iniciando sessão")
start_plugins()
start_longpoll()
if __name__ == "__main__":
main()