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

Merge pull request #733 from Soulsuck24/development

Browse Source
This commit is contained in:
Serene
2022-12-27 12:04:41 +10:00
committed by GitHub
parent 7cd1a70f59 13887ca7e1
commit 0bb94040d6
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bdfr/site_downloaders/redgifs.py
View File

@@ -27,7 +27,9 @@ class Redgifs(BaseDownloader):
try: try:
if url.endswith("/"): if url.endswith("/"):
url = url.removesuffix("/") url = url.removesuffix("/")
redgif_id = re.match(r".*/(.*?)(?:\?.*|\..{0,})?$", url).group(1) redgif_id = re.match(r".*/(.*?)(?:\?.*|\..{0,})?$", url).group(1).lower()
if redgif_id.endswith("-mobile"):
redgif_id = redgif_id.removesuffix("-mobile")
except AttributeError: except AttributeError:
raise SiteDownloaderError(f"Could not extract Redgifs ID from {url}") raise SiteDownloaderError(f"Could not extract Redgifs ID from {url}")
return redgif_id return redgif_id

2
tests/site_downloaders/test_redgifs.py
View File

@@ -16,6 +16,8 @@ from bdfr.site_downloaders.redgifs import Redgifs
("https://redgifs.com/watch/frighteningvictorioussalamander", "frighteningvictorioussalamander"), ("https://redgifs.com/watch/frighteningvictorioussalamander", "frighteningvictorioussalamander"),
("https://www.redgifs.com/watch/genuineprivateguillemot/", "genuineprivateguillemot"), ("https://www.redgifs.com/watch/genuineprivateguillemot/", "genuineprivateguillemot"),
("https://www.redgifs.com/watch/marriedcrushingcob?rel=u%3Akokiri.girl%3Bo%3Arecent", "marriedcrushingcob"), ("https://www.redgifs.com/watch/marriedcrushingcob?rel=u%3Akokiri.girl%3Bo%3Arecent", "marriedcrushingcob"),
("https://thumbs4.redgifs.com/DismalIgnorantDrongo.mp4", "dismalignorantdrongo"),
("https://thumbs4.redgifs.com/DismalIgnorantDrongo-mobile.mp4", "dismalignorantdrongo"),
), ),
) )
def test_get_id(test_url: str, expected: str): def test_get_id(test_url: str, expected: str):