Mirrors/gallery-dl
1
0
Fork 0
mirror of https://github.com/mikf/gallery-dl.git synced 2024-11-22 02:32:33 +01:00
Code Issues Releases Wiki Activity

[pixiv] add bookmark extractor

Browse Source
This commit is contained in:
Mike Fährmann 2015-11-23 02:58:31 +01:00
parent daf5fcd416
commit a4c880f4c5
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
gallery_dl/extractor/pixiv.py
View File

@ -165,6 +165,17 @@ class PixivFavoriteExtractor(PixivUserExtractor):
return PixivUserExtractor.prepare_work(self, work["work"])
class PixivBookmarkExtractor(PixivFavoriteExtractor):
"""Extract all favorites/bookmarks of your own account"""
pattern = [r"(?:https?://)?(?:www\.)?pixiv\.net/bookmark\.php()$"]
def __init__(self, match):
PixivFavoriteExtractor.__init__(self, match)
self.api.login()
self.artist_id = self.api.user_id
def require_login(func):
"""Decorator: auto-login before api-calls"""
def wrap(self, *args):
@ -219,7 +230,10 @@ class PixivAPI():
raise Exception("login() failed! check username and password.\n"
"HTTP %s: %s" % (response.status_code, response.text))
try:
token = self._parse(response)["response"]["access_token"]
response = self._parse(response)["response"]
token = response["access_token"]
self.token_timeout = response["expires_in"]
self.user_id = response["user"]["id"]
except:
raise Exception("Get access_token error! Response: %s" % (token))
pconf["access-token"] = token