reimplemented fetch_unread with fetch_articles

Author: Linusp

inoreader/client.py

@@ -163,7 +163,7 @@ def __get_stream_contents(self, stream_id, continuation=''):
         else:
             return response['items'], None
 
-    def fetch_unread(self, folder=None, tags=None):
+    def fetch_articles(self, folder=None, tags=None, unread=True, starred=False):
         self.check_token()
 
         url = urljoin(BASE_URL, self.STREAM_CONTENTS_PATH)
@@ -172,7 +172,13 @@ def fetch_unread(self, folder=None, tags=None):
                 url,
                 quote_plus(self.GENERAL_TAG_TEMPLATE.format(folder))
             )
-        params = {'xt': self.READ_TAG, 'c': str(uuid4())}
+
+        params = {'c': str(uuid4())}
+        if unread:
+            params['xt'] = self.READ_TAG
+
+        if starred:
+            params['it'] = self.STARRED_TAG
 
         response = self.parse_response(self.session.post(url, params=params, proxies=self.proxies))
         for data in response['items']:
@@ -182,6 +188,7 @@ def fetch_unread(self, folder=None, tags=None):
             ])
             if tags and not categories.issuperset(set(tags)):
                 continue
+
             yield Article.from_json(data)
 
         continuation = response.get('continuation')
@@ -198,8 +205,13 @@ def fetch_unread(self, folder=None, tags=None):
                 if tags and not categories.issuperset(set(tags)):
                     continue
                 yield Article.from_json(data)
+
             continuation = response.get('continuation')
 
+    def fetch_unread(self, folder=None, tags=None):
+        for article in self.fetch_articles(folder=folder, tags=tags, unread=True):
+            yield article
+
     def add_general_label(self, articles, label):
         self.check_token()