diff --git a/bdfr/site_downloaders/redgifs.py b/bdfr/site_downloaders/redgifs.py index 625cf7d..84fb3c3 100644 --- a/bdfr/site_downloaders/redgifs.py +++ b/bdfr/site_downloaders/redgifs.py @@ -24,6 +24,8 @@ class Redgifs(BaseDownloader): @staticmethod def _get_link(url: str) -> set[str]: try: + if url.endswith("/"): + url = url.removesuffix("/") redgif_id = re.match(r".*/(.*?)(\..{0,})?$", url).group(1) except AttributeError: raise SiteDownloaderError(f"Could not extract Redgifs ID from {url}") diff --git a/tests/site_downloaders/test_redgifs.py b/tests/site_downloaders/test_redgifs.py index 0e1a497..bfe683f 100644 --- a/tests/site_downloaders/test_redgifs.py +++ b/tests/site_downloaders/test_redgifs.py @@ -29,6 +29,7 @@ from bdfr.site_downloaders.redgifs import Redgifs "UnripeUnkemptWoodpecker-large.jpg", }, ), + ("https://www.redgifs.com/watch/genuineprivateguillemot/", {"GenuinePrivateGuillemot.mp4"}), ), ) def test_get_link(test_url: str, expected: set[str]):