Spaces:
Sleeping
Sleeping
File size: 1,563 Bytes
a03c9b4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
import os
import json
import numpy as np
from pytube import YouTube
def downloadMp3(yt, idx, askPath=0):
# extract only audio
video = yt.streams.filter(only_audio=True).first()
destination = 'mp3File'
# check for destination to save file
if (askPath == 1):
print("Enter the destination (leave blank for default dir mp3File)")
destination = str(input(">> ")) or 'mp3File'
# download the file
out_file = video.download(output_path=destination)
# save the file
# base, ext = os.path.splitext(out_file)
dir_path, file_base = os.path.split(out_file)
new_file = os.path.join(dir_path, f'{idx}.mp3')
os.rename(out_file, new_file)
# result of success
print(yt.title + " has been successfully downloaded.")
MISSING_FILE_IDS = [
16, 26, 33, 38, 40, 50, 53, 55, 60, 81, 82, 98, 107, 122, 126, 127, 129, 141, 145, 150, 172,
201, 205, 206, 215, 216, 221, 226, 232, 240, 243, 245, 255, 257, 267, 273, 278, 279, 285, 287,
291, 304, 312, 319, 321, 325, 329, 332, 333, 336, 337, 342, 359, 375, 402, 417, 438, 445, 454,
498
]
data_link_file = '../../../data/mir_St500_yourmt3_16k/MIR-ST500_20210206/MIR-ST500_link.json'
data_link = json.load(open(data_link_file, 'r'))
download_fail = []
for i in MISSING_FILE_IDS:
print(f'Downloading {i}...')
yt = YouTube(data_link[str(i)])
try:
downloadMp3(yt, idx=i)
except:
download_fail.append(i)
print(f'Failed to download {i}.')
print(f'Failed to download {len(download_fail)} files: {download_fail}') |