Rename module

bdfr/site_downloaders/download_factory.py

@@ -0,0 +1,50 @@
+#!/usr/bin/env python3
+# coding=utf-8
+
+import re
+from typing import Type
+
+from bdfr.exceptions import NotADownloadableLinkError
+from bdfr.site_downloaders.base_downloader import BaseDownloader
+from bdfr.site_downloaders.direct import Direct
+from bdfr.site_downloaders.erome import Erome
+from bdfr.site_downloaders.gallery import Gallery
+from bdfr.site_downloaders.gfycat import Gfycat
+from bdfr.site_downloaders.gif_delivery_network import GifDeliveryNetwork
+from bdfr.site_downloaders.imgur import Imgur
+from bdfr.site_downloaders.redgifs import Redgifs
+from bdfr.site_downloaders.self_post import SelfPost
+from bdfr.site_downloaders.vreddit import VReddit
+from bdfr.site_downloaders.youtube import Youtube
+
+
+class DownloadFactory:
+    @staticmethod
+    def pull_lever(url: str) -> Type[BaseDownloader]:
+        url_beginning = r'\s*(https?://(www\.)?)'
+        if re.match(url_beginning + r'(i\.)?imgur.*\.gifv$', url):
+            return Imgur
+        elif re.match(url_beginning + r'.*/.*\.\w{3,4}(\?[\w;&=]*)?$', url):
+            return Direct
+        elif re.match(url_beginning + r'erome\.com.*', url):
+            return Erome
+        elif re.match(url_beginning + r'reddit\.com/gallery/.*', url):
+            return Gallery
+        elif re.match(url_beginning + r'gfycat\.', url):
+            return Gfycat
+        elif re.match(url_beginning + r'gifdeliverynetwork', url):
+            return GifDeliveryNetwork
+        elif re.match(url_beginning + r'(m\.)?imgur.*', url):
+            return Imgur
+        elif re.match(url_beginning + r'redgifs.com', url):
+            return Redgifs
+        elif re.match(url_beginning + r'reddit\.com/r/', url):
+            return SelfPost
+        elif re.match(url_beginning + r'v\.redd\.it', url):
+            return VReddit
+        elif re.match(url_beginning + r'(m\.)?youtu\.?be', url):
+            return Youtube
+        elif re.match(url_beginning + r'i\.redd\.it.*', url):
+            return Direct
+        else:
+            raise NotADownloadableLinkError(f'No downloader module exists for url {url}')