From afce25267d8423a4fc3aab10ce2edfaa19b7ce44 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Sun, 13 Mar 2016 08:48:08 -0700 Subject: [PATCH] Fix for FutureProduceResult.await on python2.6 --- kafka/producer/future.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/kafka/producer/future.py b/kafka/producer/future.py index 52c4ffc..5a7a9dc 100644 --- a/kafka/producer/future.py +++ b/kafka/producer/future.py @@ -25,7 +25,8 @@ class FutureProduceResult(Future): return ret def await(self, timeout=None): - return self._latch.wait(timeout) + # wait() on python2.6 returns None instead of the flag value + return self._latch.wait(timeout) or self._latch.is_set() class FutureRecordMetadata(Future):