Merge pull request #726 from Soulsuck24/development

This commit is contained in:
Serene
2022-12-20 10:28:10 +10:00
committed by GitHub
2 changed files with 3 additions and 0 deletions

View File

@@ -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}")

View File

@@ -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]):