AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-13 06:19:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix-app-site-missing command (#5714)

Browse Source
This commit is contained in:
takatost 2024-06-28 20:33:53 +08:00 committed by GitHub
parent d30c13891b
commit 8e5569f773
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
api/commands.py
View File

@ -593,8 +593,8 @@ def fix_app_site_missing():
""" """
click.echo(click.style('Start fix app related site missing issue.', fg='green')) click.echo(click.style('Start fix app related site missing issue.', fg='green'))
failed_app_ids = []
while True: while True:
try:
sql = """select apps.id as id from apps left join sites on sites.app_id=apps.id sql = """select apps.id as id from apps left join sites on sites.app_id=apps.id
where sites.id is null limit 1000""" where sites.id is null limit 1000"""
with db.engine.begin() as conn: with db.engine.begin() as conn:
@ -604,6 +604,11 @@ where sites.id is null limit 1000"""
for i in rs: for i in rs:
processed_count += 1 processed_count += 1
app_id = str(i.id) app_id = str(i.id)
if app_id in failed_app_ids:
continue
try:
app = db.session.query(App).filter(App.id == app_id).first() app = db.session.query(App).filter(App.id == app_id).first()
tenant = app.tenant tenant = app.tenant
if tenant: if tenant:
@ -615,13 +620,15 @@ where sites.id is null limit 1000"""
account = accounts[0] account = accounts[0]
print("Fix app {} related site missing issue.".format(app.id)) print("Fix app {} related site missing issue.".format(app.id))
app_was_created.send(app, account=account) app_was_created.send(app, account=account)
except Exception as e:
failed_app_ids.append(app_id)
click.echo(click.style('Fix app {} related site missing issue failed!'.format(app_id), fg='red'))
logging.exception(f'Fix app related site missing issue failed, error: {e}')
continue
if not processed_count: if not processed_count:
break break
except Exception as e:
click.echo(click.style('Fix app related site missing issue failed!', fg='red'))
logging.exception(f'Fix app related site missing issue failed, error: {e}')
continue
click.echo(click.style('Congratulations! Fix app related site missing issue successful!', fg='green')) click.echo(click.style('Congratulations! Fix app related site missing issue successful!', fg='green'))