mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-15 21:12:30 +08:00
29 lines
717 B
Python
29 lines
717 B
Python
"""Research Workbench database session management."""
|
|
from pathlib import Path
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy.pool import StaticPool
|
|
|
|
_COMFYUI_ROOT = Path(__file__).parent.parent
|
|
_DB_PATH = _COMFYUI_ROOT / "research_workbench.db"
|
|
|
|
DATABASE_URL = f"sqlite:///{_DB_PATH}"
|
|
|
|
engine = create_engine(
|
|
DATABASE_URL,
|
|
connect_args={"check_same_thread": False},
|
|
poolclass=StaticPool,
|
|
)
|
|
session_maker = sessionmaker(bind=engine)
|
|
|
|
|
|
def create_session():
|
|
"""Create a new database session."""
|
|
return session_maker()
|
|
|
|
|
|
def init_db():
|
|
"""Create all research tables."""
|
|
from research_api.models import Base
|
|
Base.metadata.create_all(engine)
|