完善 aiosqlite 异步数据库驱动

Author: lan-air

.gitignore

@@ -8,7 +8,7 @@ __pycache__/
 # Created by .ignore support plugin (hsz.mobi)
 ### Python template
 # Byte-compiled / optimized / DLL files
-database.db
+databasea.db
 # C extensions
 *.so
 *.env

database.py

@@ -1,3 +1,4 @@
+import asyncio
 import datetime
 
 from sqlalchemy import Boolean, Column, Integer, String, DateTime
@@ -9,7 +10,10 @@
 
 Base = declarative_base()
 
-Base.metadata.create_all(bind=engine)
+
+async def init_models():
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)
 
 
 async def get_session():

main.py

@@ -11,13 +11,19 @@
 from sqlalchemy import or_, select, update, delete
 from sqlalchemy.ext.asyncio.session import AsyncSession
 
-from database import get_session, Codes
+from database import get_session, Codes, init_models
 
 app = FastAPI()
 if not os.path.exists('./static'):
     os.makedirs('./static')
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
+
+@app.on_event('startup')
+async def startup():
+    await init_models()
+
+
 ############################################
 # 需要修改的参数
 # 允许错误次数