diff options
author | Dana Powers <dana.powers@rd.io> | 2015-12-29 16:58:38 -0800 |
---|---|---|
committer | Dana Powers <dana.powers@rd.io> | 2015-12-29 17:05:07 -0800 |
commit | aa35c4e424424687b3ff4a89a10e51d96f365279 (patch) | |
tree | 233f09cc8da385fc4d33a40b4425042065485095 /kafka/consumer | |
parent | 357073cc180e3d80fa1f7ece58fc67dbb8f42af7 (diff) | |
download | kafka-python-aa35c4e424424687b3ff4a89a10e51d96f365279.tar.gz |
Make add_assigned_partition private(ish) in SubscriptionState
Diffstat (limited to 'kafka/consumer')
-rw-r--r-- | kafka/consumer/subscription_state.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/kafka/consumer/subscription_state.py b/kafka/consumer/subscription_state.py index a90d9b3..a562093 100644 --- a/kafka/consumer/subscription_state.py +++ b/kafka/consumer/subscription_state.py @@ -143,7 +143,7 @@ class SubscriptionState(object): for partition in partitions: if partition not in self.assignment: - self.add_assigned_partition(partition) + self._add_assigned_partition(partition) for tp in set(self.assignment.keys()) - self._user_assignment: del self.assignment[tp] @@ -163,7 +163,7 @@ class SubscriptionState(object): raise ValueError("Assigned partition %s for non-subscribed topic." % tp) self.assignment.clear() for tp in assignments: - self.add_assigned_partition(tp) + self._add_assigned_partition(tp) self.needs_partition_assignment = False def unsubscribe(self): @@ -250,7 +250,7 @@ class SubscriptionState(object): def resume(self, partition): self.assignment[partition].resume() - def add_assigned_partition(self, partition): + def _add_assigned_partition(self, partition): self.assignment[partition] = TopicPartitionState() |