65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
from os import getenv
|
||
|
||
from loguru import logger
|
||
|
||
from aiogram import Bot, Dispatcher
|
||
from aiogram.methods import DeleteWebhook
|
||
from aiogram.fsm.storage.redis import RedisStorage
|
||
|
||
from .settings import set_command_menu, set_profile_info
|
||
|
||
from .Handlers import setup_routers
|
||
from .Middlewares import i18n_middleware, DatabaseSessionMiddleware
|
||
|
||
from Mousey.Redis import redis
|
||
from Mousey.Database import create_db, drop_db, session_maker
|
||
|
||
|
||
@logger.catch
|
||
async def start(
|
||
to_drop_db: bool,
|
||
to_update_telegram_profile: bool,
|
||
) -> None:
|
||
"""
|
||
Запуск бота
|
||
"""
|
||
# Загрузка базы данных
|
||
if to_drop_db:
|
||
await drop_db()
|
||
logger.warning("База данных очищена.")
|
||
|
||
await create_db()
|
||
|
||
# Настройка бота
|
||
bot = Bot(token=getenv("BOT_TOKEN"))
|
||
await set_command_menu(bot=bot)
|
||
|
||
# Обновление профиля
|
||
if to_update_telegram_profile:
|
||
await set_profile_info(bot=bot)
|
||
|
||
# Настройка диспетчера
|
||
dp = Dispatcher(
|
||
storage=RedisStorage(redis=redis),
|
||
)
|
||
|
||
# Добавление мидлвари для диспетчера
|
||
dp.message.middleware(i18n_middleware)
|
||
dp.callback_query.middleware(i18n_middleware)
|
||
|
||
dp.update.middleware(
|
||
DatabaseSessionMiddleware(session_pool=session_maker)
|
||
)
|
||
|
||
logger.success("Все мидлвари успешно добавлены.")
|
||
|
||
# Добавление роутеров для диспетчера
|
||
setup_routers(dispatcher=dp)
|
||
logger.success("Все роутеры успешно добавлены.")
|
||
|
||
# Старт пуллинга бота
|
||
await bot(DeleteWebhook(drop_pending_updates=True))
|
||
logger.debug("Все Telegram запросы для бота удалены.")
|
||
|
||
logger.success("Старт пуллинга бота.")
|
||
await dp.start_polling(bot)
|