add spoiler bot

bots/migrations/0024_spoilerbotmoduleconfig_spoilermessage.py

@@ -0,0 +1,29 @@
+# Generated by Django 3.0rc1 on 2020-12-11 10:36
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('bots', '0023_overlaybotmoduleconfig_send_avatar_on_start'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='SpoilerBotModuleConfig',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+            ],
+            options={
+                'abstract': False,
+            },
+        ),
+        migrations.CreateModel(
+            name='SpoilerMessage',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('text', models.TextField()),
+            ],
+        ),
+    ]

bots/modules/__init__.py

@@ -3,6 +3,7 @@ from .channel_helper import ChannelHelperBotModuleConfig, QueuedItem, ChannelHel
 from .echo import EchoBotModuleConfig
 from .cyberlina import CyberLinaBotModuleConfig
 from .ping import PingBotModuleConfig
+from .spoiler import SpoilerBotModuleConfig
 
 BOT_MODULES = [EchoBotModuleConfig, ChannelHelperBotModuleConfig, OverlayBotModuleConfig, CyberLinaBotModuleConfig,
-               PingBotModuleConfig]
+               PingBotModuleConfig, SpoilerBotModuleConfig]

bots/modules/spoiler.py

@@ -0,0 +1,36 @@
+from django.db import models
+from telegram import Update, InlineQueryResultArticle, InputTextMessageContent, InlineKeyboardMarkup, \
+    InlineKeyboardButton
+from telegram.ext import Dispatcher, CallbackContext, InlineQueryHandler, CallbackQueryHandler
+
+from bots.models import TelegramBotModuleConfig
+
+
+class SpoilerMessage(models.Model):
+    text = models.TextField()
+
+
+class SpoilerBotModuleConfig(TelegramBotModuleConfig):
+    MODULE_NAME = 'Spoiler'
+
+    def inline_handler(self, update: Update, ctx: CallbackContext):
+        sm = SpoilerMessage.objects.create(text=update.inline_query.query)
+        update.inline_query.answer(results=[InlineQueryResultArticle(
+            id=str(sm.pk),
+            title='Send spoiler',
+            input_message_content=InputTextMessageContent('This is a spoiler'),
+            reply_markup=InlineKeyboardMarkup([InlineKeyboardButton('Read content', callback_data='read {}'.format(sm.pk))])
+        )])
+
+    def read_handler(self, update: Update, ctx: CallbackContext):
+        _, smid = update.callback_query.data.split()
+        try:
+            sm = SpoilerMessage.objects.get(pk=smid)
+        except SpoilerMessage.DoesNotExist:
+            return update.callback_query.answer('Message does not exist')
+        update.callback_query.answer(sm.text, alert=True)
+
+    def build_dispatcher(self, dispatcher: Dispatcher):
+        dispatcher.add_handler(InlineQueryHandler(self.inline_handler))
+        dispatcher.add_handler(CallbackQueryHandler(self.read_handler, pattern=r'^read \d+$'))
+        return dispatcher