Fix: timeout

app/crud.py

@@ -35,7 +35,7 @@ def create_announcement(db: Session, announcement: schemas.AnnouncementCreate):
         "to": tokens,
         "title": announcement.title,
         "body": announcement.description
-    })
+    }, timeout=10)
     return db_announcement
 
 

app/database.py

@@ -12,3 +12,11 @@
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
 Base = declarative_base()
+
+
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()

app/main.py

@@ -5,20 +5,12 @@
 import models
 import schemas
 from auth import get_user
-from database import engine, SessionLocal
+from database import engine, get_db
 
 models.Base.metadata.create_all(bind=engine)
 app = FastAPI()
 
 
-def get_db():
-    db = SessionLocal()
-    try:
-        yield db
-    finally:
-        db.close()
-
-
 @app.post("/private/announcements", response_model=schemas.Announcement)
 def create_announcement(announcement: schemas.AnnouncementCreate, db: Session = Depends(get_db)):
     return crud.create_announcement(db=db, announcement=announcement)