Spaces:
Sleeping
Sleeping
import os | |
from logging.config import fileConfig | |
from sqlalchemy import engine_from_config | |
from sqlalchemy import pool | |
from alembic import context | |
from apps.webui.models.auths import Auth | |
from apps.webui.models.chats import Chat | |
from apps.webui.models.documents import Document | |
from apps.webui.models.memories import Memory | |
from apps.webui.models.models import Model | |
from apps.webui.models.prompts import Prompt | |
from apps.webui.models.tags import Tag, ChatIdTag | |
from apps.webui.models.tools import Tool | |
from apps.webui.models.users import User | |
from apps.webui.models.files import File | |
from apps.webui.models.functions import Function | |
from config import DATABASE_URL | |
# this is the Alembic Config object, which provides | |
# access to the values within the .ini file in use. | |
config = context.config | |
# Interpret the config file for Python logging. | |
# This line sets up loggers basically. | |
if config.config_file_name is not None: | |
fileConfig(config.config_file_name, disable_existing_loggers=False) | |
# add your model's MetaData object here | |
# for 'autogenerate' support | |
# from myapp import mymodel | |
# target_metadata = mymodel.Base.metadata | |
target_metadata = Auth.metadata | |
# other values from the config, defined by the needs of env.py, | |
# can be acquired: | |
# my_important_option = config.get_main_option("my_important_option") | |
# ... etc. | |
DB_URL = DATABASE_URL | |
if DB_URL: | |
config.set_main_option("sqlalchemy.url", DB_URL.replace("%", "%%")) | |
def run_migrations_offline() -> None: | |
"""Run migrations in 'offline' mode. | |
This configures the context with just a URL | |
and not an Engine, though an Engine is acceptable | |
here as well. By skipping the Engine creation | |
we don't even need a DBAPI to be available. | |
Calls to context.execute() here emit the given string to the | |
script output. | |
""" | |
url = config.get_main_option("sqlalchemy.url") | |
context.configure( | |
url=url, | |
target_metadata=target_metadata, | |
literal_binds=True, | |
dialect_opts={"paramstyle": "named"}, | |
) | |
with context.begin_transaction(): | |
context.run_migrations() | |
def run_migrations_online() -> None: | |
"""Run migrations in 'online' mode. | |
In this scenario we need to create an Engine | |
and associate a connection with the context. | |
""" | |
connectable = engine_from_config( | |
config.get_section(config.config_ini_section, {}), | |
prefix="sqlalchemy.", | |
poolclass=pool.NullPool, | |
) | |
with connectable.connect() as connection: | |
context.configure(connection=connection, target_metadata=target_metadata) | |
with context.begin_transaction(): | |
context.run_migrations() | |
if context.is_offline_mode(): | |
run_migrations_offline() | |
else: | |
run_migrations_online() | |