fix main
This commit is contained in:
parent
caebd09c80
commit
b1ee865124
@ -1,7 +1,3 @@
|
|||||||
from contextlib import asynccontextmanager
|
|
||||||
|
|
||||||
from fastapi import FastAPI
|
|
||||||
|
|
||||||
from pyrogram import Client, filters, idle
|
from pyrogram import Client, filters, idle
|
||||||
from pyrogram.handlers import MessageHandler
|
from pyrogram.handlers import MessageHandler
|
||||||
|
|
||||||
@ -10,26 +6,22 @@ from src.core.tg_service.messages_handler import message_listener
|
|||||||
from src.core.rabbitmq.connect import broker, init_queue_and_publisher
|
from src.core.rabbitmq.connect import broker, init_queue_and_publisher
|
||||||
|
|
||||||
|
|
||||||
tg_app = Client(
|
app = Client(
|
||||||
name=settings.ACCOUNT.NAME,
|
name=settings.ACCOUNT.NAME,
|
||||||
api_id=settings.ACCOUNT.API_ID,
|
api_id=settings.ACCOUNT.API_ID,
|
||||||
api_hash=settings.ACCOUNT.API_HASH,
|
api_hash=settings.ACCOUNT.API_HASH,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
app.add_handler(MessageHandler(
|
||||||
@asynccontextmanager
|
|
||||||
async def lifespan(fastapi_app: FastAPI):
|
|
||||||
await tg_app.start()
|
|
||||||
await broker.start()
|
|
||||||
await init_queue_and_publisher()
|
|
||||||
await idle()
|
|
||||||
yield
|
|
||||||
await tg_app.stop()
|
|
||||||
|
|
||||||
|
|
||||||
tg_app.add_handler(MessageHandler(
|
|
||||||
callback=message_listener,
|
callback=message_listener,
|
||||||
filters=filters.all
|
filters=filters.all
|
||||||
))
|
))
|
||||||
|
|
||||||
app = FastAPI(lifespan=lifespan)
|
async def main():
|
||||||
|
await app.start()
|
||||||
|
await broker.start()
|
||||||
|
await init_queue_and_publisher()
|
||||||
|
await idle()
|
||||||
|
await app.stop()
|
||||||
|
|
||||||
|
app.run(main())
|
||||||
|
Loading…
x
Reference in New Issue
Block a user