Spaces:
Running
Running
import os | |
from tempfile import mkdtemp | |
from yt_dlp import YoutubeDL | |
from yt_dlp.postprocessor import PostProcessor | |
class FilenameCollectorPP(PostProcessor): | |
def __init__(self): | |
super(FilenameCollectorPP, self).__init__(None) | |
self.filenames = [] | |
def run(self, information): | |
self.filenames.append(information["filepath"]) | |
return [], information | |
def downloadUrl(url: str): | |
destinationDirectory = mkdtemp() | |
ydl_opts = { | |
"format": "bestaudio/best", | |
'playlist_items': '1', | |
'paths': { | |
'home': destinationDirectory | |
} | |
} | |
filename_collector = FilenameCollectorPP() | |
with YoutubeDL(ydl_opts) as ydl: | |
ydl.add_post_processor(filename_collector) | |
ydl.download([url]) | |
if len(filename_collector.filenames) <= 0: | |
raise Exception("Cannot download " + url) | |
result = filename_collector.filenames[0] | |
print("Downloaded " + result) | |
return result |