From e10a79493b19af58d4ecb6ef2de17f3b02e514be Mon Sep 17 00:00:00 2001
From: Mike McLean <mikem@redhat.com>
Date: Jan 22 2024 08:22:21 +0000
Subject: handle new task refs in clean_scratch_tasks


---

diff --git a/util/koji-sweep-db b/util/koji-sweep-db
index 11a759d..69da15a 100755
--- a/util/koji-sweep-db
+++ b/util/koji-sweep-db
@@ -111,10 +111,17 @@ def clean_scratch_tasks(cursor, vacuum, test, age):
     if not ids:
         return
 
-    # delete standard buildroots
-    delete = DeleteProcessor(table='standard_buildroot',
-                             clauses=['task_id IN (SELECT task_id FROM temp_scratch_tasks)'])
-    delete.execute()
+    # delete from referring tables
+    ref_tables = (
+        'standard_buildroot',
+        'scheduler_task_runs',
+        'scheduler_task_refusals',
+        'scheduler_log_messages',
+    )
+    for table in ref_tables:
+        delete = DeleteProcessor(table=table,
+                                 clauses=['task_id IN (SELECT task_id FROM temp_scratch_tasks)'])
+        delete.execute()
 
     # delete tasks finally
     delete = DeleteProcessor(table='task',