Add some tests for base_download

This commit is contained in:
Serene-Arc authored and Ali Parlakci committed 2021-04-18 16:43:15 +03:00
1 parent f573038a21
commit 5ef58f147f
4 files changed
+45 -2

No files matched your search

@@ -96,8 +96,9 @@ class BaseDownloader(ABC):
@staticmethod
def _get_extension(url: str) -> str:
pattern = re.compile(r'(\.(jpg|jpeg|png|mp4|webm|gif))')
if len(results := re.search(pattern, url).groups()) > 1:
return results[1]
if results := re.search(pattern, url):
if len(results.groups()) > 1:
return results[0]
if "v.redd.it" not in url:
return '.jpg'
else: