From 492fe094737c7ef5be316a988cbf477a5c874e90 Mon Sep 17 00:00:00 2001
From: Tomas Kopecek <tkopecek@redhat.com>
Date: Jan 22 2024 08:00:35 +0000
Subject: Fix koji-sweep-db for task scheduler


Related: https://pagure.io/koji/issue/3978

---

diff --git a/util/koji-sweep-db b/util/koji-sweep-db
index 11a759d..aafee25 100755
--- a/util/koji-sweep-db
+++ b/util/koji-sweep-db
@@ -111,10 +111,15 @@ 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()
+    clauses = ['task_id IN (SELECT task_id FROM temp_scratch_tasks)']
+    for table in (
+        'standard_buildroot',
+        'scheduler_task_runs',
+        'scheduler_task_refusals',
+        'scheduler_log_messages',
+        'repo'):
+        delete = DeleteProcessor(table=table, clauses=clauses)
+        delete.execute()
 
     # delete tasks finally
     delete = DeleteProcessor(table='task',