Hana Celeste commited on
Commit
259a469
·
verified ·
1 Parent(s): b4f4752

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +8 -0
main.py CHANGED
@@ -1,17 +1,21 @@
1
  from fastapi import FastAPI
2
  from contextlib import asynccontextmanager
3
  from app.enka_logic import EnkaApp
 
4
  from app.fetch import Fetcher
5
 
6
  enka = EnkaApp()
 
7
  fetcher = Fetcher()
8
 
9
  @asynccontextmanager
10
  async def lifespan(app: FastAPI):
11
  await enka.start()
 
12
  await fetcher.start()
13
  yield
14
  await fetcher.stop()
 
15
  await enka.stop()
16
 
17
  app = FastAPI(lifespan=lifespan)
@@ -24,6 +28,10 @@ async def get_info(uid: str):
24
  async def get_gen(uid: str, char: str = None):
25
  return await enka.fetch_gen(uid, char)
26
 
 
 
 
 
27
  @app.get("/fetch")
28
  async def fetch_url(url: str):
29
  return await fetcher.fetch(url)
 
1
  from fastapi import FastAPI
2
  from contextlib import asynccontextmanager
3
  from app.enka_logic import EnkaApp
4
+ from app.akasa_logic import AkasaApp
5
  from app.fetch import Fetcher
6
 
7
  enka = EnkaApp()
8
+ akasa = AkasaApp()
9
  fetcher = Fetcher()
10
 
11
  @asynccontextmanager
12
  async def lifespan(app: FastAPI):
13
  await enka.start()
14
+ await akasa.start()
15
  await fetcher.start()
16
  yield
17
  await fetcher.stop()
18
+ await akasa.stop()
19
  await enka.stop()
20
 
21
  app = FastAPI(lifespan=lifespan)
 
28
  async def get_gen(uid: str, char: str = None):
29
  return await enka.fetch_gen(uid, char)
30
 
31
+ @app.get("/akasa")
32
+ async def refresh_akasa(uid: str):
33
+ return await akasa.refresh_akasha(uid)
34
+
35
  @app.get("/fetch")
36
  async def fetch_url(url: str):
37
  return await fetcher.fetch(url)