download_factory.py: check if url has ext first
This commit is contained in:
parent
4e35f0db2b
commit
ab29e17511
1 changed files with 5 additions and 5 deletions
|
@ -22,7 +22,11 @@ class DownloadFactory:
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def pull_lever(url: str) -> Type[BaseDownloader]:
|
def pull_lever(url: str) -> Type[BaseDownloader]:
|
||||||
url_beginning = r'\s*(https?://(www\.)?)'
|
url_beginning = r'\s*(https?://(www\.)?)'
|
||||||
if re.match(url_beginning + r'erome\.com.*', url):
|
if re.match(url_beginning + r'i\.imgur.*\.gifv$', url):
|
||||||
|
return Imgur
|
||||||
|
elif re.match(url_beginning + r'.*/.*\.\w{3,4}$', url):
|
||||||
|
return Direct
|
||||||
|
elif re.match(url_beginning + r'erome\.com.*', url):
|
||||||
return Erome
|
return Erome
|
||||||
elif re.match(url_beginning + r'reddit\.com/gallery/.*', url):
|
elif re.match(url_beginning + r'reddit\.com/gallery/.*', url):
|
||||||
return Gallery
|
return Gallery
|
||||||
|
@ -32,8 +36,6 @@ class DownloadFactory:
|
||||||
return GifDeliveryNetwork
|
return GifDeliveryNetwork
|
||||||
elif re.match(url_beginning + r'imgur.*', url):
|
elif re.match(url_beginning + r'imgur.*', url):
|
||||||
return Imgur
|
return Imgur
|
||||||
elif re.match(url_beginning + r'i\.imgur.*\.gifv$', url):
|
|
||||||
return Imgur
|
|
||||||
elif re.match(url_beginning + r'redgifs.com', url):
|
elif re.match(url_beginning + r'redgifs.com', url):
|
||||||
return Redgifs
|
return Redgifs
|
||||||
elif re.match(url_beginning + r'reddit\.com/r/', url):
|
elif re.match(url_beginning + r'reddit\.com/r/', url):
|
||||||
|
@ -44,7 +46,5 @@ class DownloadFactory:
|
||||||
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
|
||||||
elif re.match(url_beginning + r'.*/.*\.\w{3,4}$', url):
|
|
||||||
return Direct
|
|
||||||
else:
|
else:
|
||||||
raise NotADownloadableLinkError(f'No downloader module exists for url {url}')
|
raise NotADownloadableLinkError(f'No downloader module exists for url {url}')
|
||||||
|
|
Loading…
Reference in a new issue