import os import youtube_dl from importer.providers.providerbase import ProviderBase class YoutubeDlProviderBase(ProviderBase): regex = ".*" output_template: str = 'source_%(id)s.%(ext)s' format: str = "best" merge_format_output: str = "mp4" _TEST = [{ "url": "https://www.youtube.com/watch?v=Wjrrgrvq1ew", "paths": ["source_Wjrrgrvq1ew.mp4"] }] def download(self): ydl_opts = { 'quiet': True, 'format': self.format, 'merge_output_format': self.merge_format_output, 'outtmpl': self.output_template } with youtube_dl.YoutubeDL(ydl_opts) as ydl: info = ydl.extract_info(self.url, download=True) if info.get('_type', None) == 'playlist': for entry in info['entries']: r = ydl.prepare_filename(entry) self.paths.append(f'{os.path.splitext(r)[0]}.{entry["ext"]}') else: r = ydl.prepare_filename(info) self.paths.append(f'{os.path.splitext(r)[0]}.{info["ext"]}') self.downloaded = True