mirror of
https://github.com/open-webui/open-webui
synced 2024-11-16 21:42:58 +00:00
20 lines
539 B
Python
20 lines
539 B
Python
|
from alembic import command
|
||
|
|
from alembic.config import Config
|
||
|
|
|
||
|
|
from open_webui.env import OPEN_WEBUI_DIR
|
||
|
|
|
||
|
|
alembic_cfg = Config(OPEN_WEBUI_DIR / "alembic.ini")
|
||
|
|
|
||
|
|
# Set the script location dynamically
|
||
|
|
migrations_path = OPEN_WEBUI_DIR / "migrations"
|
||
|
|
alembic_cfg.set_main_option("script_location", str(migrations_path))
|
||
|
|
|
||
|
|
|
||
|
|
def revision(message: str) -> None:
|
||
|
|
command.revision(alembic_cfg, message=message, autogenerate=False)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
input_message = input("Enter the revision message: ")
|
||
|
|
revision(input_message)
|