cyberlina bot

bots/modules/__init__.py

@@ -1,5 +1,6 @@
 from .overlay import OverlayBotModuleConfig
 from .channel_helper import ChannelHelperBotModuleConfig, QueuedItem
 from .echo import EchoBotModuleConfig
+from .cyberlina import CyberLinaBotModuleConfig
 
-BOT_MODULES = [EchoBotModuleConfig, ChannelHelperBotModuleConfig, OverlayBotModuleConfig]
+BOT_MODULES = [EchoBotModuleConfig, ChannelHelperBotModuleConfig, OverlayBotModuleConfig, CyberLinaBotModuleConfig]

bots/modules/cyberlina.py

@@ -0,0 +1,128 @@
+import os
+from datetime import timedelta, datetime, time
+from random import choice, seed
+from uuid import uuid4
+
+import humanize
+from django.db import models
+from django.utils.timezone import localdate, now, make_aware
+from jsoneditor.forms import JSONEditor
+from jsonfield import JSONField
+from telegram import Update, Chat, User, InlineQueryResultArticle, InputTextMessageContent
+from telegram.error import BadRequest
+from telegram.ext import Dispatcher, CallbackContext, MessageHandler, Filters, CommandHandler, InlineQueryHandler
+from telegram.utils.helpers import mention_html
+
+from bots.models import TelegramBotModuleConfig
+
+
+class CyberLinaBotModuleConfig(TelegramBotModuleConfig):
+    first_part = JSONField(default='{"items": []}')
+    second_part = JSONField(default='{"items": []}')
+    third_part = JSONField(default='{"items": []}')
+    emoji = JSONField(default='{"items": []}')
+    already_ran = JSONField(default='{"items": []}')
+    welcome_reactions = JSONField(default='{"items": []}')
+    inline_reactions = JSONField(default='{"items": []}')
+
+    MODULE_NAME = 'Киберлиночка'
+    CUSTOM_WIDGETS = {
+        'first_part': JSONEditor(),
+        'second_part': JSONEditor(),
+        'third_part': JSONEditor(),
+        'emoji': JSONEditor(),
+        'already_ran': JSONEditor(),
+        'welcome_reactions': JSONEditor(),
+        'inline_reactions': JSONEditor(),
+    }
+
+    def message_handler(self, update: Update, ctx: CallbackContext):
+        if not update.effective_chat or not update.effective_user:
+            return
+        CyberLinaUser.from_tg_obj(self, update.effective_chat, update.effective_user)
+
+    def goodmorning_handler(self, update: Update, ctx: CallbackContext):
+        if not all([self.first_part['items'], self.second_part['items'],
+                    self.third_part['items'], self.emoji['items']]):
+            return update.effective_message.reply_text('Я не настроена :c')
+        seed(os.urandom(128))
+        self.message_handler(update, ctx)
+        chat = self.chats.get(chat_id=update.effective_chat.id)
+        if chat.last_run and (chat.last_run >= localdate() or
+                              chat.last_run + timedelta(1) == localdate() and now().hour < 6):
+            humanize.i18n.activate('ru_RU')
+            time_left = make_aware(datetime.combine(chat.last_run + timedelta(1), time(6, 0))) - now()
+            return update.effective_message.reply_text(
+                choice(self.already_ran['items']).format(
+                    name=chat.last_choice.name,
+                    time=humanize.naturaldelta(time_left)
+                )
+            )
+        while True:
+            user = chat.users.order_by('?').first()  # type: CyberLinaUser
+            if not user:
+                return update.effective_message.reply_text('Нет известных юзеров в чате')
+            try:
+                member = ctx.bot.get_chat_member(chat.chat_id, user.user_id)
+                CyberLinaUser.from_tg_obj(self, update.effective_chat, member.user)
+                break
+            except BadRequest:
+                user.delete()
+        msg = '{}, {}! {}, {} {}'.format(
+            choice(self.first_part['items']),
+            choice(self.second_part['items']),
+            mention_html(user.user_id, user.name),
+            choice(self.third_part['items']),
+            choice(self.emoji['items']),
+        )
+        update.effective_chat.send_message(msg, parse_mode='html')
+        chat.last_run = localdate()
+        chat.last_choice = user
+        chat.save()
+
+    def inline_query_handler(self, update: Update, ctx: CallbackContext):
+        if not self.inline_reactions:
+            return
+        seed(os.urandom(128))
+        results = [
+            InlineQueryResultArticle(
+                id=uuid4(),
+                title='Не нажимай >_<',
+                input_message_content=InputTextMessageContent(choice(self.inline_reactions['items']))
+            )
+        ]
+        update.inline_query.answer(results)
+
+    def build_dispatcher(self, dispatcher: Dispatcher):
+        dispatcher.add_handler(CommandHandler('goodmorning', self.goodmorning_handler))
+        dispatcher.add_handler(MessageHandler(Filters.all, self.message_handler))
+        dispatcher.add_handler(InlineQueryHandler(self.inline_query_handler))
+        return dispatcher
+
+
+class CyberLinaChat(models.Model):
+    config = models.ForeignKey(CyberLinaBotModuleConfig, on_delete=models.CASCADE, related_name='chats')
+    name = models.TextField()
+    chat_id = models.BigIntegerField(db_index=True)
+    last_run = models.DateField(null=True, blank=True)
+    last_choice = models.ForeignKey('CyberLinaUser', on_delete=models.SET_NULL, null=True, blank=True, related_name='+')
+
+    class Meta:
+        unique_together = 'config', 'chat_id',
+
+
+class CyberLinaUser(models.Model):
+    chat = models.ForeignKey(CyberLinaChat, on_delete=models.CASCADE, related_name='users')
+    user_id = models.BigIntegerField(db_index=True)
+    name = models.TextField()
+
+    @staticmethod
+    def from_tg_obj(config: CyberLinaBotModuleConfig, chat: Chat, user: User):
+        chat_title = chat.title or user.full_name
+        chat, _ = CyberLinaChat.objects.update_or_create(config=config, chat_id=chat.id,
+                                                         defaults={'name': chat_title})
+        CyberLinaUser.objects.update_or_create(chat=chat, user_id=user.id,
+                                               defaults={'name': user.full_name})
+
+    class Meta:
+        unique_together = 'chat', 'user_id',