forked from computertech/techdj
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import yt_dlp
|
|
import os
|
|
import re
|
|
|
|
def clean_filename(title):
|
|
# Remove quotes and illegal characters
|
|
title = title.strip("'").strip('"')
|
|
return re.sub(r'[\\/*?:"<>|]', "", title)
|
|
|
|
def download_mp3(url, quality='320'):
|
|
print(f"\n🔍 Processing: {url}")
|
|
|
|
# 1. Fetch Title First (Clean Name)
|
|
try:
|
|
with yt_dlp.YoutubeDL({'quiet':True}) as ydl:
|
|
info = ydl.extract_info(url, download=False)
|
|
if 'entries' in info:
|
|
title = '%(title)s' # Playlist default
|
|
else:
|
|
title = clean_filename(info['title'])
|
|
print(f"📝 Renaming to: {title}")
|
|
except Exception as e:
|
|
print(f"❌ Error fetching info: {e}")
|
|
return {"success": False, "error": "Invalid URL or video unavailable"}
|
|
|
|
# 2. Download Options (adjustable quality)
|
|
ydl_opts = {
|
|
'format': 'bestaudio/best',
|
|
'postprocessors': [{
|
|
'key': 'FFmpegExtractAudio',
|
|
'preferredcodec': 'mp3',
|
|
'preferredquality': quality,
|
|
}],
|
|
'outtmpl': f'music/{title}.%(ext)s',
|
|
'quiet': False,
|
|
'no_warnings': True,
|
|
}
|
|
|
|
try:
|
|
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
|
ydl.download([url])
|
|
print("✅ Success! (Hit Refresh in the App)")
|
|
return {"success": True, "title": title}
|
|
except Exception as e:
|
|
print(f"❌ Error: {e}")
|
|
return {"success": False, "error": str(e)}
|
|
|
|
if __name__ == "__main__":
|
|
if not os.path.exists("music"):
|
|
os.makedirs("music")
|
|
|
|
print("--- TECHDJ DOWNLOADER (320kbps) ---")
|
|
while True:
|
|
url = input("\n🔗 URL (q to quit): ").strip()
|
|
if url.lower() == 'q': break
|
|
if url: download_mp3(url)
|