ModerateWinGuy/BDFR_Web
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Redgifs fix

Browse Source 
Handle redgifs link with trailing / causing id to return empty string.
This commit is contained in:
Soulsuck24
2022-12-19 11:02:06 -05:00
parent 7cf096012e
commit 603e7de04d
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bdfr/site_downloaders/redgifs.py
View File

@@ -24,6 +24,8 @@ class Redgifs(BaseDownloader):
@staticmethod @staticmethod
def _get_link(url: str) -> set[str]: def _get_link(url: str) -> set[str]:
try: try:
if url.endswith("/"):
url = url.removesuffix("/")
redgif_id = re.match(r".*/(.*?)(\..{0,})?$", url).group(1) redgif_id = re.match(r".*/(.*?)(\..{0,})?$", url).group(1)
except AttributeError: except AttributeError:
raise SiteDownloaderError(f"Could not extract Redgifs ID from {url}") raise SiteDownloaderError(f"Could not extract Redgifs ID from {url}")

1
tests/site_downloaders/test_redgifs.py
View File

@@ -29,6 +29,7 @@ from bdfr.site_downloaders.redgifs import Redgifs
"UnripeUnkemptWoodpecker-large.jpg", "UnripeUnkemptWoodpecker-large.jpg",
}, },
), ),
("https://www.redgifs.com/watch/genuineprivateguillemot/", {"GenuinePrivateGuillemot.mp4"}),
), ),
) )
def test_get_link(test_url: str, expected: set[str]): def test_get_link(test_url: str, expected: set[str]):