github/gallery-dl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[hentaifoundry] update 'YII_CSRF_TOKEN' cookie handling

Browse Source 
(fixes #1083)
This commit is contained in:
Mike Fährmann
2020-10-28 21:49:03 +01:00
parent d83b95fd28
commit 0211af7ca8
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
gallery_dl/extractor/hentaifoundry.py
View File

@@ -20,6 +20,7 @@ class HentaifoundryExtractor(Extractor):
directory_fmt = ("{category}", "{user}") directory_fmt = ("{category}", "{user}")
filename_fmt = "{category}_{index}_{title}.{extension}" filename_fmt = "{category}_{index}_{title}.{extension}"
archive_fmt = "{index}" archive_fmt = "{index}"
cookiedomain = "www.hentai-foundry.com"
root = "https://www.hentai-foundry.com" root = "https://www.hentai-foundry.com"
per_page = 25 per_page = 25
@@ -121,7 +122,13 @@ class HentaifoundryExtractor(Extractor):
def _init_site_filters(self): def _init_site_filters(self):
"""Set site-internal filters to show all images""" """Set site-internal filters to show all images"""
url = self.root + "/?enterAgree=1" url = self.root + "/?enterAgree=1"
response = self.request(url, method="HEAD") self.request(url, method="HEAD")
csrf_token = self.session.cookies.get(
"YII_CSRF_TOKEN", domain=self.cookiedomain)
if not csrf_token:
self.log.warning("Unable to update site content filters")
return
url = self.root + "/site/filters" url = self.root + "/site/filters"
data = { data = {
@@ -148,7 +155,7 @@ class HentaifoundryExtractor(Extractor):
"filter_order" : "date_new", "filter_order" : "date_new",
"filter_type" : "0", "filter_type" : "0",
"YII_CSRF_TOKEN" : text.unquote(text.extract( "YII_CSRF_TOKEN" : text.unquote(text.extract(
response.cookies["YII_CSRF_TOKEN"], "%22", "%22")[0]), csrf_token, "%22", "%22")[0]),
} }
self.request(url, method="POST", data=data) self.request(url, method="POST", data=data)