于小丘
209cb24ab4
- Implement PhysicalFile model to manage physical file references and reference counting. - Create Policy model with associated options and group links for storage policies. - Introduce Redeem and Report models for handling redeem codes and reports. - Add Settings model for site configuration and user settings management. - Develop Share model for sharing objects with unique codes and associated metadata. - Implement SourceLink model for managing download links associated with objects. - Create StoragePack model for managing user storage packages. - Add Tag model for user-defined tags with manual and automatic types. - Implement Task model for managing background tasks with status tracking. - Develop User model with comprehensive user management features including authentication. - Introduce UserAuthn model for managing WebAuthn credentials. - Create WebDAV model for managing WebDAV accounts associated with users.
33 lines
973 B
Python
33 lines
973 B
Python
from sqlmodel import SQLModel
|
|
from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
|
|
from sqlmodel.ext.asyncio.session import AsyncSession
|
|
from utils.conf import appmeta
|
|
from sqlalchemy.orm import sessionmaker
|
|
from typing import AsyncGenerator
|
|
|
|
ASYNC_DATABASE_URL = appmeta.database_url
|
|
|
|
engine: AsyncEngine = create_async_engine(
|
|
ASYNC_DATABASE_URL,
|
|
echo=appmeta.debug,
|
|
connect_args={
|
|
"check_same_thread": False
|
|
} if ASYNC_DATABASE_URL.startswith("sqlite") else {},
|
|
future=True,
|
|
# pool_size=POOL_SIZE,
|
|
# max_overflow=64,
|
|
)
|
|
|
|
_async_session_factory = sessionmaker(engine, class_=AsyncSession)
|
|
|
|
async def get_session() -> AsyncGenerator[AsyncSession, None]:
|
|
async with _async_session_factory() as session:
|
|
yield session
|
|
|
|
async def init_db(
|
|
url: str = ASYNC_DATABASE_URL
|
|
):
|
|
"""创建数据库结构"""
|
|
async with engine.begin() as conn:
|
|
await conn.run_sync(SQLModel.metadata.create_all)
|
|
|