From d2af5b37ce53fcde6e9f3ee1b587c1ee1de19d2c Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Mon, 21 Dec 2015 01:43:57 -0800 Subject: Use attributes on exception classes to signal retriable and invalid metadata errors --- kafka/future.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'kafka/future.py') diff --git a/kafka/future.py b/kafka/future.py index 24173bb..20c31cf 100644 --- a/kafka/future.py +++ b/kafka/future.py @@ -1,4 +1,4 @@ -from kafka.common import RetriableError, IllegalStateError +from kafka.common import IllegalStateError class Future(object): @@ -16,7 +16,10 @@ class Future(object): return self.is_done and self.exception def retriable(self): - return isinstance(self.exception, RetriableError) + try: + return self.exception.retriable + except AttributeError: + return False def success(self, value): if self.is_done: -- cgit v1.2.1