initial commit

src/core/ydl.py

@@ -0,0 +1,29 @@
+from __future__ import unicode_literals
+from urllib.parse import urlparse
+
+import youtube_dl
+
+from src.exceptions.download_exceptions import SiteNotImplementedException
+
+
+class VideoDownloader:
+    def __init__(self, link: str, ydl_opts: dict = None, username: str = None, password: str = None):
+        self.link = link
+        self.ydl_opts = ydl_opts
+        self.username = username
+        self.password = password
+
+        self.SUPPORTING_WEBSITES = [
+            "ok.ru", "vk.com", "www.youtube.com"
+        ]
+
+    def download(self):
+        domain = urlparse(self.link).netloc
+        if domain not in self.SUPPORTING_WEBSITES:
+            raise SiteNotImplementedException
+
+        with youtube_dl.YoutubeDL(self.ydl_opts if self.ydl_opts else {}) as ydl:
+            ydl.download([self.link])
+            result = ydl.extract_info(self.link, download=False)
+            url = result['formats'][-1]
+        return url