88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
import asyncio
|
|
import contextlib
|
|
import logging
|
|
|
|
from aiogram import Bot, Dispatcher
|
|
from aiogram.client.default import DefaultBotProperties
|
|
from aiogram.client.session.aiohttp import AiohttpSession
|
|
from remnawave import RemnawaveSDK
|
|
|
|
from bot.handlers import admin_routers, client_routers, system_routers
|
|
from bot.middlewares import DIMiddleware
|
|
from bot.scheduler import start_scheduler
|
|
from config import settings
|
|
from db.session import async_session
|
|
from misc.pally import PallyClient
|
|
from repositories.bills import BillsRepository
|
|
from repositories.subscriptions import SubscriptionRepository
|
|
from repositories.users import UserRepository
|
|
from schemas.di import DependenciesDTO
|
|
from services.billing_service import BillingService
|
|
from services.renewal_service import RenewalService
|
|
from services.user_service import UserService
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
contextlib.suppress(asyncio.CancelledError)
|
|
|
|
|
|
async def main():
|
|
dp = Dispatcher()
|
|
|
|
pally_client = PallyClient(settings.pally_token, payer_pays_commission=True)
|
|
|
|
user_repository = UserRepository()
|
|
bills_repository = BillsRepository()
|
|
subscriptions_repository = SubscriptionRepository()
|
|
|
|
user_service = UserService(user_repository, subscription_repository=subscriptions_repository)
|
|
billing_service = BillingService(
|
|
user_repository=user_repository, bills_repository=bills_repository
|
|
)
|
|
|
|
rw_sdk = RemnawaveSDK(
|
|
base_url=settings.remnawave_url,
|
|
token=settings.remnawave_token,
|
|
)
|
|
|
|
aiohttp_session = AiohttpSession(proxy=settings.proxy) if settings.proxy else None
|
|
default = DefaultBotProperties(parse_mode="HTML")
|
|
bot = Bot(token=settings.bot_token, default=default, session=aiohttp_session)
|
|
|
|
renewal_service = RenewalService(
|
|
subscription_repository=subscriptions_repository,
|
|
user_repository=user_repository,
|
|
rw_sdk=rw_sdk,
|
|
bot=bot,
|
|
)
|
|
|
|
deps = DependenciesDTO(
|
|
user_repository=user_repository,
|
|
user_service=user_service,
|
|
rw_sdk=rw_sdk,
|
|
pally_client=pally_client,
|
|
bills_repository=bills_repository,
|
|
billing_service=billing_service,
|
|
subscriptions_repository=subscriptions_repository,
|
|
renewal_service=renewal_service,
|
|
)
|
|
dp.update.middleware.register(DIMiddleware(deps, async_session))
|
|
|
|
for r in admin_routers:
|
|
dp.include_router(r)
|
|
for r in client_routers:
|
|
dp.include_router(r)
|
|
for r in system_routers:
|
|
dp.include_router(r)
|
|
|
|
scheduler_task = asyncio.create_task(start_scheduler(renewal_service, async_session))
|
|
|
|
try:
|
|
await dp.start_polling(bot)
|
|
finally:
|
|
scheduler_task.cancel()
|
|
await scheduler_task
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|