From 80d1aae8f0c7373b0bb7cb208954dca19d77b697 Mon Sep 17 00:00:00 2001
From: Jana Cupova <root@localhost>
Date: Jul 08 2022 09:16:57 +0000
Subject: Fix download-task with wait option


---

diff --git a/cli/koji_cli/commands.py b/cli/koji_cli/commands.py
index b59c37d..15c7788 100644
--- a/cli/koji_cli/commands.py
+++ b/cli/koji_cli/commands.py
@@ -6926,6 +6926,7 @@ def anon_handle_download_task(options, session, args):
             koji.TASK_STATES['FAILED']):
         watch_tasks(session, [base_task_id], quiet=suboptions.quiet,
                     poll_interval=options.poll_interval, topurl=options.topurl)
+        base_task = session.getTaskInfo(base_task_id)
 
     list_tasks = [base_task]
     if not suboptions.parentonly:
@@ -6999,18 +7000,6 @@ def anon_handle_download_task(options, session, args):
         print("No files for download found.")
         return
 
-    required_tasks = {}
-    for (task, nop, nop, nop, nop) in downloads:
-        if task["id"] not in required_tasks:
-            required_tasks[task["id"]] = task
-
-    for task_id in required_tasks:
-        if required_tasks[task_id]["state"] != koji.TASK_STATES.get("CLOSED"):
-            if task_id == base_task_id:
-                error("Task %d has not finished yet." % task_id)
-            else:
-                error("Child task %d has not finished yet." % task_id)
-
     downloads_new_names = [(new_filename, vol) for (_, _, vol, new_filename, _) in downloads]
     if not suboptions.dirpertask:
         not_uniques = list({x for x in downloads_new_names if downloads_new_names.count(x) > 1})