22 lines
653 B
Python
22 lines
653 B
Python
"""SQLite 数据库连接与初始化"""
|
|
|
|
from pathlib import Path
|
|
|
|
from sqlmodel import SQLModel, create_engine
|
|
|
|
BASE_DIR = Path(__file__).resolve().parent.parent
|
|
DATA_DIR = BASE_DIR / "data"
|
|
DATA_DIR.mkdir(parents=True, exist_ok=True)
|
|
|
|
DB_URL = f"sqlite:///{DATA_DIR / 'vps_manager.db'}"
|
|
|
|
engine = create_engine(DB_URL, echo=False, connect_args={"check_same_thread": False})
|
|
|
|
|
|
def init_db() -> None:
|
|
"""创建所有表（幂等，可重复调用）"""
|
|
# 显式导入模型，确保其注册到 SQLModel.metadata
|
|
from app.models.asset import AIAccount, Asset, DomainDetail, VPSDetail # noqa: F401
|
|
|
|
SQLModel.metadata.create_all(engine)
|