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

[wallpapercave] fix extraction

Browse Source
This commit is contained in:
Mike Fährmann
2023-11-24 21:54:21 +01:00
parent 5b979b5706
commit 23cd17997d
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
gallery_dl/extractor/wallpapercave.py
View File

@@ -1,6 +1,7 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# Copyright 2021 David Hoppenbrouwers # Copyright 2021 David Hoppenbrouwers
# Copyright 2023 Mike Fährmann
# #
# This program is free software; you can redistribute it and/or modify # This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as # it under the terms of the GNU General Public License version 2 as
@@ -22,7 +23,20 @@ class WallpapercaveImageExtractor(Extractor):
def items(self): def items(self):
page = self.request(text.ensure_http_scheme(self.url)).text page = self.request(text.ensure_http_scheme(self.url)).text
path = None
for path in text.extract_iter(page, 'class="download" href="', '"'): for path in text.extract_iter(page, 'class="download" href="', '"'):
image = text.nameext_from_url(path) image = text.nameext_from_url(path)
yield Message.Directory, image yield Message.Directory, image
yield Message.Url, self.root + path, image yield Message.Url, self.root + path, image
if path is None:
try:
path = text.rextract(
page, 'href="', '"', page.index('id="tdownload"'))[0]
except Exception:
pass
else:
image = text.nameext_from_url(path)
yield Message.Directory, image
yield Message.Url, self.root + path, image