fix for prev, minor refactoring

server/custom.py

@@ -35,29 +35,32 @@ def _thread_uniqie_id(bot_id: int, chat_id: int) -> str:
     return f"thread_{bot_id}_{chat_id}"
 
 
-async def message_handler(message: types.Message, *args, **kwargs):
+async def send_user_message(message: types.Message, super_chat_id: int, bot):
-    _logger.info("message handler")
+    """Переслать сообщение от пользователя, добавлять к нему user info при необходимости"""
-    bot = db_bot_instance.get()
+    if bot.enable_additional_info:
+        user_info = "От пользователя: "
+        if message.from_user.full_name:
+            user_info += message.from_user.full_name
+        if message.from_user.username:
+            user_info += " | @" + message.from_user.username
+        user_info += f" | #{message.from_user.id}"
+        new_message = await message.bot.send_message(super_chat_id, text=user_info)
+        await message.copy_to(super_chat_id, reply_to_message_id=new_message.message_id)
+        return new_message
+    else:
+        return await message.forward(super_chat_id)
 
-    if message.text and message.text == "/start":
-        # На команду start нужно ответить, не пересылая сообщение никуда
-        return SendMessage(chat_id=message.chat.id,
-                           text=bot.start_text + ServerSettings.append_text())
 
-    super_chat_id = await bot.super_chat_id()
+async def handle_user_message(message: types.Message, super_chat_id: int, bot):
+    """Обычный пользователь прислал сообщение в бот, нужно переслать его операторам"""
     is_super_group = super_chat_id < 0
 
-    if message.chat.id != super_chat_id:
-        # Это обычный чат
-
     # Проверить, не забанен ли пользователь
     banned = await bot.banned_users.filter(telegram_id=message.chat.id)
     if banned:
         return SendMessage(chat_id=message.chat.id,
                            text="Вы заблокированы в этом боте")
 
-        in_thread = False
-
     # Пересылаем сообщение в супер-чат
     if is_super_group and bot.enable_threads:
         thread_first_message = await _redis.get(_thread_uniqie_id(bot.pk, message.chat.id))
@@ -65,40 +68,32 @@ async def message_handler(message: types.Message, *args, **kwargs):
             # переслать в супер-чат, отвечая на предыдущее сообщение
             try:
                 new_message = await message.copy_to(super_chat_id, reply_to_message_id=int(thread_first_message))
-                    in_thread = True
             except exceptions.BadRequest:
-                    new_message = await message.forward(super_chat_id)
+                new_message = await send_user_message(message, super_chat_id, bot)
                 await _redis.set(_thread_uniqie_id(bot.pk, message.chat.id), new_message.message_id,
                                  pexpire=ServerSettings.thread_timeout_ms())
         else:
             # переслать супер-чат
-                new_message = await message.forward(super_chat_id)
+
+            new_message = await send_user_message(message, super_chat_id, bot)
             await _redis.set(_thread_uniqie_id(bot.pk, message.chat.id), new_message.message_id,
                              pexpire=ServerSettings.thread_timeout_ms())
     else:  # личные сообщения не поддерживают потоки сообщений: простой forward
-            new_message = await message.forward(super_chat_id)
+        new_message = await send_user_message(message, super_chat_id, bot)
 
     await _redis.set(_message_unique_id(bot.pk, new_message.message_id), message.chat.id,
                      pexpire=ServerSettings.redis_timeout_ms())
 
-        if bot.enable_additional_info and not in_thread:
-            user_info = "От пользователя: "
-            if message.from_user.full_name:
-                user_info += message.from_user.full_name
-            if message.from_user.username:
-                user_info += " | @" + message.from_user.username
-            user_info += f" | #{message.from_user.id}"
-            await message.bot.send_message(super_chat_id, text=user_info, reply_to_message_id=new_message.message_id)
-
     bot.incoming_messages_count = F("incoming_messages_count") + 1
     await bot.save(update_fields=["incoming_messages_count"])
 
     # И отправить пользователю специальный текст, если он указан
     if bot.second_text:
         return SendMessage(chat_id=message.chat.id, text=bot.second_text)
-    else:
-        # Это супер-чат
 
+
+async def handle_operator_message(message: types.Message, super_chat_id: int, bot):
+    """Оператор написал что-то, нужно переслать сообщение обратно пользователю, или забанить его и т.д."""
     if message.reply_to_message:
         # В супер-чате кто-то ответил на сообщение пользователя, нужно переслать тому пользователю
         chat_id = await _redis.get(_message_unique_id(bot.pk, message.reply_to_message.message_id))
@@ -142,6 +137,25 @@ async def message_handler(message: types.Message, *args, **kwargs):
             return SendMessage(chat_id=message.chat.id, text=bot.second_text)
 
 
+async def message_handler(message: types.Message, *args, **kwargs):
+    _logger.info("message handler")
+    bot = db_bot_instance.get()
+
+    if message.text and message.text == "/start":
+        # На команду start нужно ответить, не пересылая сообщение никуда
+        return SendMessage(chat_id=message.chat.id,
+                           text=bot.start_text + ServerSettings.append_text())
+
+    super_chat_id = await bot.super_chat_id()
+
+    if message.chat.id != super_chat_id:
+        # Это обычный чат
+        return await handle_user_message(message, super_chat_id, bot)
+    else:
+        # Это супер-чат
+        return await handle_operator_message(message, super_chat_id, bot)
+
+
 async def receive_invite(message: types.Message):
     bot = db_bot_instance.get()
     for member in message.new_chat_members: