bulk-downloader-for-reddit/src/downloaders/youtube.py

import os
import pathlib
import sys

import youtube_dl

from src.downloaders.downloaderUtils import createHash
from src.errors import FileAlreadyExistsError
from src.utils import GLOBAL
from src.utils import printToFile as print



class Youtube:
    def __init__(self, directory: pathlib.Path, post: dict):
        if not os.path.exists(directory):
            os.makedirs(directory)

        filename = GLOBAL.config['filename'].format(**post)
        print(filename)

        self.download(filename, directory, post['CONTENTURL'])

    def download(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")

        if GLOBAL.arguments.no_dupes:
            try:
                file_hash = createHash(str(location))
            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 print("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()