add cli command to remove finished export manually

HubGrep · Feb 20, 2022 · e821b00 · e821b00
1 parent 9a9f61b
commit e821b00
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 4 deletions.
diff --git a/hubgrep_indexer/cli_blueprint/repos.py b/hubgrep_indexer/cli_blueprint/repos.py
@@ -58,3 +58,19 @@ def prune_exports(keep, hosting_service=None):
         for export in old_exports_unified:
             print(f"deleting export {export}")
             export.delete_file()
+
+
+@cli_bp.cli.command(help="drop a hosting services 'finished' table")
+@click.argument("hosting_service")
+def drop_finished_run_table(hosting_service):
+    hosting_service_api_url = hosting_service
+    hosting_service: HostingService = HostingService.query.filter_by(
+        api_url=hosting_service_api_url
+    ).first()
+    if not hosting_service:
+        print('could not find hosting service!')
+        exit(1)
+
+    print(f"found {hosting_service}, dropping finished table")
+    hosting_service.drop_finished_run_table()
+
diff --git a/hubgrep_indexer/models/hosting_service.py b/hubgrep_indexer/models/hosting_service.py
@@ -164,12 +164,14 @@ def handle_finished_run(self):
         self.export_repos()
         logger.debug(f"export for {self} finished")
         if not current_app.config['KEEP_LAST_RUN_IN_DB']:
-            logger.debug(f"dropping table for exported {self}")
+            self.drop_finished_run_table()
 
-            target_table = Repository.get_finished_table_name(self)
-            with TableHelper._cursor() as cur:
-                TableHelper.drop_table(cur, target_table)
+    def drop_finished_run_table(self):
+        logger.debug(f"dropping table for exported {self}")
 
+        target_table = Repository.get_finished_table_name(self)
+        with TableHelper._cursor() as cur:
+            TableHelper.drop_table(cur, target_table)
 
     @property
     def repos(self) -> ResultProxy: