allow user to get all follows by user

tests/api/test_users.py

@@ -152,6 +152,72 @@ def test_get_follows():
     assert follow.channel.name == example_channel['name']
 
 
+@responses.activate
+def test_get_follows_with_get_all():
+    user_id = 1234
+    response = {
+        '_total': 27,
+        'follows': [example_follow]
+    }
+    responses.add(responses.GET,
+                  '{}users/{}/follows/channels'.format(BASE_URL, user_id),
+                  body=json.dumps(response),
+                  status=200,
+                  content_type='application/json')
+
+    client = TwitchClient('client id')
+
+    follows = client.users.get_follows(user_id, get_all=True)
+
+    assert len(responses.calls) == 1
+    assert len(follows) == 1
+    follow = follows[0]
+    assert isinstance(follow, Follow)
+    assert follow.notifications == example_follow['notifications']
+    assert isinstance(follow.channel, Channel)
+    assert follow.channel.id == example_channel['_id']
+    assert follow.channel.name == example_channel['name']
+
+
+@responses.activate
+def test__get_all_follows():
+    user_id = 1234
+    response_with_offset = {
+        '_total': 27,
+        '_offset': 1234,
+        'follows': [example_follow]
+    }
+    response_without_offset = {
+        '_total': 27,
+        'follows': [example_follow]
+    }
+    responses.add(responses.GET,
+                  '{}users/{}/follows/channels'.format(BASE_URL, user_id),
+                  body=json.dumps(response_with_offset),
+                  status=200,
+                  content_type='application/json')
+    responses.add(responses.GET,
+                  '{}users/{}/follows/channels'.format(BASE_URL, user_id),
+                  body=json.dumps(response_without_offset),
+                  status=200,
+                  content_type='application/json')
+
+    client = TwitchClient('client id')
+
+    follows = client.users._get_all_follows(user_id,
+                                            direction='desc',
+                                            sort_by='last_broadcast')
+
+    assert len(responses.calls) == 2
+    assert len(follows) == 2
+    follow = follows[0]
+    assert isinstance(follow, Follow)
+    assert follow.notifications == example_follow['notifications']
+    assert isinstance(follow.channel, Channel)
+    assert follow.channel.id == example_channel['_id']
+    assert follow.channel.name == example_channel['name']
+
+
 @responses.activate
 @pytest.mark.parametrize('param,value', [
     ('limit', 101),

twitch/api/users.py

@@ -1,5 +1,6 @@
 from twitch.api.base import TwitchAPI
-from twitch.constants import DIRECTIONS, DIRECTION_DESC, USERS_SORT_BY, USERS_SORT_BY_CREATED_AT
+from twitch.constants import (DIRECTIONS, DIRECTION_DESC, MAX_FOLLOWS_LIMIT,
+                              USERS_SORT_BY, USERS_SORT_BY_CREATED_AT)
 from twitch.decorators import oauth_required
 from twitch.exceptions import TwitchAttributeException
 from twitch.resources import Follow, Subscription, User, UserBlock
@@ -26,9 +27,25 @@ def check_subscribed_to_channel(self, user_id, channel_id):
         response = self._request_get('users/{}/subscriptions/{}'.format(user_id, channel_id))
         return Subscription.construct_from(response)
 
+    def _get_all_follows(self, user_id, direction, sort_by):
+        offset = 0
+        params = {
+            'limit': MAX_FOLLOWS_LIMIT,
+            'direction': direction
+        }
+        follows = []
+        while offset is not None:
+            params.update({
+                'offset': offset
+            })
+            response = self._request_get('users/{}/follows/channels'.format(user_id), params=params)
+            offset = response.get('_offset')
+            follows.extend(response['follows'])
+        return [Follow.construct_from(x) for x in follows]
+
     def get_follows(self, user_id, limit=25, offset=0, direction=DIRECTION_DESC,
-                    sort_by=USERS_SORT_BY_CREATED_AT):
-        if limit > 100:
+                    sort_by=USERS_SORT_BY_CREATED_AT, get_all=False):
+        if limit > MAX_FOLLOWS_LIMIT:
             raise TwitchAttributeException(
                 'Maximum number of objects returned in one request is 100')
         if direction not in DIRECTIONS:
@@ -37,7 +54,8 @@ def get_follows(self, user_id, limit=25, offset=0, direction=DIRECTION_DESC,
         if sort_by not in USERS_SORT_BY:
             raise TwitchAttributeException(
                 'Sort by is not valud. Valid values are {}'.format(USERS_SORT_BY))
-
+        if get_all:
+            return self._get_all_follows(user_id, direction, sort_by)
         params = {
             'limit': limit,
             'offset': offset,

twitch/constants.py

@@ -45,6 +45,7 @@
     USERS_SORT_BY_LAST_BROADCAST,
     USERS_SORT_BY_LOGIN
 ]
+MAX_FOLLOWS_LIMIT = 100
 
 DIRECTION_ASC = 'asc'
 DIRECTION_DESC = 'desc'