Move to different program structure

bulkredditdownloader/site_downloaders/youtube.py

@@ -1,64 +1,37 @@
 #!/usr/bin/env python3
 
 import logging
-import os
 import pathlib
-import sys
+import tempfile
 
 import youtube_dl
+from praw.models import Submission
 
+from bulkredditdownloader.resource import Resource
 from bulkredditdownloader.site_downloaders.base_downloader import BaseDownloader
-from bulkredditdownloader.errors import FileAlreadyExistsError
-from bulkredditdownloader.utils import GLOBAL
 
 logger = logging.getLogger(__name__)
 
 
 class Youtube(BaseDownloader):
-    def __init__(self, directory: pathlib.Path, post: dict):
+    def __init__(self, directory: pathlib.Path, post: Submission):
         super().__init__(directory, post)
-        self.download()
 
     def download(self):
-        self.directory.mkdir(exist_ok=True)
+        return self._download_video()
 
-        filename = GLOBAL.config['filename'].format(**self.post)
-        logger.info(filename)
+    def _download_video(self) -> Resource:
+        with tempfile.TemporaryDirectory() as temp_dir:
+            ydl_opts = {
+                "format": "best",
+                "outtmpl": str(temp_dir / "test.%(ext)s"),
+                "playlistend": 1,
+                "nooverwrites": True,
+                "quiet": True
+            }
+            with youtube_dl.YoutubeDL(ydl_opts) as ydl:
+                ydl.download([self.post.url])
 
-        self._download_video(filename, self.directory, self.post['CONTENTURL'])
-
-    def _download_video(self, filename: str, directory: pathlib.Path, url: str):
-        ydl_opts = {
-            "format": "best",
-            "outtmpl": str(directory / (filename + ".%(ext)s")),
-            "progress_hooks": [self._hook],
-            "playlistend": 1,
-            "nooverwrites": True,
-            "quiet": True
-        }
-        with youtube_dl.YoutubeDL(ydl_opts) as ydl:
-            ydl.download([url])
-
-        location = directory / (filename + ".mp4")
-
-        with open(location, 'rb') as file:
-            content = file.read()
-
-        if GLOBAL.arguments.no_dupes:
-            try:
-                file_hash = self._create_hash(content)
-            except FileNotFoundError:
-                return None
-            if file_hash in GLOBAL.downloadedPosts():
-                os.remove(location)
-                raise FileAlreadyExistsError
-            GLOBAL.downloadedPosts.add(file_hash)
-
-    @staticmethod
-    def _hook(d):
-        if d['status'] == 'finished':
-            return logger.info("Downloaded")
-        downloaded_mbs = int(d['downloaded_bytes'] * (10**(-6)))
-        file_size = int(d['total_bytes'] * (10**(-6)))
-        sys.stdout.write("{}Mb/{}Mb\r".format(downloaded_mbs, file_size))
-        sys.stdout.flush()
+            with open(temp_dir / 'test.mp4', 'rb') as file:
+                content = file.read()
+        return Resource(self.post, self.post.url, content)