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

Lexer guess implemented.

Browse Source
This commit is contained in:
Alexandr
2018-07-30 15:28:06 +02:00
parent 643ccfafbc
commit 977f918baa
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
worker.py
View File

@ -13,9 +13,8 @@ import settings
import re import re
from pygments import highlight from pygments import highlight
from pygments.lexers import PythonLexer from pygments.lexers import PythonLexer
from pygments.lexers import guess_lexer, get_lexer_by_name
from pygments.formatters import ImageFormatter from pygments.formatters import ImageFormatter
class MessageWorker: class MessageWorker:
@ -119,15 +118,16 @@ class MessageWorker:
self.send(id=conf_id, msg=message) self.send(id=conf_id, msg=message)
return True return True
if input_message[:5] == '/code': if input_message[:5] == '/code':
print("Going to highlight")
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:] code = msg['message']['text'][6:]
print("Code to highlight: %s" % code) lexer_guess = guess_lexer(code)
highlight(code, PythonLexer(),ImageFormatter(), outfile="code.png") if lexer_guess.name == 'Text only':
lexer_guess = get_lexer_by_name('python')
highlight(code, lexer_guess, ImageFormatter(), outfile="code.png")
self.send_img(conf_id) self.send_img(conf_id)
return True return True
except: except:
@ -146,7 +146,6 @@ class MessageWorker:
user_id = msg['message']['from']['id'] user_id = msg['message']['from']['id']
chat_id = msg['message']['chat']['id'] chat_id = msg['message']['chat']['id']
chat_title = msg['message']['chat']['title'] chat_title = msg['message']['chat']['title']
#print(self.clean_text(text))
except: except:
return False return False