src.app

  1import asyncio
  2import logging
  3from datetime import datetime, timedelta
  4
  5from aiogram import Bot, Dispatcher
  6from aiogram.client.default import DefaultBotProperties
  7from aiogram.enums import ParseMode
  8from aiogram.fsm.storage.redis import RedisStorage
  9from aiogram.utils.chat_action import ChatActionMiddleware
 10from aiohttp import ClientSession
 11from apscheduler.schedulers.asyncio import AsyncIOScheduler
 12
 13import handlers as hand
 14from core.config import settings
 15from core.err import exception_logging
 16from db import models, utils  # NOTE for subserver
 17from db.utils.tests import test_base, test_redis_base
 18from scheduler.balance import balance_decrement, users_notice
 19from scheduler.config_freezer import check_freeze_configs, validate_configs
 20from scheduler.dump import regular_dump
 21from scheduler.notices import send_notice
 22from wg.utils import SSH
 23
 24logger = logging.getLogger()
 25
 26
 27@exception_logging()
 28def create_bot():
 29    """Создает экземпляр бота и диспетчера.
 30
 31    Эта функция инициализирует бота с заданным токеном и
 32    создает диспетчер с использованием RedisStorage для хранения
 33    состояний.
 34
 35    Returns:
 36        tuple: Кортеж из экземпляра бота и диспетчера.
 37    """
 38    bot = Bot(
 39        token=settings.BOT_TOKEN.get_secret_value(),
 40        default=DefaultBotProperties(parse_mode=ParseMode.HTML),
 41    )
 42
 43    storage = RedisStorage.from_url(
 44        settings.CASHBASE_URL,
 45        state_ttl=timedelta(hours=settings.cash_ttl),
 46        data_ttl=timedelta(hours=settings.cash_ttl),
 47    )
 48
 49    # dp = Dispatcher(storage=MemoryStorage())  # Данные бота стираются при перезапуске
 50    dp = Dispatcher(storage=storage)
 51    dp.message.middleware(ChatActionMiddleware())
 52    dp.include_routers(
 53        hand.admin.router,
 54        hand.ban.router,
 55        hand.group.router,
 56        hand.account.router,
 57        hand.user_service.router,
 58        hand.info.router,
 59        hand.wg_service.router,
 60        hand.payment.router,
 61        hand.bug.router,
 62    )
 63
 64    return bot, dp
 65
 66
 67def create_scheduler(bot):
 68    """Создает планировщик для выполнения задач.
 69
 70    Эта функция инициализирует планировщик для выполнения
 71    различных периодических задач, таких как отправка уведомлений,
 72    уменьшение баланса и проверка конфигураций.
 73
 74    Args:
 75        bot (Bot): Экземпляр бота, используемый для отправки уведомлений.
 76
 77    Returns:
 78        AsyncIOScheduler: Экземпляр планировщика.
 79    """
 80    scheduler = AsyncIOScheduler()
 81    scheduler.add_job(
 82        send_notice,
 83        # trigger="cron",
 84        trigger="interval",
 85        seconds=3,
 86        # hour=00,
 87        # minute=00,
 88        # second=2,
 89        start_date=datetime.now() + timedelta(seconds=10),
 90        kwargs={"bot": bot},
 91    )
 92    scheduler.add_job(
 93        balance_decrement,
 94        trigger="interval",
 95        seconds=3600,
 96        start_date=datetime.now() + timedelta(seconds=20),
 97    )
 98    scheduler.add_job(
 99        users_notice,
100        trigger="interval",
101        seconds=3600,
102        start_date=datetime.now() + timedelta(seconds=30),
103        kwargs={"bot": bot},
104    )
105    scheduler.add_job(
106        check_freeze_configs,
107        trigger="interval",
108        seconds=60,
109        start_date=datetime.now() + timedelta(seconds=15),
110    )
111    scheduler.add_job(
112        validate_configs,
113        trigger="interval",
114        seconds=3600,
115        start_date=datetime.now() + timedelta(seconds=15),
116    )
117    scheduler.add_job(
118        regular_dump,
119        trigger="interval",
120        seconds=3600,
121        start_date=datetime.now() + timedelta(seconds=60),
122    )
123
124    return scheduler
125
126
127async def start_services(
128    bot: Bot, dp: Dispatcher, scheduler: AsyncIOScheduler, timeout: float = None
129):
130    """Запускает службы бота и планировщика.
131
132    Эта функция инициализирует и запускает подсистемы бота, а также
133    запускает планировщик для выполнения задач. Она управляет
134    асинхронным опросом обновлений от Telegram.
135
136    Args:
137        bot (Bot): Экземпляр бота, который будет использоваться для
138            взаимодействия с Telegram API.
139        dp (Dispatcher): Диспетчер, который обрабатывает обновления
140            и маршрутизирует их к соответствующим обработчикам.
141        scheduler (AsyncIOScheduler): Планировщик для управления
142            асинхронными задачами.
143        timeout (float, optional): Максимальное время ожидания в
144            секундах для выполнения задач. Если None, будет ожидать
145            бесконечно.
146
147    Returns:
148        set: Множество задач, которые были созданы для выполнения.
149
150    Raises:
151        RuntimeError: Если происходит ошибка при остановке опроса.
152    """
153    await test_subsystem()
154    scheduler.start()
155
156    tasks = await asyncio.wait(
157        [
158            asyncio.create_task(job)
159            for job in [
160                # bot.delete_webhook(drop_pending_updates=True),
161                dp.start_polling(
162                    bot,
163                    allowed_updates=dp.resolve_used_update_types(),
164                    started_at=datetime.now().strftime("%Y-%m-%d %H:%M"),
165                ),
166            ]
167        ],
168        timeout=timeout,
169    )
170    try:
171        await dp.stop_polling()
172    except RuntimeError:
173        pass
174    return tasks
175
176
177async def test_subsystem():
178    """Тестирует подключение к подсистемам.
179
180    Эта функция проверяет доступность баз данных и других
181    подсистем, необходимых для работы бота.
182
183    Raises:
184        Exception: Если возникает ошибка при подключении к БД.
185    """
186    bases = await test_base()
187    bases.extend(await test_redis_base())
188    bases.append(await test_subserver())
189    for base in bases:
190        try:
191            if isinstance(base, Exception):
192                raise base
193        except Exception:
194            logger.exception("Возникло исключение при подключении к БД")
195            raise
196
197
198async def test_subserver():
199    """Тестирует подключение к подсерверу.
200
201    Эта функция отправляет запрос к подсерверу для проверки его
202    доступности и корректности ответа.
203
204    Raises:
205        AssertionError: Если ответ от подсервера не соответствует ожиданиям.
206    """
207    async with ClientSession() as session:
208        url = f"{settings.subserver_url}test"
209        # url = "http://127.0.0.1:5000/test"
210        params = {"name": "test"}
211
212        async with session.get(url=url, params=params) as response:
213            result: dict = await response.json()
214            assert result.get("message") == "Hello test"
215
216
217async def start_bot():
218    """Запускает бота и его службы.
219
220    Эта функция инициализирует подключение по SSH к wireguard серверу,
221    создает бота, диспетчер и планировщик, а затем запускает все службы.
222
223    Returns:
224        set: Множество задач, которые были созданы для выполнения.
225    """
226    await SSH.connect()
227    bot, dp = create_bot()
228    scheduler = create_scheduler(bot)
229    return await start_services(bot, dp, scheduler)