Третья часть: сделаны обработчики команд

This commit is contained in:
Verum
2025-02-22 11:10:16 +07:00
parent 3dcd49a3cc
commit 3bb01751c3
20 changed files with 280 additions and 28 deletions

View File

@@ -1,6 +1,11 @@
# BotCode/__init__.py
# Инициализация пакета BotCode, для создания кода проекта
# Инициализация пакета BotCode, для работы с главными частями кода
# Экспортирование модулей во внешние слои проекта
from .keyboards import *
from .routers import *
from aiogram import Router
from .routers import router as all_routers
# Объявление главного роутера
router = Router(name="main_router")
# Список подключаемых роутеров сверху-вниз
router.include_routers(all_routers)

View File

@@ -1,4 +1,4 @@
# BotCode/inline/__init__.py
# Инициализация модуля inline, для основных роутеров
# Инициализация модуля inline, для создания inline-команд
# Экспортирование модулей во внешние слои проекта

View File

@@ -1,4 +1,18 @@
# BotCode/routers/__init__.py
# Инициализация модуля routers, для основных роутеров
# Инициализация пакета routers, для работы с асинхронными обработчиками
# Экспортирование модулей во внешние слои проекта
from aiogram import Router
from .commands import router as commands_head_router
from .common import router as common_head_router
# Объявление главного роутера и настройка экспорта модулей
__all__ = ("router",)
router = Router(name="all_routers")
# Список подключаемых роутеров сверху-вниз
router.include_routers(
commands_head_router,
)
router.include_routers(common_head_router)

View File

@@ -0,0 +1,16 @@
# BotCode/routers/commands/__init__.py
# Инициализация модуля commands, для основных команд бота
from aiogram import Router
from .user_cmd import router as user_cmd_router
# Объявление роутера и настройка экспорта модулей
__all__ = ("router",)
router = Router(name="commands_head_router")
# Список подключаемых роутеров сверху-вниз
router.include_routers(
user_cmd_router,
)

View File

@@ -0,0 +1,32 @@
# BotCode/routers/commands/list_commands.py
# Создание списка команд для бота
from aiogram import Router, types
from aiogram.filters import Command
from BotLibrary import *
from .user_cmd.start_cmd import start_cmd
from .user_cmd.help_cmd import help_cmd
# Создание роутера и настройка экспорта модулей
__all__ = ("router", "set_commands")
router = Router(name="list_cmd_routers")
# Список ключевых слов для команды "setcommands"
secret_keywords = ["setcommands", "setcommand", "ыуесщььфтвы", "ыуесщььфтв",
"setcmd", "setcmds", "ыуесьв",]
# Хэндлер на команду /setcommands для использования в чате
@router.message(
#F.from_user.id.func(lambda user_id: str(user_id) in DataID.important.keys()),
Command(*secret_keywords, prefix=BotVar.prefix, ignore_case=True))
@router.message(
#F.from_user.id.func(lambda user_id: str(user_id) in DataID.important.keys()),
F.text.lower().in_(secret_keywords))
async def set_commands():
bot_commands = [
types.BotCommand(command=start_cmd.name, description=start_cmd.description),
types.BotCommand(command=help_cmd.name, description=help_cmd.description),
]
await bot.set_my_commands(bot_commands)

View File

@@ -0,0 +1,19 @@
# BotCode/routers/commands/user_cmd/__init__.py
# Инициализация модуля user_cmd, для пользовательских команд бота
# Экспортирование модулей во внешние слои проекта
from aiogram import Router
from .start_cmd import start_cmd
from .help_cmd import help_cmd
# Объявление роутера и настройка экспорта модулей
__all__ = ("router",)
router = Router(name="user_cmd_router")
# Список подключаемых роутеров сверху-вниз
router.include_routers(
help_cmd.router,
)
router.include_routers(start_cmd.router)

View File

@@ -0,0 +1,12 @@
# BotCode/routers/commands/user_cmd/help_cmd.py
# Работа с командой /help, для вывода помощи пользователю
from .user_cmd_class import CommandHandler
# Создание команды /help с нужными параметрами
help_cmd = CommandHandler(
name="help",
description="Получить помощь",
keywords=["help", "info", "помощь", "инфо", "информация", "рудз", "штащ", "byaj", "gjvjom", "byajhvfwbz"],
text_msg="Привет! Это команда помощи. Тут ты можешь узнать, как пользоваться ботом.",
)

View File

@@ -0,0 +1,13 @@
# BotCode/routers/commands/user_cmd/start_cmd.py
# # Работа с командой /start, для запуска бота
from .user_cmd_class import CommandHandler
# Создание команды /start с нужными параметрами
start_cmd = CommandHandler(
name="start",
description="Запустить бота",
keywords=["start", "старт", "запуск", "пуск", "on", "вкл", "с", "s", "ы",
"ыефке", "cnfhn", "pfgecr", "gecr", "щт", "drk", "restart", "куыефке"],
text_msg="Старт!",
)

View File

@@ -0,0 +1,51 @@
# BotCode/routers/commands/user_cmd_class.py
# Класс-шаблон для создания новых команд
from aiogram import Router, types
from aiogram.filters import Command
from BotLibrary import *
# Класс-шаблон для команд
class CommandHandler:
def __init__(self, name: str, description: str,
keywords: list, text_msg: str,
keyboard=None, prefix = BotVar.prefix,
ignore_case = True,
):
"""
Универсальный обработчик команд для бота.
:param name: Имя команды (например, "help").
:param description: Описание команды.
:param keywords: Список ключевых слов, которые активируют команду.
:param text_msg: Текст сообщения, который отправляется пользователю.
:param keyboard: Клавиатура, если требуется.
"""
self.router = Router(name=f"{name}_router")
self.name = name
self.log_type = name.capitalize()
self.description = description
self.keywords = keywords
self.text_msg = text_msg
self.keyboard = keyboard
# Привязываем хэндлер к роутеру
self.router.message(
Command(*keywords, prefix=prefix, ignore_case=ignore_case)
)(self.handler)
self.router.message(F.text.lower().in_(keywords))(self.handler)
async def handler(self, message: types.Message):
"""Основной хэндлер команды."""
user = f"@{message.from_user.username or message.from_user.id}"
try:
logger.bind(log_type=self.name.capitalize(), user=user).info(f"использовал(а) команду /{self.name}")
await message.reply(
text=self.text_msg,
reply_markup=self.keyboard() if self.keyboard else None,
)
# Проверка на ошибку
except Exception as e:
logger.bind(log_type=self.name.capitalize(), user=user).error(f"Ошибка команды: {e}")

View File

@@ -0,0 +1,13 @@
# BotCode/routers/common/__init__.py
# Инициализация пакета common, для работы со всеми сообщениями
from aiogram import Router
from .messages import router as common_messages_router
# Объявление роутера и настройка экспорта модулей
__all__ = ("router",)
router = Router(name="common_head_router")
# Идет самым последним, если другие роутеры не сработали
router.include_router(common_messages_router)

View File

@@ -0,0 +1,15 @@
# BotCode/routers/common/messages.py
# Обработчик всех сообщений
from aiogram import Router, types
from BotLibrary import *
# Настройка экспорта модулей и роутера
__all__ = ("router",)
router = Router(name="common_msg_router")
# Хэндлер на все сообщения и записывает данные
@router.message()
async def handle_all_messages(message: types.Message):
await logger_msg(message)

View File

@@ -3,6 +3,11 @@
from aiogram.types import ContentType
# Настройка экспорта из модуля
__all__ = ("types_message",)
# Функция определения типа сообщения
def types_message(message):
# Словарь для соответствия типов сообщений
content_types = {

View File

@@ -3,4 +3,5 @@
# Экспортирование модулей во внешние слои проекта
from .logs import *
from .msg_logger import *
from .start_info_out import *

View File

@@ -3,7 +3,7 @@
import sys
from loguru import logger
from ProjectsFiles.configs.config import BotLogs
from ProjectsFiles import BotLogs
# Создание обычного логгера + логгер в файл
async def setup_logger():
@@ -36,18 +36,3 @@ async def setup_logger():
diagnose=True,
level="ERROR",
filter=lambda record: record["level"].name == "ERROR")"""
# Создание функции логирования на обычные сообщения
async def common_msg_logginger(message,
name : str = "Пользователь",
message_type : str = "Медиа",
log_type : str = "Message"):
if BotLogs.permission_msg:
# Проверка на наличие текста и его типа
if message.text is None:
logger.bind(log_type=log_type, user=f"@{message.from_user.username}").info(
f"Получено сообщение из ({name}) : {message_type}")
else:
logger.bind(log_type=log_type, user=f"@{message.from_user.username}").info(
f"Получено сообщение из ({name}) : {message.text}")

View File

@@ -0,0 +1,19 @@
# BotLibrary/loggers/msg_logger.py
# Логгер для всех не обработанных сообщений
from .logs import logger
from ProjectsFiles import BotLogs
from ..analytics.type_msg import types_message
# Настройка экспорта из модуля
__all__ = ("logger_msg",)
# Создание функции логирования на обычные сообщения
async def logger_msg(message, log_type : str = "Message"):
user = f"@{message.from_user.username or message.from_user.id}"
if BotLogs.permission:
# Проверка на наличие текста и его типа
if message.text is None:
logger.bind(log_type=log_type, user=user).info(f"Получено сообщение из ({message.chat.id}) : {types_message(message)}")
else:
logger.bind(log_type=log_type, user=user).info(f"Получено сообщение из ({message.chat.id}) : {message.text}")

View File

@@ -4,3 +4,4 @@
# Экспортирование модулей во внешние слои проекта
from .bots import *
from .bot_edit import *
from .directory import *

View File

@@ -0,0 +1,26 @@
# BotLibrary/system/directory.py
# Создание пустых директорий при первом запуске
import os
from ProjectsFiles import ProjectPath, TypeDirectory
# Настройка экспорта из модуля
__all__ = ("create_directories", "setup_directories")
# Функция создания пустых директорий
async def create_directories(base_directory, subdirectories):
# Создание директорий и файлов в каждой из них
for subdirectory in subdirectories:
directory_path = os.path.join(base_directory, subdirectory)
# Проверка, существует ли директория, если нет - создаём
if not os.path.exists(directory_path):
os.makedirs(directory_path)
# Начальная установка пустых директорий
async def setup_directories():
await create_directories(ProjectPath.personal_media, TypeDirectory.media_directories)
# await create_directories(ProjectPath.received_media, TypeDirectory.media_directories)
# await create_directories(ProjectPath.bot_files, TypeDirectory.avatar_directories)
# await create_directories(ProjectPath.msg, TypeDirectory.msg_directories)

View File

@@ -3,9 +3,14 @@
import pytz
from datetime import datetime
from tzlocal import get_localzone
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from ProjectsFiles import BotVar
# Создание планировщика
scheduler = AsyncIOScheduler(timezone=get_localzone().key)
# Функция получение времени по Московскому времени
def get_moscow_time():
moscow_tz = pytz.timezone('Europe/Moscow')
@@ -17,7 +22,3 @@ def get_moscow_time():
def get_host_time():
host_time = datetime.now()
return host_time.strftime(BotVar.time_format)
# Создание планировщика
scheduler = AsyncIOScheduler(timezone=get_moscow_time())

View File

@@ -8,7 +8,7 @@ class Permissions:
logging = True # Вывод логов в консоль (True)
logging_to_file = False # Вывод логов в файл (True)
msg_logging = False # Логирование сообщений (В разработке)
msg_logging = False # Логирование сообщений в консоль (В разработке)
start_info_console = True # Вывод информации о боте в начале (True)
@@ -48,10 +48,28 @@ class BotVar:
prefix = ('$', '!', '.', '%', '&', ':', '|', '+', '-', '/', '~', '?')
# Класс для хранения типов директорий
class TypeDirectory:
# Типы сообщений и список директорий для создания
private_msg = "Личные"
group_msg = "Группы"
# Названия директорий-хранилищ
avatar = "Avatar"
photo = "Photo"
video = "Video"
videonote = "VideoNote"
gif = "GIF"
files = "Document"
voice = "Voice"
media_directories = [avatar, photo, video, videonote, gif, files, voice]
# Класс создания директорий проекта
class ProjectPath:
BotLogs = "BotLogs"
personal_media = "ProjectsFiles/media"
# Настройки логирования бота
class BotLogs:

View File

@@ -3,6 +3,8 @@
import asyncio
from BotLibrary import *
from BotCode import router as main_router
# Запуск основного кода
async def main():
@@ -11,7 +13,11 @@ async def main():
await bot_get_info()
logger.bind(log_type="AEP", user="@Console").info(f"Начало запуска бота @{BotInfo.username}...")
# Создание пустых директорий
await setup_directories()
# Подключение главного маршрутизатора
dp.include_router(main_router)
# Нужно ли удалить веб-хук
if Permissions.delete_webhook: