pgq.Consumer: Provide self.batch_info with batch details

diff --git a/python/pgq/consumer.py b/python/pgq/consumer.py
index 1f6615d40937fc2e4092094f3de5b291d6ed1c4b..20b83fe6922cba6f3bd3ee24666b193ff6c00a23 100644 (file)
--- a/python/pgq/consumer.py
+++ b/python/pgq/consumer.py
@@ -144,6 +144,8 @@ class Consumer(skytools.DBScript):
     pgq_min_interval = None
     pgq_min_lag = None
 
+    batch_info = None
+
     def __init__(self, service_name, db_name, args):
         """Initialize new consumer.
         
@@ -298,8 +300,8 @@ class Consumer(skytools.DBScript):
         q = "select * from pgq.next_batch_custom(%s, %s, %s, %s, %s)"
         curs.execute(q, [self.queue_name, self.consumer_name,
                          self.pgq_min_lag, self.pgq_min_count, self.pgq_min_interval])
-        inf = curs.fetchone()
-        return inf['batch_id']
+        self.batch_info = curs.fetchone()
+        return self.batch_info['batch_id']
 
     def _flush_retry(self, curs, batch_id, list):
         """Tag retry events."""