add stickers bot
This commit is contained in:
parent
509eb027e5
commit
d63d9268da
@ -6,6 +6,7 @@ from .cyberlina import CyberLinaBotModuleConfig
|
|||||||
from .ping import PingBotModuleConfig
|
from .ping import PingBotModuleConfig
|
||||||
from .robot import RobotBotModuleConfig
|
from .robot import RobotBotModuleConfig
|
||||||
from .spoiler import SpoilerBotModuleConfig
|
from .spoiler import SpoilerBotModuleConfig
|
||||||
|
from .stickers import StickersBotModuleConfig
|
||||||
|
|
||||||
BOT_MODULES = [EchoBotModuleConfig, ChannelHelperBotModuleConfig, OverlayBotModuleConfig, CyberLinaBotModuleConfig,
|
BOT_MODULES = [EchoBotModuleConfig, ChannelHelperBotModuleConfig, OverlayBotModuleConfig, CyberLinaBotModuleConfig,
|
||||||
PingBotModuleConfig, SpoilerBotModuleConfig, RobotBotModuleConfig, FeedbackBotModuleConfig]
|
PingBotModuleConfig, SpoilerBotModuleConfig, RobotBotModuleConfig, FeedbackBotModuleConfig, StickersBotModuleConfig]
|
||||||
|
42
bots/modules/stickers.py
Normal file
42
bots/modules/stickers.py
Normal file
@ -0,0 +1,42 @@
|
|||||||
|
import tempfile
|
||||||
|
|
||||||
|
from PIL import Image
|
||||||
|
from telegram import Update
|
||||||
|
from telegram.ext import Dispatcher, CallbackContext, MessageHandler, Filters
|
||||||
|
|
||||||
|
from bots.models import TelegramBotModuleConfig
|
||||||
|
|
||||||
|
|
||||||
|
class StickersBotModuleConfig(TelegramBotModuleConfig):
|
||||||
|
MODULE_NAME = 'Stickers'
|
||||||
|
|
||||||
|
def resize_image(self, update: Update, ctx: CallbackContext):
|
||||||
|
doc = update.message.document
|
||||||
|
if update.message.photo:
|
||||||
|
with tempfile.NamedTemporaryFile(suffix='.jpg') as f:
|
||||||
|
f.write(update.message.photo[-1].get_file().download_as_bytearray())
|
||||||
|
f.seek(0)
|
||||||
|
im = Image.open(f).convert('RGBA')
|
||||||
|
else:
|
||||||
|
if doc.mime_type == 'image/png':
|
||||||
|
suffix = '.png'
|
||||||
|
elif doc.mime_type == 'image/jpeg':
|
||||||
|
suffix = '.jpg'
|
||||||
|
else:
|
||||||
|
return
|
||||||
|
with tempfile.NamedTemporaryFile(suffix=suffix) as f:
|
||||||
|
f.write(doc.get_file().download_as_bytearray())
|
||||||
|
f.seek(0)
|
||||||
|
im = Image.open(f).convert('RGBA')
|
||||||
|
width, height = im.size
|
||||||
|
im.thumbnail((512, 512), Image.ANTIALIAS)
|
||||||
|
target_width, target_height = im.size
|
||||||
|
with tempfile.NamedTemporaryFile(suffix='.png') as f:
|
||||||
|
im.save(f, 'png')
|
||||||
|
f.seek(0)
|
||||||
|
update.message.reply_document(f, caption=f'{width}x{height} -> {target_width}x{target_height}')
|
||||||
|
|
||||||
|
def build_dispatcher(self, dispatcher: Dispatcher):
|
||||||
|
dispatcher.add_handler(MessageHandler(Filters.document, self.resize_image))
|
||||||
|
dispatcher.add_handler(MessageHandler(Filters.photo, self.resize_image))
|
||||||
|
return dispatcher
|
Loading…
Reference in New Issue
Block a user