CodeMirror/bulk-downloader-for-reddit
1
0
Fork 0
mirror of synced 2024-06-01 18:09:47 +12:00
Code Issues Releases Wiki Activity

Tweak regex to fix incorrect match (#237)

Browse source
This commit is contained in:
Serene 2021-03-30 17:08:08 +10:00 committed by Ali Parlakci
parent 5d86c2d400
commit 44889d5264
2 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
bulkredditdownloader/site_downloaders/download_factory.py
View file

@ -44,7 +44,7 @@ class DownloadFactory:
return Youtube
elif re.match(url_beginning + r'i\.redd\.it.*', url):
return Direct
elif re.match(url_beginning + r'.*\..{3,4}$', url):
elif re.match(url_beginning + r'.*/.*\.\w{3,4}$', url):
return Direct
else:
raise NotADownloadableLinkError('No downloader module exists for url {}'.format(url))

4
bulkredditdownloader/tests/downloaders/test_download_factory.py
View file

@ -46,6 +46,10 @@ def test_factory_lever_good(test_submission_id: str, expected_class: BaseDownloa
@pytest.mark.parametrize('test_url', (
'random.com',
'bad',
'https://www.google.com/',
'https://www.google.com',
'https://www.google.com/test',
'https://www.google.com/test/',
))
def test_factory_lever_bad(test_url: str):
with pytest.raises(NotADownloadableLinkError):