Skip to content

Commit

Permalink
Add favlist url user check
Browse files Browse the repository at this point in the history
  • Loading branch information
mmmpipi committed Jan 6, 2025
1 parent ddecf8d commit 3f30009
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions gallery_dl/extractor/bilibili.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,6 +97,9 @@ def _init(self):
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"])
Expand Down Expand Up @@ -159,3 +162,15 @@ def user_favlist(self):
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")

0 comments on commit 3f30009

Please sign in to comment.