Add tests, fix style.

This commit is contained in:
BlipRanger
2022-04-25 12:53:59 -04:00
parent dbd0c6cd42
commit d64acc25f5
4 changed files with 44 additions and 3 deletions

View File

@@ -18,7 +18,7 @@ from bdfr.site_downloaders.redgifs import Redgifs
from bdfr.site_downloaders.self_post import SelfPost
from bdfr.site_downloaders.vidble import Vidble
from bdfr.site_downloaders.youtube import Youtube
from bdfr.site_downloaders.vreddit import vreddit
from bdfr.site_downloaders.vreddit import Vreddit
class DownloadFactory:
@@ -49,7 +49,7 @@ class DownloadFactory:
elif re.match(r'i\.redd\.it.*', sanitised_url):
return Direct
elif re.match(r'v\.redd\.it.*', sanitised_url):
return vreddit
return Vreddit
elif re.match(r'pornhub\.com.*', sanitised_url):
return PornHub
elif re.match(r'vidble\.com', sanitised_url):

View File

@@ -16,7 +16,7 @@ from bdfr.site_downloaders.base_downloader import BaseDownloader
logger = logging.getLogger(__name__)
class vreddit(BaseDownloader):
class Vreddit(BaseDownloader):
def __init__(self, post: Submission):
super().__init__(post)