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

/code feature. send coloryzed code snipet as image

Browse Source
This commit is contained in:
root
2018-07-27 15:20:40 +02:00
parent 3aa7982116
commit 643ccfafbc
2 changed files with 27 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -4,3 +4,4 @@ assets/cert*
main.db main.db
main.db-journal main.db-journal
settings.ini settings.ini
code.png

27
worker.py
View File

@ -7,10 +7,17 @@ from database import DataBase
import os import os
import urllib.request import urllib.request
from urllib.parse import urlencode from urllib.parse import urlencode
import requests
import json import json
import settings import settings
import re import re
from pygments import highlight
from pygments.lexers import PythonLexer
from pygments.formatters import ImageFormatter
class MessageWorker: class MessageWorker:
def __init__(self, db, stop_words = 'assets/stop-word.ru', settings = settings): def __init__(self, db, stop_words = 'assets/stop-word.ru', settings = settings):
self.stop_words = stop_words self.stop_words = stop_words
@ -111,6 +118,18 @@ class MessageWorker:
message += ' @%s ' % (user[0]) message += ' @%s ' % (user[0])
self.send(id=conf_id, msg=message) self.send(id=conf_id, msg=message)
return True return True
if input_message[:5] == '/code':
print("Going to highlight")
conf_id = msg['message']['chat']['id']
user_id = msg['message']['from']['id']
chat_title = msg['message']['chat']['title']
self.db.add_conf(conf_id, chat_title)
if len(msg['message']['text'][6:]) < 10000:
code = msg['message']['text'][6:]
print("Code to highlight: %s" % code)
highlight(code, PythonLexer(),ImageFormatter(), outfile="code.png")
self.send_img(conf_id)
return True
except: except:
return False return False
try: try:
@ -174,7 +193,7 @@ class MessageWorker:
return collection return collection
def send(self, id, msg): def send(self, id, msg):
print(msg) #print(msg)
url = self.telegram_api + 'bot' + self.telegram_key + '/sendMessage' url = self.telegram_api + 'bot' + self.telegram_key + '/sendMessage'
post_fields = { post_fields = {
'text': msg, 'text': msg,
@ -186,3 +205,9 @@ class MessageWorker:
request = urllib.request.Request(url, urlencode(post_fields).encode()) request = urllib.request.Request(url, urlencode(post_fields).encode())
json = urllib.request.urlopen(request).read().decode() json = urllib.request.urlopen(request).read().decode()
return json return json
def send_img(self, id):
url = self.telegram_api + 'bot' + self.telegram_key + '/sendPhoto'
data = {'chat_id': id}
files = {'photo': open('code.png', 'rb')}
r = requests.post(url, files=files, data=data)