wip

src/ocab_modules/standard/message_processing/message_api.py

@@ -0,0 +1,117 @@
+# flake8: noqa
+
+from aiogram import Bot, F, Router, types
+
+from ocab_core.logger import log
+from ocab_modules.external.yandexgpt.handlers import answer_to_message
+from ocab_modules.standard.config.config import (
+    get_aproved_chat_id,
+    get_yandexgpt_in_words,
+    get_yandexgpt_start_words,
+)
+from ocab_modules.standard.database.db_api import *
+
+
+async def chat_check(message: types.Message):
+    # Проверка наличия id чата в базе данных чатов
+    # Если чата нет в базе данных, то проверяем его в наличии в конфиге и если он там есть то добавляем его в БД
+    # Если чат есть в базе данных, то pass
+    if get_chat(message.chat.id) is None:
+        if message.chat.id in get_aproved_chat_id():
+            # print(f"Chat in approve list: {message.chat.id} {message.chat.title}")
+            await log(f"Chat in approve list: {message.chat.id} {message.chat.title}")
+            add_chat(message.chat.id, message.chat.title)
+            # print(f"Chat added: {message.chat.id} {message.chat.title}")
+            await log(f"Chat added: {message.chat.id} {message.chat.title}")
+        else:
+            # print(f"Chat not in approve list: {message.chat.id} {message.chat.title}")
+            await log(
+                f"Chat not in approve list: {message.chat.id} {message.chat.title}"
+            )
+        pass
+    else:
+        # Проверяем обновление названия чата
+        chat = get_chat(message.chat.id)
+        if chat.chat_name != message.chat.title:
+            chat.chat_name = message.chat.title
+            chat.save()
+            # print(f"Chat updated: {message.chat.id} {message.chat.title}")
+            await log(f"Chat updated: {message.chat.id} {message.chat.title}")
+        else:
+            # print(f"Chat already exists: {message.chat.id} {message.chat.title}")
+            await log(f"Chat already exists: {message.chat.id} {message.chat.title}")
+            pass
+
+
+async def user_check(message: types.Message):
+    # Проверка наличия id пользователя в базе данных пользователей
+    # Если пользователя нет в базе данных, то добавляем его
+    # Если пользователь есть в базе данных, то pass
+    current_user_name = ""
+    if message.from_user.last_name is None:
+        current_user_name = message.from_user.first_name
+    else:
+        current_user_name = (
+            message.from_user.first_name + " " + message.from_user.last_name
+        )
+
+    if get_user(message.from_user.id) is None:
+        add_user(
+            message.from_user.id,
+            message.from_user.first_name,
+            message.from_user.last_name,
+            message.from_user.username,
+        )
+        # print(f"User added: {message.from_user.id} {message.from_user.first_name} {message.from_user.last_name}")
+        await log(f"User added: {message.from_user.id} {current_user_name}")
+    else:
+        # print(f"User already exists: {message.from_user.id} {message.from_user.first_name} {message.from_user.last_name} {message.from_user.username}")
+        await log(
+            f"User already exists: {message.from_user.id} {current_user_name} {message.from_user.username}"
+        )
+        # Проверяем обновление имени пользователя
+        if get_user_name(message.from_user.id) != current_user_name:
+            change_user_name(message.from_user.id, current_user_name)
+            # print(f"User updated: {message.from_user.id} {message.from_user.first_name} {message.from_user.last_name}")
+            await log(f"User name updated: {message.from_user.id} {current_user_name}")
+        # Проверяем обновление username пользователя
+        if get_user_tag(message.from_user.id) != message.from_user.username:
+            change_user_tag(message.from_user.id, message.from_user.username)
+            # print(f"User updated: {message.from_user.id} {message.from_user.username}")
+            await log(
+                f"User tag updated: {message.from_user.id} {message.from_user.username}"
+            )
+        pass
+
+
+async def add_stats(message: types.Message):
+    # Добавляем пользователю и чату статистику
+    update_chat_all_stat(message.chat.id)
+    update_user_all_stat(message.from_user.id)
+
+
+async def message_processing(message: types.Message, bot: Bot):
+    await chat_check(message)
+    await user_check(message)
+    await add_stats(message)
+
+    add_message(message)
+    # Если сообщение в начале содержит слово из списка или внутри сообщения содержится слово из списка или сообщение отвечает на сообщение бота
+
+    if (message.text.split(" ")[0] in get_yandexgpt_start_words()) or (
+        any(word in message.text for word in get_yandexgpt_in_words())
+    ):
+        # print("message_processing")
+        await log("message_processing")
+        await answer_to_message(message, bot)
+
+    elif message.reply_to_message is not None:
+        if message.reply_to_message.from_user.is_bot:
+            # print("message_processing")
+            await log("message_processing")
+            await answer_to_message(message, bot)
+
+
+router = Router()
+# Если сообщение содержит текст то вызывается функция message_processing
+router.message.register(message_processing, F.text)