AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-18 22:55:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

optimize: add friendly logs

Browse Source
This commit is contained in:
Yeuoly 2024-12-26 17:39:13 +08:00
parent b34d815883
commit 2f523dd29f
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
api/services/plugin/plugin_migration.py
View File

@ -30,6 +30,11 @@ class PluginMigration:
started_at = datetime.datetime(2023, 4, 3, 8, 59, 24) started_at = datetime.datetime(2023, 4, 3, 8, 59, 24)
current_time = started_at current_time = started_at
with Session(db.engine) as session:
total_tenant_count = session.query(Tenant.id).count()
handled_tenant_count = 0
while current_time < ended_at: while current_time < ended_at:
# Initial interval of 1 day, will be dynamically adjusted based on tenant count # Initial interval of 1 day, will be dynamically adjusted based on tenant count
interval = datetime.timedelta(days=1) interval = datetime.timedelta(days=1)
@ -91,6 +96,15 @@ class PluginMigration:
plugins = cls.extract_installed_plugin_ids(tenant_id) plugins = cls.extract_installed_plugin_ids(tenant_id)
print(plugins) print(plugins)
handled_tenant_count += len(tenants)
click.echo(
click.style(
f"Processed {handled_tenant_count} tenants ({(handled_tenant_count/total_tenant_count)*100:.1f}%), {handled_tenant_count}/{total_tenant_count}",
fg="green",
)
)
current_time = batch_end current_time = batch_end
@classmethod @classmethod