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

[kemonoparty] ensure all files have an 'extension' (#2740)

Browse Source
This commit is contained in:
Mike Fährmann
2022-07-10 13:33:55 +02:00
parent 74865adae5
commit 7c0505868c
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
gallery_dl/extractor/kemonoparty.py
View File

@@ -96,12 +96,14 @@ class KemonopartyExtractor(Extractor):
post["num"] += 1 post["num"] += 1
post["_http_headers"] = headers post["_http_headers"] = headers
text.nameext_from_url(file.get("name", url), post)
if not post["extension"]:
post["extension"] = text.ext_from_url(url)
if url[0] == "/": if url[0] == "/":
url = self.root + "/data" + url url = self.root + "/data" + url
elif url.startswith(self.root): elif url.startswith(self.root):
url = self.root + "/data" + url[20:] url = self.root + "/data" + url[20:]
text.nameext_from_url(file.get("name", url), post)
yield Message.Url, url, post yield Message.Url, url, post
def login(self): def login(self):
@@ -377,12 +379,15 @@ class KemonopartyDiscordExtractor(KemonopartyExtractor):
for post["num"], file in enumerate(files, 1): for post["num"], file in enumerate(files, 1):
post["type"] = file["type"] post["type"] = file["type"]
url = file["path"] url = file["path"]
text.nameext_from_url(file.get("name", url), post)
if not post["extension"]:
post["extension"] = text.ext_from_url(url)
if url[0] == "/": if url[0] == "/":
url = self.root + "/data" + url url = self.root + "/data" + url
elif url.startswith(self.root): elif url.startswith(self.root):
url = self.root + "/data" + url[20:] url = self.root + "/data" + url[20:]
text.nameext_from_url(file["name"], post)
yield Message.Url, url, post yield Message.Url, url, post
def posts(self): def posts(self):