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

[imgur] simplify gallery extraction

Browse Source
This commit is contained in:
Mike Fährmann
2019-08-20 20:00:43 +02:00
parent 3284c62f22
commit 7d6af936c5
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
gallery_dl/extractor/imgur.py
View File

@@ -161,7 +161,6 @@ class ImgurAlbumExtractor(ImgurExtractor):
def items(self): def items(self):
album = self._extract_data("/a/" + self.key + "/all") album = self._extract_data("/a/" + self.key + "/all")
album["title"] = album["title"].strip()
images = album["album_images"]["images"] images = album["album_images"]["images"]
del album["album_images"] del album["album_images"]
@@ -195,15 +194,14 @@ class ImgurGalleryExtractor(ImgurExtractor):
def items(self): def items(self):
url = self.root + "/a/" + self.key url = self.root + "/a/" + self.key
response = self.request(url, method="HEAD", fatal=False) with self.request(url, method="HEAD", fatal=False) as response:
code = response.status_code
if response.status_code < 400: if code < 400:
extr = ImgurAlbumExtractor extr = ImgurAlbumExtractor
path = "/a/"
else: else:
extr = ImgurImageExtractor extr = ImgurImageExtractor
path = "/" url = self.root + "/" + self.key
response.close()
yield Message.Version, 1 yield Message.Version, 1
yield Message.Queue, self.root + path + self.key, {"_extractor": extr} yield Message.Queue, url, {"_extractor": extr}