91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
import yt_dlp
|
|
import os
|
|
import shutil
|
|
import re
|
|
|
|
def sanitize_filename(name):
|
|
return re.sub(r'[\\/*?:"<>|@#$%^&+=~`\';]', '', name).strip()
|
|
|
|
def get_download_path():
|
|
paths = [
|
|
'/mnt/sdcard/Music',
|
|
'/storage/emulated/0/Music',
|
|
'/sdcard/Music',
|
|
os.path.join(os.environ.get('HOME', ''), 'Music'),
|
|
os.path.join(os.getcwd(), 'music')
|
|
]
|
|
|
|
for path in paths:
|
|
if os.path.exists(path):
|
|
return path
|
|
os.makedirs(paths[-1], exist_ok=True)
|
|
return paths[-1]
|
|
|
|
def load_music_list():
|
|
music_list = []
|
|
with open('musiclist.txt', 'r') as f:
|
|
for line in f:
|
|
line = line.strip()
|
|
if not line:
|
|
continue
|
|
try:
|
|
parts = line.split('---')
|
|
music_list.append({
|
|
'name': sanitize_filename(parts[0].strip()),
|
|
'id': parts[1].strip(),
|
|
'provider': parts[2].strip().upper()
|
|
})
|
|
except IndexError:
|
|
print(f"Invalid line format: {line}")
|
|
return music_list
|
|
|
|
def download_track(music, download_path):
|
|
base_url = f"https://{'music.' if music['provider'] == 'YTM' else ''}youtube.com/watch?v={music['id']}"
|
|
output_path = os.path.join(download_path, f"{music['name']}.opus")
|
|
|
|
if os.path.exists(output_path):
|
|
print(f"Skipping existing track: {music['name']}")
|
|
return
|
|
|
|
ytdl_opts = {
|
|
'format': 'bestaudio/best',
|
|
'outtmpl': f"{music['name']}.%(ext)s",
|
|
'postprocessors': [
|
|
{'key': 'FFmpegExtractAudio', 'preferredcodec': 'opus'},
|
|
{'key': 'FFmpegMetadata'},
|
|
{'key': 'EmbedThumbnail'}
|
|
],
|
|
'writethumbnail': True,
|
|
'quiet': True,
|
|
'no_warnings': True,
|
|
'logger': MyLogger()
|
|
}
|
|
|
|
try:
|
|
with yt_dlp.YoutubeDL(ytdl_opts) as ydl:
|
|
ydl.download([base_url])
|
|
shutil.move(f"{music['name']}.opus", output_path)
|
|
print(f"Downloaded: {music['name']}")
|
|
except Exception as e:
|
|
print(f"Error downloading {music['name']}: {str(e)}")
|
|
|
|
class MyLogger:
|
|
def debug(self, msg):
|
|
pass
|
|
|
|
def warning(self, msg):
|
|
pass
|
|
|
|
def error(self, msg):
|
|
print(msg)
|
|
|
|
if __name__ == "__main__":
|
|
download_path = get_download_path()
|
|
print(f"Download location: {download_path}")
|
|
|
|
music_list = load_music_list()
|
|
total = len(music_list)
|
|
|
|
for index, music in enumerate(music_list, 1):
|
|
print(f"\nProcessing track {index}/{total}")
|
|
download_track(music, download_path)
|