mirror of
https://github.com/nexus-stc/hyperboria
synced 2024-12-02 16:02:53 +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
79 lines
3.0 KiB
Python
79 lines
3.0 KiB
Python
import asyncio
|
|
|
|
import uvloop
|
|
from library.aiogrpctools import AioGrpcServer
|
|
from library.aiopostgres import AioPostgresPoolHolder
|
|
from library.configurator import Configurator
|
|
from library.logging import configure_logging
|
|
from library.telegram.base import BaseTelegramClient
|
|
from nexus.hub.configs import get_config
|
|
from nexus.hub.services.delivery import DeliveryService
|
|
from nexus.hub.services.submitter import SubmitterService
|
|
|
|
|
|
class GrpcServer(AioGrpcServer):
|
|
def __init__(self, config: Configurator):
|
|
self.log_config(config)
|
|
super().__init__(address=config['grpc']['address'], port=config['grpc']['port'])
|
|
|
|
self.pool_holder = None
|
|
if config['database']['enabled']:
|
|
self.pool_holder = AioPostgresPoolHolder(
|
|
dsn=f'dbname={config["database"]["database"]} '
|
|
f'user={config["database"]["username"]} '
|
|
f'password={config["database"]["password"]} '
|
|
f'host={config["database"]["host"]}',
|
|
timeout=30,
|
|
pool_recycle=60,
|
|
maxsize=4,
|
|
)
|
|
self.waits.append(self.pool_holder)
|
|
|
|
self.telegram_client = BaseTelegramClient(
|
|
app_id=config['telegram']['app_id'],
|
|
app_hash=config['telegram']['app_hash'],
|
|
bot_token=config['telegram']['bot_token'],
|
|
database=config['telegram'].get('database'),
|
|
mtproxy=config['telegram'].get('mtproxy'),
|
|
)
|
|
self.starts.append(self.telegram_client)
|
|
|
|
self.delivery_service = DeliveryService(
|
|
server=self.server,
|
|
service_name=config['application']['service_name'],
|
|
bot_external_name=config['telegram']['bot_external_name'],
|
|
ipfs_config=config['ipfs'],
|
|
is_sharience_enabled=config['application']['is_sharience_enabled'],
|
|
maintenance_picture_url=config['application'].get('maintenance_picture_url', ''),
|
|
pool_holder=self.pool_holder,
|
|
pylon_config=config['pylon'],
|
|
should_store_hashes=config['application']['should_store_hashes'],
|
|
should_use_telegram_file_id=config['telegram']['should_use_telegram_file_id'],
|
|
telegram_client=self.telegram_client,
|
|
)
|
|
self.starts.append(self.delivery_service)
|
|
|
|
if config['grobid']['enabled']:
|
|
self.submitter_service = SubmitterService(
|
|
server=self.server,
|
|
service_name=config['application']['service_name'],
|
|
bot_external_name=config['telegram']['bot_external_name'],
|
|
grobid_config=config['grobid'],
|
|
ipfs_config=config['ipfs'],
|
|
meta_api_config=config['meta_api'],
|
|
telegram_client=self.telegram_client,
|
|
)
|
|
self.starts.append(self.submitter_service)
|
|
|
|
|
|
def main():
|
|
config = get_config()
|
|
configure_logging(config)
|
|
uvloop.install()
|
|
grpc_server = GrpcServer(config)
|
|
asyncio.get_event_loop().run_until_complete(grpc_server.start_and_wait())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|