Fix tortoise example

uriyyo · Sep 26, 2023 · 636620a · 636620a
1 parent ee740e3
commit 636620a
Showing 1 changed file with 9 additions and 11 deletions.
diff --git a/examples/pagination_tortoise.py b/examples/pagination_tortoise.py
@@ -33,17 +33,7 @@ class Config:
         orm_mode = True
 
 
-@asynccontextmanager
-async def lifespan(_: Any) -> None:
-    for _ in range(100):
-        await User.create(
-            name=faker.name(),
-            email=faker.email(),
-        )
-    yield
-
-
-app = FastAPI(lifespan=lifespan)
+app = FastAPI()
 
 register_tortoise(
     app,
@@ -52,6 +42,14 @@ async def lifespan(_: Any) -> None:
     generate_schemas=True,
 )
 
+@app.on_event("startup")
+async def on_startup() -> None:
+    for _ in range(100):
+        await User.create(
+            name=faker.name(),
+            email=faker.email(),
+        )
+
 
 @app.post("/users", response_model=UserOut)
 async def create_user(user_in: UserIn) -> Any: