open-webui/backend/open_webui/migrations/util.py
2024-10-01 17:35:10 -07:00

16 lines
309 B
Python

from alembic import op
from sqlalchemy import Inspector
def get_existing_tables():
con = op.get_bind()
inspector = Inspector.from_engine(con)
tables = set(inspector.get_table_names())
return tables
def get_revision_id():
import uuid
return str(uuid.uuid4()).replace("-", "")[:12]