Skip to content
Snippets Groups Projects
Commit 48cce30f authored by Eliot Berriot's avatar Eliot Berriot
Browse files

FIxed broken import

parent c6de0539
No related branches found
No related tags found
No related merge requests found
......@@ -11,7 +11,6 @@ def download(
url,
target_directory=settings.MEDIA_ROOT,
name="%(id)s.%(ext)s",
extension='vorbis',
bitrate=192):
target_path = os.path.join(target_directory, name)
ydl_opts = {
......@@ -19,10 +18,10 @@ def download(
'outtmpl': target_path,
'postprocessors': [{
'key': 'FFmpegExtractAudio',
'preferredcodec': extension,
'preferredcodec': 'vorbis',
}],
}
_downloader = youtube_dl.YoutubeDL(ydl_opts)
info = _downloader.extract_info(url)
info['audio_file_path'] = target_path.format(**info)
info['audio_file_path'] = target_path % {'id': info['id'], 'ext': 'ogg'}
return info
......@@ -8,6 +8,7 @@ class TestDownloader(TMPDirTestCaseMixin, TestCase):
def test_can_download_audio_from_youtube_url_to_vorbis(self):
data = downloader.download('https://www.youtube.com/watch?v=tPEE9ZwTmy0', target_directory=self.download_dir)
self.assertTrue(
os.path.exists(os.path.join(self.download_dir, 'tPEE9ZwTmy0.ogg'))
)
self.assertEqual(
data['audio_file_path'],
os.path.join(self.download_dir, 'tPEE9ZwTmy0.ogg'))
self.assertTrue(os.path.exists(data['audio_file_path']))
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment