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

[imgur] add "mp4" config key to decide between GIF and MP4

Browse Source 
possible values:
    - false   : always use GIF
    - true    : use MP4 if "prefer_video" flag is set,
                GIF otherwise (default)
    - "always": always use MP4
This commit is contained in:
Mike Fährmann
2017-05-29 08:48:07 +02:00
parent 48ccee2505
commit 011659ced5
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gallery_dl/extractor/imgur.py
View File

@@ -20,6 +20,7 @@ class ImgurExtractor(Extractor):
def __init__(self, match): def __init__(self, match):
Extractor.__init__(self) Extractor.__init__(self)
self.item_id = match.group(1) self.item_id = match.group(1)
self.mp4 = self.config("mp4", True)
def _get_data(self, urlpart): def _get_data(self, urlpart):
response = self.session.get("https://imgur.com/" + urlpart) response = self.session.get("https://imgur.com/" + urlpart)
@@ -28,9 +29,10 @@ class ImgurExtractor(Extractor):
data = text.extract(response.text, "image : ", ",\n")[0] data = text.extract(response.text, "image : ", ",\n")[0]
return self._clean(json.loads(data)) return self._clean(json.loads(data))
@staticmethod def _prepare(self, image):
def _prepare(image): if image["ext"] == ".gif" and (
if image["prefer_video"]: (self.mp4 and image["prefer_video"]) or self.mp4 == "always"
):
image["ext"] = ".mp4" image["ext"] = ".mp4"
url = "https://i.imgur.com/" + image["hash"] + image["ext"] url = "https://i.imgur.com/" + image["hash"] + image["ext"]
image["extension"] = image["ext"][1:] image["extension"] = image["ext"][1:]