Merge pull request #51 from UW-Macrostrat/delete-source

Update the 'sources delete' subcommand to handle ingest processes

map-integration/macrostrat/map_integration/__init__.py

@@ -46,7 +46,14 @@ def set_active_map(map: MapInfo = None):
 cli.add_command(apply_srid, name="apply-srid")
 
 cli.add_command(ingest_file, name="ingest-file")
-cli.add_command(ingest_from_csv, name="ingest-from-csv", context_settings={"allow_extra_args": True, "ignore_unknown_options": True})
+cli.add_command(
+    ingest_from_csv,
+    name="ingest-from-csv",
+    context_settings={
+        "allow_extra_args": True,
+        "ignore_unknown_options": True,
+    },
+)
 cli.add_command(ingest_object, name="ingest-object")
 cli.add_command(run_polling_loop, name="run-polling-loop")
 
@@ -104,6 +111,29 @@ def delete_sources(
                 dict(table=Identifier("sources", table)),
             )
 
+        ingest_process = db.run_query(
+            """
+            SELECT id FROM maps_metadata.ingest_process
+            JOIN maps.sources ON maps.sources.source_id = maps_metadata.ingest_process.source_id
+            WHERE maps.sources.slug = :slug
+            """,
+            dict(slug=slug),
+        ).fetchone()
+
+        if ingest_process:
+            ingest_process_id = ingest_process[0]
+
+            print("Ingest Process ID", ingest_process_id)
+
+            db.run_sql(
+                "DELETE FROM maps_metadata.ingest_process_tag WHERE ingest_process_id = :ingest_process_id",
+                dict(ingest_process_id=ingest_process_id),
+            )
+            db.run_sql(
+                "DELETE FROM maps_metadata.ingest_process WHERE id = :ingest_process_id",
+                dict(ingest_process_id=ingest_process_id),
+            )
+
         db.run_sql("DELETE FROM maps.sources WHERE slug = :slug", dict(slug=slug))