-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.py
114 lines (91 loc) · 3.8 KB
/
main.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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""Simple Bot to reply to Telegram messages.
This program is dedicated to the public domain under the CC0 license.
This Bot uses the Updater class to handle the bot.
First, a few handler functions are defined. Then, those functions are passed to
the Dispatcher and registered at their respective places.
Then, the bot is started and runs until we press Ctrl-C on the command line.
Usage:
Basic Echobot example, repeats messages.
Press Ctrl-C on the command line or send a signal to the process to stop the
bot.
"""
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
import logging
import os
import errno
import uuid
from PIL import Image
import requests
# Enable logging
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.INFO)
logger = logging.getLogger(__name__)
TOKEN = ""
PATH_DIRECTORY = "tmp"
def start(bot, update):
"""Send a message when the command /start is issued."""
update.message.reply_text('Hello dear. If you want to start send me a sticker and I will give you the png without touching the trasparency!')
def help(bot, update):
"""Send a message when the command /help is issued."""
update.message.reply_text('Send me a sticker!')
def download_sticker(stickerId):
url = 'https://api.telegram.org/bot{}/getFile?file_id={}'.format(TOKEN, stickerId)
logger.info(url)
r = requests.get(url)
if r.status_code != 200:
return {'success': False, 'msg': 'Connection error'}
data = r.json()
if data['ok']:
url = 'https://api.telegram.org/file/bot{}/{}'.format(TOKEN, data['result']['file_path'])
logger.info(url)
response = requests.get(url)
if response.status_code == 200:
path = PATH_DIRECTORY + '/' +uuid.uuid4().hex[0:8]+'.webp'
with open(path, 'wb') as f:
f.write(response.content)
return {'success': True, 'msg': 'OK', 'path': path}
def convert_png(path):
im = Image.open(path)
im.load()
alpha = im.split()[-1]
im = im.convert('RGB').convert('P', palette=Image.ADAPTIVE, colors=255)
mask = Image.eval(alpha, lambda a: 255 if a <=128 else 0)
im.paste(255, mask)
newPath = path.replace(".webp",".png")
im.save(newPath, transparency=255)
return newPath
def stickers(bot, update):
update.message.reply_text('Well, let me do some nerdy operation 🤓')
result = download_sticker(update.message.sticker.file_id)
if result['success']:
image = convert_png(result['path'])
bot.send_document(chat_id=update.message.chat.id, document=open(image, 'rb'))
bot.send_message(chat_id=update.message.chat.id, text='Here there is your sticker!\nThe associated emoji is {}!'.format(update.message.sticker.emoji))
else:
bot.send_mesage(chat_idupdate.message.chat.id, text='There was an error while processing your request... Try again!')
def error(bot, update, error):
"""Log Errors caused by Updates."""
logger.warning('Update "%s" caused error "%s"', update, error)
def main():
"""Start the bot."""
try:
os.makedirs(PATH_DIRECTORY)
except OSError as exception:
if exception.errno != errno.EEXIST:
raise
updater = Updater(TOKEN)
dp = updater.dispatcher
dp.add_handler(CommandHandler("start", start))
dp.add_handler(CommandHandler("help", help))
dp.add_handler(MessageHandler(Filters.sticker, stickers))
dp.add_error_handler(error)
# Start the Bot
updater.start_polling()
# Run the bot until you press Ctrl-C or the process receives SIGINT,
# SIGTERM or SIGABRT. This should be used most of the time, since
# start_polling() is non-blocking and will stop the bot gracefully.
updater.idle()
if __name__ == '__main__':
main()