90 lines
3.6 KiB
Python
90 lines
3.6 KiB
Python
from telebot.async_telebot import AsyncTeleBot
|
|
from telebot.asyncio_handler_backends import BaseMiddleware
|
|
import asyncio
|
|
|
|
import os
|
|
import sys
|
|
import importlib
|
|
from dotenv import load_dotenv
|
|
|
|
import logging
|
|
from logger import setup_logging
|
|
|
|
from database import db
|
|
|
|
from config import MODULES_DIR
|
|
|
|
load_dotenv() # Загружаем токен бота из .env
|
|
bot = AsyncTeleBot(os.getenv("BOT_TOKEN"), parse_mode="html")
|
|
|
|
logger = logging.getLogger(__name__) # Получаем логгер для текущего модуля
|
|
|
|
class UserUpdateMiddleware(BaseMiddleware):
|
|
def __init__(self, db):
|
|
super().__init__()
|
|
# message - все обычные сообщения
|
|
# chat_member - события изменения статуса участников чата
|
|
self.update_types = ['message', 'chat_member']
|
|
self.db = db
|
|
|
|
async def pre_process(self, message, data):
|
|
|
|
# Обработка пользователей, отправившие сообщение
|
|
if message.content_type == 'text':
|
|
self.db.add_or_update_user(
|
|
user_id=message.from_user.id,
|
|
nickname=message.from_user.first_name,
|
|
tag=message.from_user.username
|
|
)
|
|
|
|
# Обработка новых участников группы
|
|
elif message.content_type == 'new_chat_members':
|
|
for new_member in message.new_chat_members:
|
|
self.db.add_or_update_user(
|
|
user_id=new_member.id,
|
|
nickname=new_member.first_name,
|
|
tag=new_member.username
|
|
)
|
|
return data
|
|
|
|
async def post_process(self, message, data, exception):
|
|
pass
|
|
|
|
# Регистрируем middleware
|
|
bot.setup_middleware(UserUpdateMiddleware(db))
|
|
|
|
async def load_modules(): # Загружает все модули из директории /modules
|
|
|
|
setup_logging() # Инициализация логирования
|
|
|
|
loaded_count = 0 # Переменная для подсчёта модулей
|
|
modules_path = os.path.join(os.path.dirname(__file__), MODULES_DIR) # Импортируем относительный путь проекта
|
|
|
|
for filename in os.listdir(modules_path):
|
|
if filename.endswith(".py") and filename != "__init__.py":
|
|
module_name = filename[:-3] # Убираем расширение .py
|
|
try:
|
|
module = importlib.import_module(f"{MODULES_DIR}.{module_name}")
|
|
if hasattr(module, "register_handlers"):
|
|
module.register_handlers(bot)
|
|
loaded_count += 1
|
|
logger.info(f"Модуль {module_name} успешно загружен")
|
|
else:
|
|
logger.warning(f"Модуль {module_name} не содержит функцию register_handlers")
|
|
except Exception as e:
|
|
logger.error(f"Ошибка при загрузке модуля {module_name}: {str(e)}")
|
|
logger.info(f"Загружено модулей: {loaded_count} шт. Бот запущен.")
|
|
|
|
async def main():
|
|
os.system('clear') # Очищаем терминал
|
|
try:
|
|
await load_modules() # Проверяем и загружаем модули
|
|
await bot.infinity_polling() # Запускаем бота
|
|
except (KeyboardInterrupt, asyncio.CancelledError):
|
|
logger.info("Бот остановлен.")
|
|
except Exception as e:
|
|
logger.critical(f"Критическая ошибка: {str(e)}")
|
|
sys.exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |