1
0
Fork 0
mirror of synced 2024-06-25 17:40:17 +12:00

Add shortened Youtube links to downloader factory

This commit is contained in:
Serene-Arc 2021-03-12 13:29:12 +10:00 committed by Ali Parlakci
parent ae0269e13b
commit ba6cf42096
3 changed files with 3 additions and 1 deletions

View file

@ -38,7 +38,7 @@ class DownloadFactory:
return SelfPost
elif re.match(url_beginning + r'v\.redd\.it', url):
return VReddit
elif re.match(url_beginning + r'youtube', url):
elif re.match(url_beginning + r'youtu\.?be', url):
return Youtube
elif re.match(url_beginning + r'i\.redd\.it.*', url) or re.match(url_beginning + r'.*\..{3,4}$', url):
return Direct

View file

@ -35,6 +35,7 @@ from bulkredditdownloader.site_downloaders.youtube import Youtube
('lupb4r', Youtube),
('lul6l7', Redgifs),
('luu376', GifDeliveryNetwork),
('m2l5oo', Youtube),
))
def test_factory_lever_good(test_submission_id: str, expected_class: BaseDownloader, reddit_instance: praw.Reddit):
submission = reddit_instance.submission(id=test_submission_id)

View file

@ -13,6 +13,7 @@ from bulkredditdownloader.site_downloaders.youtube import Youtube
@pytest.mark.long
@pytest.mark.parametrize(('test_submission_id', 'expected_hash'), (
('ltnoqp', '468136300a106c67f1463a7011a6db4a'),
('m2l5oo', 'a70512f7782f13922258297bb12055d9'),
))
def test_find_resources(test_submission_id: str, expected_hash: str, reddit_instance: praw.Reddit):
test_submission = reddit_instance.submission(id=test_submission_id)