pooling update handlers, add overlay bot module

bots/modules/__init__.py

@@ -1,4 +1,5 @@
+from bots.modules.overlay import OverlayBotModuleConfig
 from .channel_helper import ChannelHelperBotModuleConfig
 from .echo import EchoBotModuleConfig
 
-BOT_MODULES = [EchoBotModuleConfig, ChannelHelperBotModuleConfig]
+BOT_MODULES = [EchoBotModuleConfig, ChannelHelperBotModuleConfig, OverlayBotModuleConfig]

bots/modules/overlay.py

@@ -0,0 +1,48 @@
+import os
+from tempfile import TemporaryDirectory
+from time import sleep
+
+from PIL import Image
+from django.db import models
+from telegram import Update
+from telegram.ext import Dispatcher, CallbackContext, MessageHandler, Filters, CommandHandler
+
+from bots.models import TelegramBotModuleConfig
+
+
+class OverlayBotModuleConfig(TelegramBotModuleConfig):
+    start_text = models.TextField(null=True, blank=True)
+    type_error_text = models.TextField(null=True, blank=True)
+    comment = models.TextField(blank=True, null=True)
+    image = models.ImageField()
+
+    MODULE_NAME = 'Overlay'
+
+    def start_handler(self, update: Update, ctx: CallbackContext):
+        if self.start_text:
+            update.effective_message.reply_text(self.start_text)
+
+    def message_handler(self, update: Update, ctx: CallbackContext):
+        if self.type_error_text:
+            update.effective_message.reply_text(self.type_error_text)
+
+    def photo_handler(self, update: Update, ctx: CallbackContext):
+        with TemporaryDirectory() as d:
+            src = os.path.join(d, 'src.jpg')
+            out = os.path.join(d, 'out.png')
+            q = update.effective_message.photo[-1].get_file().download(src)
+            im = Image.open(q).convert('RGBA')  # type: Image.Image
+            overlay = Image.open(self.image.path).convert('RGBA')  # type: Image.Image
+            w, h = im.size
+            min_side = min(w, h)
+            overlay = overlay.resize((min_side, min_side), Image.LANCZOS)
+            im.paste(overlay, ((w - min_side) // 2, (h - min_side) // 2), overlay)
+            im.save(out)
+            update.effective_message.reply_photo(open(out, 'rb'), caption=self.comment)
+        raise Exception('hi')
+
+    def build_dispatcher(self, dispatcher: Dispatcher):
+        dispatcher.add_handler(CommandHandler('start', self.start_handler))
+        dispatcher.add_handler(MessageHandler(Filters.photo, self.photo_handler))
+        dispatcher.add_handler(MessageHandler(Filters.all, self.message_handler))
+        return dispatcher