imgur: download videos as mp4 instead of gif

Some imgur URLS have the extension ".gifv" and show up as a gif,
even though they're actually supposed to be mp4 videos. Imgur
serves all videos/gifs as both .gif and .mp4. The image dict has
a key "prefer_video" to distinguish the two. This commit
overrides the .gif extension if "prefer_video" is true to ensure
we download the submission as originally intended.
This commit is contained in:
Eli Lipsitz
2021-09-12 16:50:31 -05:00
parent 063caf0126
commit 33312687ac
2 changed files with 14 additions and 2 deletions

View File

@@ -32,7 +32,11 @@ class Imgur(BaseDownloader):
return out
def _compute_image_url(self, image: dict) -> Resource:
image_url = 'https://i.imgur.com/' + image['hash'] + self._validate_extension(image['ext'])
ext = self._validate_extension(image['ext'])
if image.get('prefer_video', False):
ext = '.mp4'
image_url = 'https://i.imgur.com/' + image['hash'] + ext
return Resource(self.post, image_url, Resource.retry_download(image_url))
@staticmethod