randydev commited on
Commit
c5b19bf
·
verified ·
1 Parent(s): 99d08e8

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +7 -5
main.py CHANGED
@@ -1,4 +1,5 @@
1
  import logging
 
2
  from akenoai import AkenoXToJs as js
3
  from akenoai.runner import run_fast
4
  from config import API_ID, API_HASH, BOT_TOKEN
@@ -7,8 +8,6 @@ logger = logging.getLogger(__name__)
7
  LOGS = logging.getLogger("[akenox]")
8
  logger.setLevel(logging.DEBUG)
9
 
10
- app = js.get_app()
11
-
12
  client = js.create_pyrogram(
13
  name="fastapi-bot",
14
  api_id=API_ID,
@@ -16,10 +15,13 @@ client = js.create_pyrogram(
16
  bot_token=BOT_TOKEN
17
  )
18
 
19
- @app.on_event("startup")
20
- async def startup_event():
21
  user = await client.start()
22
- LOGS.info(f"Started Bot: {user.me.first_name}")
 
 
 
23
 
24
  @app.get("/api/send")
25
  async def send_message():
 
1
  import logging
2
+ from contextlib import asynccontextmanager
3
  from akenoai import AkenoXToJs as js
4
  from akenoai.runner import run_fast
5
  from config import API_ID, API_HASH, BOT_TOKEN
 
8
  LOGS = logging.getLogger("[akenox]")
9
  logger.setLevel(logging.DEBUG)
10
 
 
 
11
  client = js.create_pyrogram(
12
  name="fastapi-bot",
13
  api_id=API_ID,
 
15
  bot_token=BOT_TOKEN
16
  )
17
 
18
+ @asynccontextmanager
19
+ async def lifespan(app: js.get_app()):
20
  user = await client.start()
21
+ LOGS.info(f"Started UserBot: {user.me.first_name}")
22
+ yield
23
+
24
+ app = js.get_app(lifespan=lifespan)
25
 
26
  @app.get("/api/send")
27
  async def send_message():