opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-06-26 06:41:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

flask upgrade-db: try_create_db_if_not_exists before flask_migrate

Browse Source
This commit is contained in:
fangshiyuan.fsy@alibaba-inc.com 2026-04-02 17:24:05 +08:00
parent 3613c7bbab
commit 465db67d2b
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/commands/system.py
View File

@ -10,6 +10,7 @@ from events.app_event import app_was_created
from extensions.ext_database import db
from extensions.ext_redis import redis_client
from libs.db_migration_lock import DbMigrationAutoRenewLock
from libs.db_migration_utils import try_create_db_if_not_exists
from libs.rsa import generate_key_pair
from models import Tenant
from models.model import App, AppMode, Conversation
@ -136,6 +137,16 @@ def upgrade_db():
try:
click.echo(click.style("Starting database migration.", fg="green"))
# ensure the target database exists before migrations
try_create_db_if_not_exists(
db_type=dify_config.DB_TYPE,
host=dify_config.DB_HOST,
port=dify_config.DB_PORT,
username=dify_config.DB_USERNAME,
password=dify_config.DB_PASSWORD,
database=dify_config.DB_DATABASE,
)
# run db migration
import flask_migrate