From 2e108cf952582509684281ea0aa9c7fad9173daf Mon Sep 17 00:00:00 2001
From: Tomas Kopecek <tkopecek@redhat.com>
Date: Jun 28 2023 04:41:14 +0000
Subject: use Task interface for correct state hooks


---

diff --git a/kojihub/scheduler.py b/kojihub/scheduler.py
index 08a572b..89d89e0 100644
--- a/kojihub/scheduler.py
+++ b/kojihub/scheduler.py
@@ -583,13 +583,8 @@ class TaskScheduler(object):
         insert.execute()
 
         # mark the task assigned
-        update = UpdateProcessor(
-            'task',
-            data={'host_id': host['id'], 'state': koji.TASK_STATES['ASSIGNED']},
-            clauses=['id=%(task_id)s', 'state=%(free)s'],
-            values={'task_id': task['task_id'], 'free': koji.TASK_STATES['FREE']},
-        )
-        update.execute()
+        task = kojihub.Task(task['task_id'])
+        task.assign(host['id'])
 
 
 class SchedulerExports: