From 35ed2e75dab2ba44b57f4e7183ebc020f3121124 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Sun, 10 Jan 2016 00:34:23 -0800 Subject: Add error checking to mark_for_reassignment --- kafka/consumer/subscription_state.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/kafka/consumer/subscription_state.py b/kafka/consumer/subscription_state.py index c60f192..bb6034c 100644 --- a/kafka/consumer/subscription_state.py +++ b/kafka/consumer/subscription_state.py @@ -157,6 +157,9 @@ class SubscriptionState(object): self._group_subscription.update(topics) def mark_for_reassignment(self): + if self._user_assignment: + raise IllegalStateError(self._SUBSCRIPTION_EXCEPTION_MESSAGE) + assert self.subscription is not None, 'Subscription required' self._group_subscription.intersection_update(self.subscription) self.needs_partition_assignment = True -- cgit v1.2.1