generated from Templates/Docker_Image
feat: dashboard via NiceGUI, refs NOISSUE
This commit is contained in:
@@ -66,20 +66,6 @@ def get_session() -> Session:
|
||||
return session_factory
|
||||
|
||||
|
||||
def init_db() -> None:
|
||||
"""Initialize database tables."""
|
||||
engine = get_engine()
|
||||
Base.metadata.create_all(bind=engine)
|
||||
print("Database tables created successfully.")
|
||||
|
||||
|
||||
def drop_db() -> None:
|
||||
"""Drop all database tables (use with caution!)."""
|
||||
engine = get_engine()
|
||||
Base.metadata.drop_all(bind=engine)
|
||||
print("Database tables dropped successfully.")
|
||||
|
||||
|
||||
def get_db() -> Session:
|
||||
"""Dependency for FastAPI routes that need database access."""
|
||||
engine = get_engine()
|
||||
@@ -92,12 +78,34 @@ def get_db() -> Session:
|
||||
session.close()
|
||||
|
||||
|
||||
def get_db_sync() -> Session:
|
||||
"""Get a database session directly (for non-FastAPI/NiceGUI usage)."""
|
||||
engine = get_engine()
|
||||
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||
session = SessionLocal()
|
||||
return session
|
||||
|
||||
|
||||
def get_db_session() -> Session:
|
||||
"""Get a database session directly (for non-FastAPI usage)."""
|
||||
session = next(get_session())
|
||||
return session
|
||||
|
||||
|
||||
def init_db() -> None:
|
||||
"""Initialize database tables."""
|
||||
engine = get_engine()
|
||||
Base.metadata.create_all(bind=engine)
|
||||
print("Database tables created successfully.")
|
||||
|
||||
|
||||
def drop_db() -> None:
|
||||
"""Drop all database tables (use with caution!)."""
|
||||
engine = get_engine()
|
||||
Base.metadata.drop_all(bind=engine)
|
||||
print("Database tables dropped successfully.")
|
||||
|
||||
|
||||
def create_migration_script() -> str:
|
||||
"""Generate a migration script for database schema changes."""
|
||||
return '''-- Migration script for AI Software Factory database
|
||||
|
||||
Reference in New Issue
Block a user