Ver código fonte

replace register_tortoise in sqlite example (fixes #1901)

Falko Schindler 1 ano atrás
pai
commit
58ab41a78b
1 arquivos alterados com 12 adições e 7 exclusões
  1. 12 7
      examples/sqlite_database/main.py

+ 12 - 7
examples/sqlite_database/main.py

@@ -2,16 +2,21 @@
 from typing import List
 
 import models
-from tortoise.contrib.fastapi import register_tortoise
+from tortoise import Tortoise
 
 from nicegui import app, ui
 
-register_tortoise(
-    app,
-    db_url='sqlite://db.sqlite3',
-    modules={'models': ['models']},  # tortoise will look for models in this main module
-    generate_schemas=True,  # in production you should use version control migrations instead
-)
+
+async def init_db() -> None:
+    await Tortoise.init(db_url='sqlite://db.sqlite3', modules={'models': ['models']})
+    await Tortoise.generate_schemas()
+
+
+async def close_db() -> None:
+    await Tortoise.close_connections()
+
+app.on_startup(init_db)
+app.on_shutdown(close_db)
 
 
 @ui.refreshable