📝 Update Peewee docs, simplify with double dependency with yield (#947)

docs/src/sql_databases_peewee/sql_app/main.py

@@ -2,7 +2,6 @@ import time
 from typing import List
 
 from fastapi import Depends, FastAPI, HTTPException
-from starlette.requests import Request
 
 from . import crud, database, models, schemas
 from .database import db_state_default
@@ -16,7 +15,12 @@ app = FastAPI()
 sleep_time = 10
 
 
-def get_db():
+async def reset_db_state():
+    database.db._state._state.set(db_state_default.copy())
+    database.db._state.reset()
+
+
+def get_db(db_state=Depends(reset_db_state)):
     try:
         database.db.connect()
         yield
@@ -25,14 +29,6 @@ def get_db():
             database.db.close()
 
 
-@app.middleware("http")
-async def reset_db_middleware(request: Request, call_next):
-    database.db._state._state.set(db_state_default.copy())
-    database.db._state.reset()
-    response = await call_next(request)
-    return response
-
-
 @app.post("/users/", response_model=schemas.User, dependencies=[Depends(get_db)])
 def create_user(user: schemas.UserCreate):
     db_user = crud.get_user_by_email(email=user.email)