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

Update regex to be more concise, add test for m.youtube.

Browse Source
This commit is contained in:
BlipRanger
2021-04-02 11:51:00 -04:00
committed by Ali Parlakci
parent a49d87e154
commit c9c864b71b
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bulkredditdownloader/site_downloaders/download_factory.py
View File

@@ -40,9 +40,7 @@ class DownloadFactory:
return SelfPost return SelfPost
elif re.match(url_beginning + r'v\.redd\.it', url): elif re.match(url_beginning + r'v\.redd\.it', url):
return VReddit return VReddit
elif re.match(url_beginning + r'youtu\.?be', url): elif re.match(url_beginning + r'(m\.)?youtu\.?be', url):
return Youtube
elif re.match(url_beginning + r'm.youtu\.?be', url):
return Youtube return Youtube
elif re.match(url_beginning + r'i\.redd\.it.*', url): elif re.match(url_beginning + r'i\.redd\.it.*', url):
return Direct return Direct

1
bulkredditdownloader/tests/downloaders/test_download_factory.py
View File

@@ -34,6 +34,7 @@ from bulkredditdownloader.site_downloaders.youtube import Youtube
('https://redgifs.com/watch/courageousimpeccablecanvasback', Redgifs), ('https://redgifs.com/watch/courageousimpeccablecanvasback', Redgifs),
('https://www.gifdeliverynetwork.com/repulsivefinishedandalusianhorse', GifDeliveryNetwork), ('https://www.gifdeliverynetwork.com/repulsivefinishedandalusianhorse', GifDeliveryNetwork),
('https://youtu.be/DevfjHOhuFc', Youtube), ('https://youtu.be/DevfjHOhuFc', Youtube),
('https://m.youtube.com/watch?v=kr-FeojxzUM', Youtube),
)) ))
def test_factory_lever_good(test_submission_url: str, expected_class: BaseDownloader, reddit_instance: praw.Reddit): def test_factory_lever_good(test_submission_url: str, expected_class: BaseDownloader, reddit_instance: praw.Reddit):
result = DownloadFactory.pull_lever(test_submission_url) result = DownloadFactory.pull_lever(test_submission_url)