Merged with private/new-module-system

src/ocab_modules/external/create_report_apps/__init__.py

@@ -0,0 +1 @@
+from .main import module_init

src/ocab_modules/external/create_report_apps/create_report.py

@@ -0,0 +1,136 @@
+from aiogram import Bot, Router
+from aiogram.enums import ParseMode
+from aiogram.fsm.context import FSMContext
+from aiogram.fsm.state import State, StatesGroup
+from aiogram.types import (
+    BufferedInputFile,
+    KeyboardButton,
+    Message,
+    ReplyKeyboardMarkup,
+    ReplyKeyboardRemove,
+)
+
+from ocab_core.modules_system.public_api import Utils, get_fsm_context
+
+from .report import Report
+
+router = Router()
+
+
+class ReportState(StatesGroup):
+    input_system_info = State()
+    input_app_name = State()
+    input_problem_step_by_step = State()
+    input_actual_result = State()
+    input_expected_result = State()
+    input_additional_info = State()
+
+
+system_info_code = """echo "SESSION_TYPE: ${XDG_SESSION_TYPE:-Unknown}"
+[ -f /etc/os-release ] && grep "^PRETTY_NAME=" /etc/os-release | cut -d= -f2 \
+| tr -d '"' | xargs echo "OS: "
+echo "Kernel: $(uname -r)"
+echo "DE: ${XDG_CURRENT_DESKTOP:-Unknown}"
+grep "^model name" /proc/cpuinfo | head -n1 | cut -d: -f2 \
+| xargs echo "CPU: "
+lspci | grep "VGA compatible controller" | cut -d: -f3 \
+| xargs -I{} echo "GPU: {}"
+"""
+
+
+system_info_message = """Укажите параметры свой системы.
+Собрать информацию о системе можно с помощью данного скрипта:
+""" + Utils.code_format(
+    system_info_code,
+    "shell",
+)
+
+
+async def start_report(chat_id: int, bot: Bot):
+    await bot.send_message(
+        chat_id=chat_id,
+        text=system_info_message,
+        parse_mode=ParseMode.HTML,
+        reply_markup=ReplyKeyboardRemove(),
+    )
+    state = await get_fsm_context(chat_id, chat_id)
+
+    await state.set_state(ReportState.input_system_info)
+
+
+app_info_message = """Укажите название и версию приложения.
+Узнать можно с помощью данной команды:""" + Utils.code_format(
+    "rpm -qa | grep -i НАЗВАНИЕ_ПРИЛОЖЕНИЯ", "shell"
+)
+
+
+@router.message(ReportState.input_system_info)
+async def system_entered(message: Message, state: FSMContext):
+    await state.update_data(system=message.text)
+    await message.answer(
+        text=app_info_message,
+        parse_mode=ParseMode.HTML,
+    )
+    await state.set_state(ReportState.input_app_name)
+
+
+step_by_step_message = (
+    """Опиши проблему пошагово, что ты делал, что происходило, что не так."""
+)
+
+
+@router.message(ReportState.input_app_name)
+async def app_name_entered(message: Message, state: FSMContext):
+    await state.update_data(app=message.text)
+    await message.answer(text=step_by_step_message)
+    await state.set_state(ReportState.input_problem_step_by_step)
+
+
+@router.message(ReportState.input_problem_step_by_step)
+async def problem_step_by_step_entered(message: Message, state: FSMContext):
+    await state.update_data(problem_step_by_step=message.text)
+    await message.answer(text="Опиши, что произошло (фактический результат).")
+    await state.set_state(ReportState.input_actual_result)
+
+
+@router.message(ReportState.input_actual_result)
+async def actual_result_entered(message: Message, state: FSMContext):
+    await state.update_data(actual=message.text)
+    await message.answer(text="Опиши ожидаемый результат.")
+    await state.set_state(ReportState.input_expected_result)
+
+
+@router.message(ReportState.input_expected_result)
+async def expected_result_entered(message: Message, state: FSMContext):
+    await state.update_data(expected=message.text)
+    await message.answer(
+        text="Если есть дополнительная информация, то напиши ее.",
+        reply_markup=ReplyKeyboardMarkup(
+            resize_keyboard=True,
+            keyboard=[
+                [KeyboardButton(text="Дополнительной информации нет")],
+            ],
+        ),
+    )
+    await state.set_state(ReportState.input_additional_info)
+
+
+@router.message(ReportState.input_additional_info)
+async def additional_info_entered(message: Message, state: FSMContext):
+    if message.text == "Дополнительной информации нет":
+        additional_info = ""
+    else:
+        additional_info = message.text
+    await state.update_data(additional=additional_info)
+    await message.answer(
+        text="Вот твой отчет сообщением, а также файлом:",
+        reply_markup=ReplyKeyboardRemove(),
+    )
+    data = await state.get_data()
+
+    report = Report(data)
+    file_report = report.export().encode()
+
+    await message.answer(text=report.export())
+    await message.answer_document(document=BufferedInputFile(file_report, "report.txt"))
+    await state.clear()

src/ocab_modules/external/create_report_apps/info.json

@@ -0,0 +1,14 @@
+{
+    "id": "external.create_report_apps",
+    "name": "Create Report Apps",
+    "description": "Модуль для создания отчетов о ошибках в приложениях",
+    "author": [
+        "OCAB Team",
+        "Maxim Slipenko"
+    ],
+    "version": "1.0.0",
+    "privileged": false,
+    "dependencies": {
+        "standard.command_helper": "^1.0.0"
+    }
+}

src/ocab_modules/external/create_report_apps/main.py

@@ -0,0 +1,115 @@
+from typing import Union
+
+from aiogram import Bot, F, Router
+from aiogram.exceptions import TelegramForbiddenError
+from aiogram.filters import BaseFilter, Command, CommandStart
+from aiogram.types import (
+    CallbackQuery,
+    InlineKeyboardButton,
+    InlineKeyboardMarkup,
+    Message,
+)
+
+from ocab_core.modules_system.public_api import get_module, register_router
+
+from .create_report import router as create_report_router
+from .create_report import start_report
+
+register_command = get_module("standard.command_helper", "register_command")
+
+router = Router()
+
+
+class ChatTypeFilter(BaseFilter):
+    def __init__(self, chat_type: Union[str, list]):
+        self.chat_type = chat_type
+
+    async def __call__(self, message: Message) -> bool:
+        if isinstance(self.chat_type, str):
+            return message.chat.type == self.chat_type
+        return message.chat.type in self.chat_type
+
+
+@router.message(
+    ChatTypeFilter(chat_type=["group", "supergroup"]), Command("create_report_apps")
+)
+async def create_report_apps_command_group(message: Message):
+    keyboard = InlineKeyboardMarkup(
+        inline_keyboard=[
+            [
+                InlineKeyboardButton(
+                    text="Да", callback_data=f"create_report:{message.from_user.id}"
+                ),
+                InlineKeyboardButton(
+                    text="Нет", callback_data=f"cancel_report:{message.from_user.id}"
+                ),
+            ]
+        ]
+    )
+    await message.answer(
+        "Я могу отправить тебе пару вопросов "
+        "для помощи в составлении репорта личными "
+        "сообщениями.",
+        reply_markup=keyboard,
+    )
+
+
+@router.message(
+    ChatTypeFilter(chat_type=["private"]),
+    CommandStart(deep_link=True, magic=F.args == "create_report_apps"),
+)
+@router.message(ChatTypeFilter(chat_type=["private"]), Command("create_report_apps"))
+async def create_report_apps_command(message: Message, bot: Bot):
+    await start_report(message.from_user.id, bot)
+
+
+@router.callback_query(F.data.startswith("cancel_report"))
+async def cancel_report_callback(callback_query: CallbackQuery):
+    callback_user_id = int(callback_query.data.split(":")[1])
+    if callback_query.from_user.id != callback_user_id:
+        await callback_query.answer("Эта кнопка не для вас.", show_alert=True)
+        return
+
+    await callback_query.message.delete()
+
+
+@router.callback_query(F.data.startswith("create_report"))
+async def create_report_callback(callback_query: CallbackQuery, bot: Bot):
+    callback_user_id = int(callback_query.data.split(":")[1])
+    if callback_query.from_user.id != callback_user_id:
+        await callback_query.answer("Эта кнопка не для вас.", show_alert=True)
+        return
+
+    user_id = callback_query.from_user.id
+
+    async def on_chat_unavailable():
+        await callback_query.message.edit_text(
+            "Я в личных сообщениях задам тебе вопросы "
+            "для помощи в составлении репорта. "
+            'Но перед этим ты должен нажать кнопку "Запустить"'
+        )
+        info = await bot.get_me()
+        await callback_query.answer(
+            url=f"https://t.me/{info.username}?start=create_report_apps"
+        )
+
+    try:
+        chat_member = await bot.get_chat_member(chat_id=user_id, user_id=user_id)
+        if chat_member.status != "left":
+            await start_report(user_id, bot)
+            await callback_query.message.edit_text(
+                "Я в личных сообщениях задам тебе "
+                "вопросы для помощи в составлении "
+                "репорта."
+            )
+        else:
+            await on_chat_unavailable()
+    except TelegramForbiddenError:
+        await on_chat_unavailable()
+
+
+async def module_init():
+    router.include_router(create_report_router)
+
+    register_router(router)
+    register_command("create_report_apps", "Написать репорт о приложении")

src/ocab_modules/external/create_report_apps/report.py

@@ -0,0 +1,59 @@
+import aiogram
+
+
+class ReportFormatter:
+    def __init__(self, html=True):
+        self.html = html
+
+    def bold(self, string):
+        if self.html:
+            return f"<b>{self.text(string)}</b>"
+        return self.text(string)
+
+    def text(self, string):
+        if self.html:
+            return aiogram.html.quote(string)
+        return string
+
+
+class Report:
+    def __init__(self, data: dict):
+        self.data = data
+
+    def export(self):
+        data = self.data
+
+        report = f"""
+Стенд с ошибкой:
+==============================
+
+{data['system']}
+
+Пакет:
+==============================
+
+{data['app']}
+
+Шаги, приводящие к ошибке:
+==============================
+
+{data['problem_step_by_step']}
+
+Фактический результат:
+==============================
+
+{data['actual']}
+
+Ожидаемый результат:
+==============================
+
+{data['expected']}
+"""
+        if data["additional"] != "":
+            report += f"""
+Дополнительно:
+==============================
+
+{data['additional']}
+"""
+        return report

src/ocab_modules/external/yandexgpt/__init__.py

@@ -0,0 +1 @@
+from .handlers import answer_to_message

src/ocab_modules/external/yandexgpt/handlers.py

@@ -1,6 +1,4 @@
 # flake8: noqa
-import asyncio
-
 from aiogram import Bot
 from aiogram.types import Message
 
@@ -15,7 +13,7 @@ from ocab_modules.standard.database.db_api import add_message
 
 async def answer_to_message(message: Message, bot: Bot):
     # print("answer_to_message")
-    await log("answer_to_message")
+    log("answer_to_message")
     yagpt = YandexGPT(get_yandexgpt_token(), get_yandexgpt_catalog_id())
     text = message.text
     prompt = get_yandexgpt_prompt()

src/ocab_modules/external/yandexgpt/info.json

@@ -1,6 +1,9 @@
 {
-  "name": "YandexGPT",
+  "id": "external.yandexgpt",
+  "name": "Yandex GPT",
   "description": "Модуль для работы с Yandex GPT",
   "author": "OCAB Team",
-  "version": "1.0"
+  "version": "1.0.0",
+  "privileged": true,
+  "dependencies": {}
 }

src/ocab_modules/external/yandexgpt/yandexgpt.py

@@ -57,7 +57,7 @@ class YandexGPT:
                 )
             except Exception as e:  # TODO: Переделать обработку ошибок
                 # print(e)
-                await log(f"Error: {e}")
+                log(f"Error: {e}")
 
                 continue
             if int(len(response["tokens"])) < (max_tokens - answer_token):
@@ -262,7 +262,7 @@ class YandexGPT:
             )
         elif type == "yandexgpt":
             # print("yandexgpt_request")
-            await log("yandexgpt_request")
+            log("yandexgpt_request")
             messages = await self.collect_messages(message_id, chat_id)
             return await self.async_yandexgpt(
                 system_prompt=get_yandexgpt_prompt(),

src/ocab_modules/legacy/moderation/__init__.py

@@ -0,0 +1 @@
+from .moderation import ban_user, unmute_user

src/ocab_modules/legacy/moderation/moderation.py

@@ -6,8 +6,8 @@ import time
 import aiogram
 import aiohttp
 
-from ...standard.config.config import *
-from ...standard.roles.roles import *
+from ocab_modules.standard.config.config import *
+from ocab_modules.standard.roles.roles import *
 
 
 class Moderation:

src/ocab_modules/legacy/welcome/handlers.py

@@ -5,18 +5,12 @@ import random
 from threading import Thread
 
 from aiogram import Bot
-from aiogram.types import Message
 from aiogram.types import inline_keyboard_button as types
 from aiogram.utils.keyboard import InlineKeyboardBuilder
 
+from ocab_modules.legacy.moderation import ban_user, unmute_user
 from src.ocab_modules.standard.config.config import get_telegram_check_bot
 from src.ocab_modules.standard.database.db_api import *
-from src.ocab_modules.standard.moderation.moderation import (
-    ban_user,
-    mute_user,
-    unmute_user,
-)
-from src.ocab_modules.standard.roles.roles import Roles
 
 
 async def create_math_task():

src/ocab_modules/legacy/welcome/routers.py

@@ -1,6 +1,6 @@
 from aiogram import F, Router
 
-from src.ocab_modules.standard.welcome.handlers import check_new_user
+from .handlers import check_new_user
 
 router = Router()
 

src/ocab_modules/standard/admin/__init__.py

@@ -1 +1 @@
-from . import routers
+from .main import module_init

src/ocab_modules/standard/admin/handlers.py

@@ -1,10 +1,10 @@
 # flake8: noqa
-import time
-
 from aiogram import Bot
 from aiogram.types import Message
 
-from src.ocab_modules.standard.config.config import get_default_chat_tag
+from ocab_core.modules_system.public_api import get_module
+
+get_default_chat_tag = get_module("standard.config", "get_default_chat_tag")
 
 
 async def delete_message(message: Message, bot: Bot):

src/ocab_modules/standard/admin/info.json

@@ -1,6 +1,11 @@
 {
+  "id": "standard.admin",
   "name": "Admin",
   "description": "Модуль для работы с админкой",
   "author": "OCAB Team",
-  "version": "1.0"
+  "version": "1.0.0",
+  "privileged": false,
+  "dependencies": {
+    "standard.filters": "^1.0.0"
+  }
 }

src/ocab_modules/standard/admin/main.py

@@ -0,0 +1,7 @@
+from ocab_core.modules_system.public_api import register_router
+
+from .routers import router
+
+
+async def module_init():
+    register_router(router)

src/ocab_modules/standard/admin/routers.py

@@ -1,15 +1,18 @@
 # flake8: noqa
 from aiogram import F, Router
+from aiogram.filters import Command
 
-from src.ocab_modules.standard.admin.handlers import (
+from ocab_core.modules_system.public_api import get_module, log
+
+from .handlers import (
     chat_not_in_approve_list,
     delete_message,
     error_access,
     get_chat_id,
 )
-from src.ocab_modules.standard.filters.filters import (
-    ChatModerOrAdminFilter,
-    ChatNotInApproveFilter,
+
+(ChatModerOrAdminFilter, ChatNotInApproveFilter) = get_module(
+    "standard.filters", ["ChatModerOrAdminFilter", "ChatNotInApproveFilter"]
 )
 
 router = Router()
@@ -17,8 +20,8 @@ router = Router()
 # Если сообщение содержит какой либо текст и выполняется фильтр ChatNotInApproveFilter, то вызывается функция chat_not_in_approve_list
 router.message.register(chat_not_in_approve_list, ChatNotInApproveFilter(), F.text)
 
-router.message.register(get_chat_id, ChatModerOrAdminFilter(), F.text == "/chatID")
+router.message.register(get_chat_id, ChatModerOrAdminFilter(), Command("chatID"))
 
-router.message.register(delete_message, ChatModerOrAdminFilter(), F.text == "/rm")
-router.message.register(error_access, F.text == "/rm")
-router.message.register(error_access, F.text == "/chatID")
+router.message.register(delete_message, ChatModerOrAdminFilter(), Command("rm"))
+router.message.register(error_access, Command("rm"))
+router.message.register(error_access, Command("chatID"))

src/ocab_modules/standard/command_helper/__init__.py

@@ -0,0 +1 @@
+from .main import module_init, register_command

src/ocab_modules/standard/command_helper/info.json

@@ -0,0 +1,12 @@
+{
+    "id": "standard.command_helper",
+    "name": "Command helper",
+    "description": "Модуль для отображения команд при вводе '/'",
+    "author": "OCAB Team",
+    "version": "1.0.0",
+    "privileged": false,
+    "dependencies": {
+        "standard.roles": "^1.0.0",
+        "standard.database": "^1.0.0"
+    }
+}

src/ocab_modules/standard/command_helper/main.py

@@ -0,0 +1,96 @@
+from typing import Any, Awaitable, Callable, Dict
+
+from aiogram import BaseMiddleware
+from aiogram.types import BotCommand, TelegramObject
+
+from ocab_core.modules_system.public_api import (
+    get_module,
+    register_outer_message_middleware,
+    set_my_commands,
+)
+
+commands = dict()
+
+
+db_api = get_module(
+    "standard.database",
+    "db_api",
+)
+
+Roles = get_module("standard.roles", "Roles")
+
+
+def register_command(command, description, role="USER"):
+    if role not in commands:
+        commands[role] = dict()
+    commands[role][command] = {
+        "description": description,
+    }
+
+
+class OuterMiddleware(BaseMiddleware):
+    async def __call__(
+        self,
+        handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]],
+        event: TelegramObject,
+        data: Dict[str, Any],
+    ):
+
+        # if not isinstance(event, Message):
+        #     return await handler(event, data)
+        #
+        # user = db_api.get_user(event.from_user.id)
+        #
+        # if user is None:
+        #     return
+        #
+        # roles = Roles()
+        # role_name = await roles.get_role_name(role_id=user.user_role)
+        #
+        # if role_name not in commands:
+        #     return await handler(event, data)
+
+        # bot_commands = []
+
+        # for role_command in commands[role_name]:
+        #     bot_commands.append(
+        #         BotCommand(
+        #             command=role_command,
+        #             description=commands[role_name][role_command]["description"],
+        #         )
+        #     )
+
+        # await event.bot.set_my_commands(
+        #     bot_commands,
+        #     BotCommandScopeChatMember(
+        #         chat_id=event.chat.id,
+        #         user_id=event.from_user.id,
+        #     ),
+        # )
+
+        return await handler(event, data)
+
+
+async def module_init():
+    register_outer_message_middleware(OuterMiddleware())
+
+
+async def set_user_commands():
+    bot_commands = []
+    if "USER" in commands:
+        user_commands = commands["USER"]
+        for command in user_commands:
+            bot_commands.append(
+                BotCommand(
+                    command=command,
+                    description=user_commands[command]["description"],
+                )
+            )
+
+    await set_my_commands(
+        bot_commands,
+    )
+
+
+async def module_late_init():
+    await set_user_commands()

src/ocab_modules/standard/config/__init__.py

@@ -1 +1,7 @@
-from .config import config
+from .config import (
+    get_approved_chat_id,
+    get_default_chat_tag,
+    get_roles,
+    get_yandexgpt_in_words,
+    get_yandexgpt_start_words,
+)

src/ocab_modules/standard/config/config.py

@@ -27,13 +27,17 @@ def get_telegram_check_bot() -> bool:
     return config["TELEGRAM"]["CHECK_BOT"]
 
 
-def get_aproved_chat_id() -> list:
+def get_approved_chat_id() -> list:
     # Возваращем сплитованный список id чатов в формате int
     return [
         int(chat_id) for chat_id in config["TELEGRAM"]["APPROVED_CHAT_ID"].split(" | ")
     ]
 
 
+def get_roles():
+    return config["ROLES"]
+
+
 def get_user_role_name(role_number) -> dict:
     # Возвращаем название роли пользвателя по номеру роли, если такой роли нет, возвращаем неизвестно
     return config["ROLES"].get(role_number, "Неизвестно")

src/ocab_modules/standard/database/__init__.py

@@ -1,5 +1,5 @@
-from . import db_api, models
+from . import db_api, models, repositories
 
 
-def module_init():
+async def module_init():
     db_api.connect_database()

src/ocab_modules/standard/database/db_api.py

@@ -7,6 +7,7 @@ from .exceptions import MissingModuleName, NotExpectedModuleName
 from .models.chat_stats import ChatStats
 from .models.chats import Chats
 from .models.db import database_proxy
+from .models.fsm_data import FSMData
 from .models.messages import Messages
 from .models.user_stats import UserStats
 from .models.users import Users
@@ -32,7 +33,7 @@ def connect_database(is_test: bool = False, module: str | None = None):
 
 def create_tables(db: pw.SqliteDatabase):
     """Создание таблиц"""
-    for table in Chats, Messages, Users, UserStats, ChatStats:
+    for table in Chats, Messages, Users, UserStats, ChatStats, FSMData:
         if not table.table_exists():
             db.create_tables([table])
 

src/ocab_modules/standard/database/models/__init__.py

@@ -0,0 +1 @@
+from .fsm_data import FSMData

src/ocab_modules/standard/database/models/fsm_data.py

@@ -0,0 +1,12 @@
+import peewee as pw
+
+from .db import database_proxy
+
+
+class FSMData(pw.Model):
+    class Meta:
+        database = database_proxy
+
+    key = pw.CharField(primary_key=True)
+    state = pw.CharField(null=True)
+    data = pw.CharField(null=True)

src/ocab_modules/standard/database/repositories/__init__.py

@@ -0,0 +1 @@
+from .fsm_data import FSMDataRepository

src/ocab_modules/standard/database/repositories/fsm_data.py

@@ -0,0 +1,32 @@
+from peewee import fn
+
+from ..models import FSMData
+
+
+class FSMDataRepository:
+    def get(self, key: str):
+        return FSMData.get_or_none(key=key)
+
+    def set_state(self, key: str, state: str):
+        FSMData.insert(
+            key=key,
+            state=state,
+            data=fn.COALESCE(
+                FSMData.select(FSMData.data).where(FSMData.key == key), None
+            ),
+        ).on_conflict(
+            conflict_target=[FSMData.key],
+            update={FSMData.state: state},
+        ).execute()
+
+    def set_data(self, key: str, data: str):
+        FSMData.insert(
+            key=key,
+            data=data,
+            state=fn.COALESCE(
+                FSMData.select(FSMData.state).where(FSMData.key == key), None
+            ),
+        ).on_conflict(
+            conflict_target=[FSMData.key],
+            update={FSMData.data: data},
+        ).execute()

src/ocab_modules/standard/filters/__init__.py

@@ -0,0 +1 @@
+from .filters import ChatModerOrAdminFilter, ChatNotInApproveFilter

src/ocab_modules/standard/filters/filters.py

@@ -2,9 +2,10 @@ from aiogram import Bot
 from aiogram.filters import BaseFilter
 from aiogram.types import Message
 
-from ocab_core.logger import log
-from ocab_modules.standard.config.config import get_aproved_chat_id
-from ocab_modules.standard.roles.roles import Roles
+from ocab_core.modules_system.public_api import get_module, log
+
+get_approved_chat_id = get_module("standard.config", "get_approved_chat_id")
+Roles = get_module("standard.roles", "Roles")
 
 
 class ChatModerOrAdminFilter(BaseFilter):
@@ -21,14 +22,11 @@ class ChatModerOrAdminFilter(BaseFilter):
 
 class ChatNotInApproveFilter(BaseFilter):
     async def __call__(self, message: Message, bot: Bot) -> bool:
-        # print("chat_check")
-        await log("chat_check")
+        log("chat_check")
         chat_id = message.chat.id
-        if chat_id in get_aproved_chat_id():
-            # print(f"Chat in approve list: {chat_id}")
-            await log(f"Chat in approve list: {chat_id}")
+        if chat_id in get_approved_chat_id():
+            log(f"Chat in approve list: {chat_id}")
             return False
         else:
-            # print(f"Chat not in approve list: {chat_id}")
-            await log(f"Chat not in approve list: {chat_id}")
+            log(f"Chat not in approve list: {chat_id}")
             return True

src/ocab_modules/standard/filters/info.json

@@ -1,6 +1,12 @@
 {
+  "id": "standard.filters",
   "name": "Filters",
   "description": "Модуль с фильтрами",
   "author": "OCAB Team",
-  "version": "1.0"
+  "version": "1.0.0",
+  "privileged": false,
+  "dependencies": {
+    "standard.roles": "^1.0.0",
+    "standard.config": "^1.0.0"
+  }
 }

src/ocab_modules/standard/fsm_database_storage/__init__.py

@@ -0,0 +1 @@
+from .fsm import module_init

src/ocab_modules/standard/fsm_database_storage/fsm.py

@@ -0,0 +1,122 @@
+import json
+from typing import Any, Dict, Optional
+
+from aiogram.fsm.state import State
+from aiogram.fsm.storage.base import BaseStorage, StorageKey
+
+from ocab_core.modules_system.public_api import get_module, log
+from ocab_core.modules_system.public_api.public_api import set_fsm
+
+FSMDataRepository = get_module("standard.database", "repositories.FSMDataRepository")
+
+
+def serialize_key(key: StorageKey) -> str:
+    return f"{key.bot_id}:{key.chat_id}:{key.user_id}"
+
+
+def serialize_object(obj: object) -> str | None:
+    try:
+        return json.dumps(obj)
+    except Exception as e:
+        log(f"Serializing error! {e}")
+        return None
+
+
+def deserialize_object(obj):
+    try:
+        return json.loads(obj)
+    except Exception as e:
+        log(f"Deserializing error! {e}")
+        return None
+
+
+class SQLStorage(BaseStorage):
+    def __init__(self):
+        super().__init__()
+        self.repo = FSMDataRepository()
+
+    async def set_state(self, key: StorageKey, state: State | None = None) -> None:
+        """
+        Set state for specified key
+
+        :param key: storage key
+        :param state: new state
+        """
+        s_key = serialize_key(key)
+        s_state = state.state if isinstance(state, State) else state
+
+        try:
+            self.repo.set_state(s_key, s_state)
+        except Exception as e:
+            log(f"FSM Storage database error: {e}")
+
+    async def get_state(self, key: StorageKey) -> Optional[str]:
+        """
+        Get key state
+
+        :param key: storage key
+        :return: current state
+        """
+        s_key = serialize_key(key)
+
+        try:
+            s_state = self.repo.get(s_key)
+            return s_state.state if s_state else None
+        except Exception as e:
+            log(f"FSM Storage database error: {e}")
+            return None
+
+    async def set_data(self, key: StorageKey, data: Dict[str, Any]) -> None:
+        """
+        Write data (replace)
+
+        :param key: storage key
+        :param data: new data
+        """
+        s_key = serialize_key(key)
+        s_data = serialize_object(data)
+
+        try:
+            self.repo.set_data(s_key, s_data)
+        except Exception as e:
+            log(f"FSM Storage database error: {e}")
+
+    async def get_data(self, key: StorageKey) -> Optional[Dict[str, Any]]:
+        """
+        Get current data for key
+
+        :param key: storage key
+        :return: current data
+        """
+        s_key = serialize_key(key)
+
+        try:
+            s_data = self.repo.get(s_key)
+            return deserialize_object(s_data.data) if s_data else None
+        except Exception as e:
+            log(f"FSM Storage database error: {e}")
+            return None
+
+    async def update_data(
+        self, key: StorageKey, data: Dict[str, Any]
+    ) -> Dict[str, Any]:
+        """
+        Update data in the storage for key (like dict.update)
+
+        :param key: storage key
+        :param data: partial data
+        :return: new data
+        """
+        current_data = await self.get_data(key=key)
+        if not current_data:
+            current_data = {}
+        current_data.update(data)
+        await self.set_data(key=key, data=current_data)
+        return current_data.copy()
+
+    async def close(self) -> None:  # pragma: no cover
+        pass
+
+
+async def module_init():
+    set_fsm(SQLStorage())

src/ocab_modules/standard/fsm_database_storage/info.json

@@ -0,0 +1,11 @@
+{
+    "id": "standard.fsm_database_storage",
+    "name": "FSM Database Storage",
+    "description": "Очень полезный модуль",
+    "author": "OCAB Team",
+    "version": "1.0.0",
+    "privileged": false,
+    "dependencies": {
+        "standard.database": "^1.0.0"
+    }
+}

src/ocab_modules/standard/info/__init__.py

@@ -1,14 +1,2 @@
-from aiogram import F, Router
-
-from ocab_core.modules_system.public_api import register_router
-
 from .handlers import get_chat_info, get_user_info
-
-
-def module_init():
-    router = Router()
-
-    router.message.register(get_user_info, F.text.startswith("/info"))
-    router.message.register(get_chat_info, F.text.startswith("/chatinfo"))
-
-    register_router(router)
+from .main import module_init

src/ocab_modules/standard/info/handlers.py

@@ -32,7 +32,7 @@ async def get_info_answer_by_id(message: Message, bot: Bot, user_id: int):
 
     if user is None:
         await message.reply("Пользователь не найден")
-        await log(f"Пользователь не найден: {user_id}, {user}")
+        log(f"Пользователь не найден: {user_id}, {user}")
         return
 
     roles = Roles()
@@ -69,7 +69,7 @@ async def get_user_info(message: Message, bot: Bot):
             " попробуйте запросить информацию о пользователе по его тегу или ответив на его сообщение"
         )
         # print(e)
-        await log(e)
+        log(e)
 
 
 async def get_chat_info(message: Message, bot: Bot):

src/ocab_modules/standard/info/info.json

@@ -7,6 +7,7 @@
   "privileged": false,
   "dependencies": {
     "standard.roles": "^1.0.0",
-    "standard.database": "^1.0.0"
+    "standard.database": "^1.0.0",
+    "standard.command_helper": "^1.0.0"
   }
 }

src/ocab_modules/standard/info/main.py

@@ -0,0 +1,20 @@
+from aiogram import Router
+from aiogram.filters import Command
+
+from ocab_core.modules_system.public_api import get_module, register_router
+
+from .handlers import get_chat_info, get_user_info
+
+register_command = get_module("standard.command_helper", "register_command")
+
+
+async def module_init():
+    router = Router()
+
+    router.message.register(get_user_info, Command("info"))
+    router.message.register(get_chat_info, Command("chatinfo"))
+
+    register_router(router)
+
+    register_command("info", "Информация о пользователе")
+    register_command("chatinfo", "Информация о чате")

src/ocab_modules/standard/message_processing/__init__.py

@@ -0,0 +1 @@
+from .message_api import module_init

src/ocab_modules/standard/message_processing/info.json

@@ -0,0 +1,13 @@
+{
+  "id": "standard.message_processing",
+  "name": "Info",
+  "description": "Модуль с информацией",
+  "author": "OCAB Team",
+  "version": "1.0.0",
+  "privileged": false,
+  "dependencies": {
+    "standard.roles": "^1.0.0",
+    "standard.database": "^1.0.0",
+    "standard.command_helper": "^1.0.0"
+  }
+}

src/ocab_modules/standard/message_processing/message_api.py

@@ -2,14 +2,45 @@
 
 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_core.modules_system.public_api import get_module, log, register_router
+
+# from ocab_modules.standard.database.db_api import *
+
+(get_approved_chat_id, get_yandexgpt_in_words, get_yandexgpt_start_words) = get_module(
+    "standard.config",
+    ["get_approved_chat_id", "get_yandexgpt_in_words", "get_yandexgpt_start_words"],
+)
+
+answer_to_message = get_module("external.yandexgpt", "answer_to_message")
+
+(
+    get_chat,
+    add_chat,
+    get_user,
+    add_user,
+    get_user_name,
+    change_user_name,
+    get_user_tag,
+    change_user_tag,
+    update_chat_all_stat,
+    update_user_all_stat,
+    add_message,
+) = get_module(
+    "standard.database",
+    [
+        "db_api.get_chat",
+        "db_api.add_chat",
+        "db_api.get_user",
+        "db_api.add_user",
+        "db_api.get_user_name",
+        "db_api.change_user_name",
+        "db_api.get_user_tag",
+        "db_api.change_user_tag",
+        "db_api.update_chat_all_stat",
+        "db_api.update_user_all_stat",
+        "db_api.add_message",
+    ],
 )
-from ocab_modules.standard.database.db_api import *
 
 
 async def chat_check(message: types.Message):
@@ -17,17 +48,15 @@ async def chat_check(message: types.Message):
     # Если чата нет в базе данных, то проверяем его в наличии в конфиге и если он там есть то добавляем его в БД
     # Если чат есть в базе данных, то pass
     if get_chat(message.chat.id) is None:
-        if message.chat.id in get_aproved_chat_id():
+        if message.chat.id in get_approved_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}")
+            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}")
+            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}"
-            )
+            log(f"Chat not in approve list: {message.chat.id} {message.chat.title}")
         pass
     else:
         # Проверяем обновление названия чата
@@ -36,10 +65,10 @@ async def chat_check(message: types.Message):
             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}")
+            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}")
+            log(f"Chat already exists: {message.chat.id} {message.chat.title}")
             pass
 
 
@@ -62,23 +91,19 @@ async def user_check(message: types.Message):
             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}")
+        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(
+        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}")
+            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(
+            log(
                 f"User tag updated: {message.from_user.id} {message.from_user.username}"
             )
         pass
@@ -101,17 +126,19 @@ async def message_processing(message: types.Message, bot: Bot):
     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")
+        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")
+            log("message_processing")
             await answer_to_message(message, bot)
 
 
 router = Router()
 # Если сообщение содержит текст то вызывается функция message_processing
 router.message.register(message_processing, F.text)
+
+
+async def module_init():
+    register_router(router)

src/ocab_modules/standard/roles/roles.py

@@ -1,7 +1,9 @@
 from ocab_core.modules_system.public_api import get_module
 
 get_user_role = get_module("standard.database", "db_api.get_user_role")
-config: dict = get_module("standard.config", "config")
+get_roles = get_module("standard.config", "get_roles")
+
+roles = get_roles()
 
 
 class Roles:
@@ -9,13 +11,12 @@ class Roles:
     moderator = "MODERATOR"
     admin = "ADMIN"
     bot = "BOT"
-    __roles = config["ROLES"]
 
     def __init__(self):
-        self.user_role_id = self.__roles[self.user]
-        self.moderator_role_id = self.__roles[self.moderator]
-        self.admin_role_id = self.__roles[self.admin]
-        self.bot_role_id = self.__roles[self.bot]
+        self.user_role_id = roles[self.user]
+        self.moderator_role_id = roles[self.moderator]
+        self.admin_role_id = roles[self.admin]
+        self.bot_role_id = roles[self.bot]
 
     async def check_admin_permission(self, user_id):
         match get_user_role(user_id):