Merge pull request #602 from zackdever/KAFKA-2698

KAFKA-2698: add paused API

1 files changed, 5 insertions, 0 deletions

diff --git a/kafka/consumer/subscription_state.py b/kafka/consumer/subscription_state.py
index a4043a1..3d170ae 100644
--- a/kafka/consumer/subscription_state.py
+++ b/kafka/consumer/subscription_state.py
@@ -265,6 +265,11 @@ class SubscriptionState(object):
         """Return set of TopicPartitions in current assignment."""
         return set(self.assignment.keys())
 
+    def paused_partitions(self):
+        """Return current set of paused TopicPartitions."""
+        return set(partition for partition in self.assignment
+                   if self.is_paused(partition))
+
     def fetchable_partitions(self):
         """Return set of TopicPartitions that should be Fetched."""
         fetchable = set()