From e44d9f6bd7644b4fd813c92bb1be0d4cd62f7f52 Mon Sep 17 00:00:00 2001
From: Tomas Kopecek <tkopecek@redhat.com>
Date: Apr 28 2021 11:33:27 +0000
Subject: PR#2823: Add default task ID to prep_repo_init/done


Merges #2823
https://pagure.io/koji/pull-request/2823

Fixes: #888
https://pagure.io/koji/issue/888
RFE: record task id for repos

---

diff --git a/plugins/hub/protonmsg.py b/plugins/hub/protonmsg.py
index 0f5521a..83b3b7c 100644
--- a/plugins/hub/protonmsg.py
+++ b/plugins/hub/protonmsg.py
@@ -279,6 +279,7 @@ def prep_untag(cbtype, *args, **kws):
 @convert_datetime
 @callback('postRepoInit')
 def prep_repo_init(cbtype, *args, **kws):
+    kws['task_id'] = kws.get('task_id')
     address = 'repo.init'
     props = {'type': cbtype[4:],
              'tag': kws['tag']['name'],
@@ -290,6 +291,7 @@ def prep_repo_init(cbtype, *args, **kws):
 @convert_datetime
 @callback('postRepoDone')
 def prep_repo_done(cbtype, *args, **kws):
+    kws['task_id'] = kws.get('task_id')
     address = 'repo.done'
     props = {'type': cbtype[4:],
              'tag': kws['repo']['tag_name'],