don't crash if consumer_group is missing

Parsely · Apr 24, 2015 · edc5e8b · edc5e8b
1 parent 334b1d9
commit edc5e8b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pykafka/simpleconsumer.py b/pykafka/simpleconsumer.py
@@ -10,8 +10,8 @@
 import base
 from .common import OffsetType
 from .exceptions import (OffsetOutOfRangeError, UnknownTopicOrPartition,
-                                OffsetMetadataTooLarge, OffsetsLoadInProgress,
-                                NotCoordinatorForConsumer)
+                         OffsetMetadataTooLarge, OffsetsLoadInProgress,
+                         NotCoordinatorForConsumer)
 
 from .utils.error_handlers import handle_partition_responses, raise_error
 from .protocol import (PartitionFetchRequest, PartitionOffsetCommitRequest,
@@ -158,7 +158,8 @@ def _handle_NotCoordinatorForConsumer(parts):
         }
 
     def _discover_offset_manager(self):
-        self._offset_manager = self._cluster.get_offset_manager(self._consumer_group)
+        if self._consumer_group is not None:
+            self._offset_manager = self._cluster.get_offset_manager(self._consumer_group)
 
     @property
     def topic(self):