From 764531951076283d2d7c37e4a1ad59b43df3516f Mon Sep 17 00:00:00 2001 From: Serene-Arc Date: Fri, 18 Feb 2022 12:49:46 +1000 Subject: [PATCH] Fix gfycat after redgifs changes --- bdfr/site_downloaders/gfycat.py | 4 ++-- tests/site_downloaders/test_gfycat.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/bdfr/site_downloaders/gfycat.py b/bdfr/site_downloaders/gfycat.py index 6accaab..c8da9df 100644 --- a/bdfr/site_downloaders/gfycat.py +++ b/bdfr/site_downloaders/gfycat.py @@ -21,7 +21,7 @@ class Gfycat(Redgifs): return super().find_resources(authenticator) @staticmethod - def _get_link(url: str) -> str: + def _get_link(url: str) -> set[str]: gfycat_id = re.match(r'.*/(.*?)/?$', url).group(1) url = 'https://gfycat.com/' + gfycat_id @@ -39,4 +39,4 @@ class Gfycat(Redgifs): raise SiteDownloaderError(f'Failed to download Gfycat link {url}: {e}') except json.JSONDecodeError as e: raise SiteDownloaderError(f'Did not receive valid JSON data: {e}') - return out + return {out,} diff --git a/tests/site_downloaders/test_gfycat.py b/tests/site_downloaders/test_gfycat.py index 981d01d..3b40840 100644 --- a/tests/site_downloaders/test_gfycat.py +++ b/tests/site_downloaders/test_gfycat.py @@ -16,7 +16,7 @@ from bdfr.site_downloaders.gfycat import Gfycat )) def test_get_link(test_url: str, expected_url: str): result = Gfycat._get_link(test_url) - assert result == expected_url + assert result.pop() == expected_url @pytest.mark.online