diff options
Diffstat (limited to 'kafka')
| -rw-r--r-- | kafka/client.py | 6 | ||||
| -rw-r--r-- | kafka/common.py | 4 |
2 files changed, 7 insertions, 3 deletions
diff --git a/kafka/client.py b/kafka/client.py index 9cb4b48..9474091 100644 --- a/kafka/client.py +++ b/kafka/client.py @@ -8,8 +8,8 @@ import kafka.common from kafka.common import (TopicAndPartition, ConnectionError, FailedPayloadsError, - PartitionUnavailableError, - LeaderUnavailableError, KafkaUnavailableError, + PartitionUnavailableError, LeaderUnavailableError, KafkaUnavailableError, + KafkaTimeoutError, UnknownTopicOrPartitionError, NotLeaderForPartitionError) from kafka.conn import collect_hosts, KafkaConnection, DEFAULT_SOCKET_TIMEOUT_SECONDS @@ -219,7 +219,7 @@ class KafkaClient(object): self.load_metadata_for_topics(topic) while not self.has_metadata_for_topic(topic): if time.time() > start_time + timeout: - raise KafkaTimeoutError("Unable to create topic {}".format(topic)) + raise KafkaTimeoutError("Unable to create topic {0}".format(topic)) self.load_metadata_for_topics(topic) time.sleep(.5) diff --git a/kafka/common.py b/kafka/common.py index 209754b..907e128 100644 --- a/kafka/common.py +++ b/kafka/common.py @@ -135,6 +135,10 @@ class KafkaUnavailableError(KafkaError): pass +class KafkaTimeoutError(KafkaError): + pass + + class LeaderUnavailableError(KafkaError): pass |