diff options
author | Dana Powers <dana.powers@rd.io> | 2016-01-10 00:34:23 -0800 |
---|---|---|
committer | Dana Powers <dana.powers@rd.io> | 2016-01-10 08:57:52 -0800 |
commit | 35ed2e75dab2ba44b57f4e7183ebc020f3121124 (patch) | |
tree | a1c56eb3b25504edcf65fd758e23555923b43671 | |
parent | b7104957f7294d3cb0e47d47ff1b6710acf5653e (diff) | |
download | kafka-python-35ed2e75dab2ba44b57f4e7183ebc020f3121124.tar.gz |
Add error checking to mark_for_reassignment
-rw-r--r-- | kafka/consumer/subscription_state.py | 3 |
1 files changed, 3 insertions, 0 deletions
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 |