limit the contents to fetch

Author: cczhong11

inoreader/client.py

@@ -143,17 +143,24 @@ def get_subscription_list(self):
         for item in response['subscriptions']:
             yield Subscription.from_json(item)
 
-    def get_stream_contents(self, stream_id, c=''):
-        while True:
+    def get_stream_contents(self, stream_id, c='', limit=None):
+        fetched_count = 0
+        stop = False
+        while not stop:
             articles, c = self.__get_stream_contents(stream_id, c)
             for a in articles:
                 try:
                     yield Article.from_json(a)
+                    fetched_count+=1
                 except Exception as e:
                     print(e)
                     continue
+                if limit and fetched_count >= limit:
+                    stop = True
+                    break
             if c is None:
                 break
+            
 
     def __get_stream_contents(self, stream_id, continuation=''):
         self.check_token()