Exception hierarchy, invalidate more md on errors

1 files changed, 25 insertions, 6 deletions

diff --git a/kafka/common.py b/kafka/common.py
index 6f0dd32..5bd9a96 100644
--- a/kafka/common.py
+++ b/kafka/common.py
@@ -69,23 +69,42 @@ class ErrorMapping(object):
 #   Exceptions  #
 #################
 
-class FailedPayloadsException(Exception):
+
+class KafkaError(RuntimeError):
+    pass
+
+
+class KafkaRequestError(KafkaError):
     pass
 
-class ConnectionError(Exception):
+
+class BrokerResponseError(KafkaError):
+    pass
+
+
+class PartitionUnavailableError(KafkaError):
     pass
 
-class BufferUnderflowError(Exception):
+
+class FailedPayloadsError(KafkaError):
+    pass
+
+
+class ConnectionError(KafkaError):
+    pass
+
+
+class BufferUnderflowError(KafkaError):
     pass
 
 
-class ChecksumError(Exception):
+class ChecksumError(KafkaError):
     pass
 
 
-class ConsumerFetchSizeTooSmall(Exception):
+class ConsumerFetchSizeTooSmall(KafkaError):
     pass
 
 
-class ConsumerNoMoreData(Exception):
+class ConsumerNoMoreData(KafkaError):
     pass