[bilibili] bilibili User Articles FavList support

gallery_dl/extractor/bilibili.py

@@ -81,6 +81,31 @@ def items(self):
             yield Message.Url, url, text.nameext_from_url(url, article)
 
 
+class BilibiliUserArticleFavlistExtractor(BilibiliExtractor):
+    subcategory = "user-article-favlist"
+    pattern = (r"(?:https?://)?(?:space.bilibili.com)"
+               r"/(\d+)/(?:favlist\?fid=opus)"
+               r"(?:&ftype=opus)?")
+    example = "https://space.bilibili.com/123456/favlist?fid=opus"
+    _warning = True
+
+    def _init(self):
+        BilibiliExtractor._init(self)
+        if self._warning:
+            if not self.cookies_check(("SESSDATA",)):
+                self.log.warning("no 'SESSDATA' cookie set")
+            BilibiliUserArticleFavlistExtractor._warning = False
+
+    def items(self):
+        mid = self.api.login_user_id()
+        if str(mid) != self.groups[0]:
+            raise exception.StopExtraction("This is not you favlist url!")
+        for article in self.api.user_favlist():
+            article["_extractor"] = BilibiliArticleExtractor
+            url = "{}/opus/{}".format(self.root, article["opus_id"])
+            yield Message.Queue, url, article
+
+
 class BilibiliAPI():
     def __init__(self, extractor):
         self.extractor = extractor
@@ -122,3 +147,30 @@ def article(self, article_id):
                     raise exception.StopExtraction(
                         "%s: Unable to extract INITIAL_STATE data", article_id)
             self.extractor.wait(seconds=300)
+
+    def user_favlist(self):
+        endpoint = "/opus/feed/fav"
+        params = {"page": 1, "page_size": 20}
+
+        while True:
+            data = self._call(endpoint, params)
+
+            for item in data["data"]["items"]:
+                yield item
+
+            if not data["data"]["has_more"]:
+                break
+
+            params["page"] = params["page"] + 1
+
+    def login_user_id(self):
+        url = "https://api.bilibili.com/x/space/v2/myinfo"
+        data = self.extractor.request(url).json()
+
+        if data["code"] != 0:
+            self.extractor.log.debug("Server response: %s", data)
+            raise exception.StopExtraction("API request failed,Are you login?")
+        try:
+            return data["data"]["profile"]["mid"]
+        except Exception:
+            raise exception.StopExtraction("API request failed")