65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import logging
|
|
import os
|
|
import pathlib
|
|
import sys
|
|
|
|
import youtube_dl
|
|
|
|
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):
|
|
super().__init__(directory, post)
|
|
self.download()
|
|
|
|
def download(self):
|
|
self.directory.mkdir(exist_ok=True)
|
|
|
|
filename = GLOBAL.config['filename'].format(**self.post)
|
|
logger.info(filename)
|
|
|
|
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()
|