mirror of
https://github.com/nexus-stc/hyperboria
synced 2024-12-11 20:27:50 +01:00
f8313f38b9
- feat(nexus): Update aiogrpcclient - feat(bot): Added option to disable IDM - feat(nexus): Detach postgres from Hub 2 internal commit(s) GitOrigin-RevId: bf55fbc64136be36fab362c5f48b7c2b78f0d08e
48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
from aiokit import AioRootThing
|
|
from idm.api.aioclient import IdmApiGrpcClient
|
|
from izihawa_utils.importlib import import_object
|
|
from library.telegram.base import BaseTelegramClient
|
|
from nexus.bot.promotioner import Promotioner
|
|
from nexus.bot.user_manager import UserManager
|
|
from nexus.hub.aioclient import HubGrpcClient
|
|
from nexus.meta_api.aioclient import MetaApiGrpcClient
|
|
|
|
|
|
class TelegramApplication(AioRootThing):
|
|
def __init__(self, config):
|
|
super().__init__()
|
|
self.config = config
|
|
self.telegram_client = BaseTelegramClient(
|
|
app_id=self.config['telegram']['app_id'],
|
|
app_hash=self.config['telegram']['app_hash'],
|
|
bot_token=self.config['telegram']['bot_token'],
|
|
database=self.config['telegram'].get('database'),
|
|
mtproxy=self.config['telegram'].get('mtproxy'),
|
|
)
|
|
|
|
self.hub_client = HubGrpcClient(base_url=self.config['hub']['url'])
|
|
self.starts.append(self.hub_client)
|
|
self.idm_client = None
|
|
if self.config['idm']['enabled']:
|
|
self.idm_client = IdmApiGrpcClient(base_url=self.config['idm']['url'])
|
|
self.starts.append(self.idm_client)
|
|
self.meta_api_client = MetaApiGrpcClient(base_url=self.config['meta_api']['url'])
|
|
self.starts.append(self.meta_api_client)
|
|
|
|
self.promotioner = Promotioner(promotions=self.config['promotions'])
|
|
self.user_manager = UserManager()
|
|
self._handlers = []
|
|
|
|
def set_handlers(self, telegram_client):
|
|
for handler in self.config['telegram']['handlers']:
|
|
import_object(handler)(self).register_for(telegram_client)
|
|
|
|
async def start(self):
|
|
self.set_handlers(self.telegram_client)
|
|
await self.telegram_client.start_and_wait()
|
|
await self.telegram_client.run_until_disconnected()
|
|
|
|
async def stop(self):
|
|
self.telegram_client.remove_event_handlers()
|
|
await self.telegram_client.stop()
|