From fa7bf18d0fd8fc1e51a108a1c4d4cd9657e9a63c Mon Sep 17 00:00:00 2001
From: Verum <sergeyzavalin@outlook.com>
Date: Fri, 28 Feb 2025 13:49:14 +0700
Subject: [PATCH] =?UTF-8?q?1.4.1=20=D0=A3=D0=BB=D1=83=D1=87=D1=88=D0=B5?=
 =?UTF-8?q?=D0=BD=D0=B8=D0=B5=20=D0=BA=D0=BE=D0=BC=D0=B0=D0=BD=D0=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 BotLibrary/samples/user_cmd_class.py | 54 +++++++++++++++++++++-------
 1 file changed, 42 insertions(+), 12 deletions(-)

diff --git a/BotLibrary/samples/user_cmd_class.py b/BotLibrary/samples/user_cmd_class.py
index c48f155..c8ea3b6 100644
--- a/BotLibrary/samples/user_cmd_class.py
+++ b/BotLibrary/samples/user_cmd_class.py
@@ -1,30 +1,28 @@
 # BotCode/routers/commands/user_cmd_class.py
 # Класс-шаблон для создания новых команд
 
+import inspect
 from aiogram import Router, types, F
 from aiogram.enums import ChatAction
 from aiogram.filters import Command
 from aiogram.types import InputMediaPhoto, InputMediaVideo, InputMediaDocument
+from typing import Optional, Callable
 
 from BotLibrary import Logs, valid_url, username
 from ProjectsFiles import BotVar
+from SQLite3 import base_sql
 
 # Настройки экспорта в модули
 __all__ = ("CommandHandler",)
 
-from SQLite3 import base_sql
-
-
-# Класс-шаблон для команд
 class CommandHandler:
-    def __init__(self, name: str, keywords: list, func=None, text_msg: str = None, chat_action: bool = False,
+    def __init__(self, name: str, keywords: list, func: Optional[list[Callable]] = None, text_msg=None, chat_action: bool = False,
                  description: str = "Описание команды", tg_links: bool = False,
                  keyboard=None, prefix=BotVar.prefix, callbackdata: list = None, only_admin: bool = False,
                  ignore_case: bool = True, activate_keywoards: bool = True,
                  activate_commands: bool = True, activate_callback: bool = True,
                  media: str = "message", path_to_media=None, parse_mode: str = BotVar.parse_mode,
                  disable_notification: bool = BotVar.disable_notification, protect: bool = BotVar.protect_content):
-
         self.router = Router(name=f"{name}_router")
         self.name = name
         self.log_type = name.capitalize()
@@ -67,12 +65,34 @@ class CommandHandler:
     async def handler(self, message: types.Message):
         """Основной хэндлер команды."""
         try:
-            if self.func:  # Проверяем, что функция не None
-                # Выполняем все функции из списка
-                for func in self.func:
-                    await func(message)
+            # Извлекаем текст после команды
+            command_text = message.text[len(message.text.split()[0]):].strip()  # Убираем команду из текста
+            args = command_text.split()  # Разделяем команду на аргументы
 
-            # Получаем текст сообщения
+            # Проверка на выполнение дополнительной функции (если она есть)
+            if self.func:  # Проверяем, что функция не None
+                # Выполняем все функции из списка, передавая команду
+                for func_item in self.func:
+                    # Используем inspect для получения информации о функции
+                    if callable(func_item):
+                        signature = inspect.signature(func_item)
+                        # Если функция ожидает аргументы
+                        if len(signature.parameters) > 0:
+                            await func_item(message, *args)  # Передаем аргументы функции
+                        else:
+                            await func_item(message)  # Если функция не требует аргументов, просто вызываем её
+
+            # Обрабатываем текстовое сообщение
+            if callable(self.text_msg):
+                text = self.text_msg()
+            else:
+                text = self.text_msg
+
+            # Обрабатываем tg_links
+            if self.tg_links:
+                text = text.replace("<users>", str(message.from_user.id))
+
+            # Обрабатываем текстовое сообщение
             if callable(self.text_msg):
                 text = self.text_msg()
             else:
@@ -85,6 +105,16 @@ class CommandHandler:
             Logs.info(log_type=self.log_type, user=username(message), text=f"использовал(а) команду /{self.name}")
             await base_sql(message)
 
+            # Обрабатываем текстовое сообщение
+            if callable(self.text_msg):
+                text = self.text_msg()
+            else:
+                text = self.text_msg
+
+            # Обрабатываем tg_links
+            if self.tg_links:
+                text = text.replace("<users>", str(message.from_user.id))
+
             if self.media == "message":
                 await message.reply(
                     text=text,
@@ -219,7 +249,7 @@ class CommandHandler:
                         if self.chat_action and is_last:
                             await message.bot.send_chat_action(
                                 chat_id=message.chat.id,
-                               action=ChatAction.UPLOAD_PHOTO,
+                                action=ChatAction.UPLOAD_PHOTO,
                             )
 
                     elif self.media == "gif":