Explorar el Código

optimize: add friendly logs

Yeuoly hace 5 meses
padre
commit
2f523dd29f
Se han modificado 1 ficheros con 14 adiciones y 0 borrados
  1. 14 0
      api/services/plugin/plugin_migration.py

+ 14 - 0
api/services/plugin/plugin_migration.py

@@ -30,6 +30,11 @@ class PluginMigration:
         started_at = datetime.datetime(2023, 4, 3, 8, 59, 24)
         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:
             # Initial interval of 1 day, will be dynamically adjusted based on tenant count
             interval = datetime.timedelta(days=1)
@@ -91,6 +96,15 @@ class PluginMigration:
                 plugins = cls.extract_installed_plugin_ids(tenant_id)
                 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
 
     @classmethod