Refactor to use async database model
This commit is contained in:
@ -13,6 +13,7 @@ import uuid
|
||||
from datetime import datetime
|
||||
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy.ext.asyncio import AsyncEngine
|
||||
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
|
||||
|
||||
|
||||
@ -186,3 +187,9 @@ class AuditLog(Base):
|
||||
def init_db(engine: sa.Engine) -> None:
|
||||
"""Initialize database."""
|
||||
Base.metadata.create_all(engine)
|
||||
|
||||
|
||||
async def init_db_async(engine: AsyncEngine) -> None:
|
||||
"""Initialize database."""
|
||||
async with engine.begin() as conn:
|
||||
await conn.run_sync(Base.metadata.create_all)
|
||||
|
||||
Reference in New Issue
Block a user