From c90d98ac72e74ff6370c44f5bc38f0e1fe1bbfbe Mon Sep 17 00:00:00 2001 From: Serene-Arc Date: Wed, 7 Apr 2021 09:19:19 +1000 Subject: [PATCH] Add fix for Imgur gifv links --- bulkredditdownloader/site_downloaders/imgur.py | 2 +- bulkredditdownloader/tests/site_downloaders/test_imgur.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/bulkredditdownloader/site_downloaders/imgur.py b/bulkredditdownloader/site_downloaders/imgur.py index 832729a..943d27b 100644 --- a/bulkredditdownloader/site_downloaders/imgur.py +++ b/bulkredditdownloader/site_downloaders/imgur.py @@ -37,7 +37,7 @@ class Imgur(BaseDownloader): @staticmethod def _get_data(link: str) -> dict: - if re.match(r'.*i\.imgur\.com.*\.gifv$', link): + if re.match(r'.*\.gifv$', link): link = link.replace('i.imgur', 'imgur') link = link.rstrip('.gifv') diff --git a/bulkredditdownloader/tests/site_downloaders/test_imgur.py b/bulkredditdownloader/tests/site_downloaders/test_imgur.py index 451fa62..2b877f6 100644 --- a/bulkredditdownloader/tests/site_downloaders/test_imgur.py +++ b/bulkredditdownloader/tests/site_downloaders/test_imgur.py @@ -42,6 +42,9 @@ def test_get_data_album(test_url: str, expected_gen_dict: dict, expected_image_d ('https://i.imgur.com/dLk3FGY.gifv', {'hash': 'dLk3FGY', 'title': '', 'ext': '.mp4', 'animated': True} ), + ('https://imgur.com/BuzvZwb.gifv', + {'hash': 'BuzvZwb', 'title': '', 'description': 'Akron Glass Works', 'animated': True, 'mimetype': 'video/mp4'}, + ) )) def test_get_data_gif(test_url: str, expected_image_dict: dict): result = Imgur._get_data(test_url)