habulaj commited on
Commit
461bb57
·
verified ·
1 Parent(s): 386654d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from fastapi import FastAPI, Request
2
  from routers import news
3
  from routers import curiosity
@@ -14,19 +15,18 @@ from routers import twitter
14
  from routers import searchterm
15
  from routers import inference_createposter
16
  from routers import db
 
17
 
18
- # Instancia a aplicação FastAPI
19
- app = FastAPI()
20
-
21
- # Conecta ao banco de dados na inicialização do servidor
22
- @app.on_event("startup")
23
- async def startup():
24
  await connect_db()
25
-
26
- # Desconecta do banco de dados ao encerrar o servidor
27
- @app.on_event("shutdown")
28
- async def shutdown():
29
  await disconnect_db()
 
 
 
30
 
31
  @app.get("/")
32
  def greet_json():
 
1
+ from contextlib import asynccontextmanager
2
  from fastapi import FastAPI, Request
3
  from routers import news
4
  from routers import curiosity
 
15
  from routers import searchterm
16
  from routers import inference_createposter
17
  from routers import db
18
+ from routers.db import connect_db, disconnect_db
19
 
20
+ @asynccontextmanager
21
+ async def lifespan(app: FastAPI):
22
+ # Startup
 
 
 
23
  await connect_db()
24
+ yield
25
+ # Shutdown
 
 
26
  await disconnect_db()
27
+
28
+ # Instancia a aplicação FastAPI
29
+ app = FastAPI(lifespan=lifespan)
30
 
31
  @app.get("/")
32
  def greet_json():