Hexor/conf_bot
1
0
Fork 0
mirror of https://github.com/house-of-vanity/conf_bot.git synced 2025-07-06 22:34:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix @here func. Now uses id instead of username.

Browse Source
This commit is contained in:
Alexandr
2019-01-09 14:39:36 +03:00
parent e1e6ee1e9d
commit 7f4d4ced56
2 changed files with 35 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
database.py
View File

@ -161,7 +161,7 @@ class DataBase:
def here(self, user_id, conf_id): def here(self, user_id, conf_id):
sql = """ sql = """
SELECT DISTINCT(u.username) FROM relations r SELECT DISTINCT(u.username), u.id, u.first_name FROM relations r
LEFT JOIN user u LEFT JOIN user u
ON u.id = r.user_id ON u.id = r.user_id
LEFT JOIN conf c LEFT JOIN conf c

52
worker.py
View File

@ -48,12 +48,35 @@ class MessageWorker:
except ValueError: except ValueError:
return False return False
def colorize(self, code):
code_tag = code[code.rfind('#') + 1:]
try:
lexer = get_lexer_by_name(code_tag)
code = code[:code.rfind('#')]
except:
lexer = guess_lexer(code)
print("Lexer is defined as %s" % lexer)
if lexer.name == 'Text only':
lexer = get_lexer_by_name('python')
try:
highlight(code, lexer, ImageFormatter(
font_size=16,
line_number_bg="#242e0c",
line_number_fg="#faddf2",
line_number_bold=True,
font_name='DejaVuSansMono',
style=get_style_by_name('monokai')), outfile="code.png")
except Exception as e:
print(e)
return e
def handleUpdate(self, msg): def handleUpdate(self, msg):
try: try:
try: try:
input_message = msg['message']['text'] input_message = msg['message']['text']
if ('@here' in input_message) or (' @'+self.me['result']['username'] in input_message): if ('@here' in input_message) or (' @'+self.me['result']['username'] in input_message):
if str(msg['message']['chat']['id']) != "-1001233797421": if str(msg['message']['chat']['id']) != "-1001233797421":
print("@here isn't available for '%s' (%s)" % (msg['message']['chat']['title'], msg['message']['chat']['id']))
return return
conf_id = msg['message']['chat']['id'] conf_id = msg['message']['chat']['id']
user_id = msg['message']['from']['id'] user_id = msg['message']['from']['id']
@ -69,7 +92,7 @@ class MessageWorker:
conf_id=conf_id conf_id=conf_id
) )
for user in users: for user in users:
message += ' @%s ' % (user[0]) message += ' [%s](tg://user?id=%s)' % (user[2], user[1])
self.send(id=conf_id, msg=message) self.send(id=conf_id, msg=message)
return True return True
@ -179,29 +202,17 @@ class MessageWorker:
return True return True
if input_message[:5] == '/code': if input_message[:5] == '/code':
# codefunc
conf_id = msg['message']['chat']['id'] conf_id = msg['message']['chat']['id']
user_id = msg['message']['from']['id'] user_id = msg['message']['from']['id']
chat_title = msg['message']['chat']['title'] chat_title = msg['message']['chat']['title']
self.db.add_conf(conf_id, chat_title) self.db.add_conf(conf_id, chat_title)
if len(msg['message']['text'][6:]) < 10000: if len(msg['message']['text'][6:]) < 10000:
code = msg['message']['text'][6:] print(msg['message']['text'][6:])
code_tag = code[code.rfind('#') + 1:]
try: try:
lexer = get_lexer_by_name(code_tag) self.colorize(msg['message']['text'][6:])
code = code[:code.rfind('#')] except Exception as e:
print("Lexer is defined as %s" % lexer) print(e)
except:
lexer = guess_lexer(code)
print("lexer is %s" % lexer)
if lexer.name == 'Text only':
lexer = get_lexer_by_name('python')
highlight(code, lexer, ImageFormatter(
font_size=16,
line_number_bg="#242e0c",
line_number_fg="#faddf2",
line_number_bold=True,
font_name='DejaVuSansMono',
style=get_style_by_name('monokai')), outfile="code.png")
self.send_img(conf_id) self.send_img(conf_id)
return True return True
except: except:
@ -282,10 +293,15 @@ class MessageWorker:
return json return json
def send_img(self, id): def send_img(self, id):
print('Going to send code.png to %s' % id)
try:
url = self.telegram_api + 'bot' + self.telegram_key + '/sendPhoto' url = self.telegram_api + 'bot' + self.telegram_key + '/sendPhoto'
data = {'chat_id': id} data = {'chat_id': id}
files = {'photo': open('code.png', 'rb')} files = {'photo': open('code.png', 'rb')}
r = requests.post(url, files=files, data=data) r = requests.post(url, files=files, data=data)
print(r)
except Exception as e:
print(e)
def cron_timer(self): def cron_timer(self):
alerts = self.db.get_alert() alerts = self.db.get_alert()