From b0b6ce7dfa8e21d06a0b7f179fe2983e61cd8944 Mon Sep 17 00:00:00 2001
From: Chenxiong Qi <cqi@redhat.com>
Date: May 24 2016 09:20:31 +0000
Subject: handle exception from getTaskInfo correctly


koji.GenericError is raised from getTaskInfo instead of returning None,
if a task id does not exist in Koji.

Fix #52

Signed-off-by: Chenxiong Qi <cqi@redhat.com>

---

diff --git a/src/pyrpkg/cli.py b/src/pyrpkg/cli.py
index 37578d2..3be4db2 100755
--- a/src/pyrpkg/cli.py
+++ b/src/pyrpkg/cli.py
@@ -1510,8 +1510,9 @@ class TaskWatcher(object):
             # Already done, nothing else to report
             return False
         last = self.info
-        self.info = self.session.getTaskInfo(self.id, request=True)
-        if self.info is None:
+        try:
+            self.info = self.session.getTaskInfo(self.id, request=True)
+        except koji.GenericError:
             raise Exception("No such task id: %i" % self.id)
         state = self.info['state']
         if last: