mirror of
https://github.com/open-webui/open-webui.git
synced 2026-07-19 23:59:21 -05:00
19 lines
538 B
Python
19 lines
538 B
Python
from __future__ import annotations
|
|
|
|
"""Alembic migration utilities."""
|
|
|
|
from alembic import op # noqa: E402 — alembic runtime context
|
|
from sqlalchemy import inspect # metadata inspection
|
|
# --- database helper functions ---
|
|
def get_existing_tables() -> set[str]:
|
|
"""Return table names already present in the database."""
|
|
conn = op.get_bind()
|
|
return set(inspect(conn).get_table_names())
|
|
|
|
|
|
def get_revision_id() -> str:
|
|
"""Generate a short random revision identifier."""
|
|
import uuid
|
|
|
|
return uuid.uuid4().hex[:12]
|