Fixes in AvroConsumer

t0mk · t0mk · commit 00f2a1f13011 · 2017-01-19T15:17:02.000+02:00
diff --git a/confluent_kafka/avro/__init__.py b/confluent_kafka/avro/__init__.py
@@ -53,8 +53,7 @@ def __str__(self):
 
 
 from confluent_kafka.avro.cached_schema_registry_client import CachedSchemaRegistryClient
-from confluent_kafka.avro.serializer import (SerializerError,
-                                             KeySerializerError,
+from confluent_kafka.avro.serializer import (KeySerializerError,
                                              ValueSerializerError)
 from confluent_kafka.avro.serializer.message_serializer import MessageSerializer
 
@@ -148,6 +147,8 @@ def poll(self, timeout=None):
         if timeout is None:
             timeout = -1
         message = super(AvroConsumer, self).poll(timeout)
+        if message is None:
+            return None
         if not message.value() and not message.key():
             return message
         if not message.error():
