85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
import logging
|
|
|
|
import yt_dlp
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def extract_audio_stream(video_id: str) -> dict | None:
|
|
"""Extract playable audio stream URL from a YouTube video using yt-dlp."""
|
|
url = f"https://www.youtube.com/watch?v={video_id}"
|
|
|
|
ydl_opts = {
|
|
"format": "bestaudio/best",
|
|
"quiet": True,
|
|
"no_warnings": True,
|
|
"extract_flat": False,
|
|
"user_agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
|
|
}
|
|
|
|
try:
|
|
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
|
info = ydl.extract_info(url, download=False)
|
|
|
|
if not info:
|
|
logger.error("No info extracted for video %s", video_id)
|
|
return None
|
|
|
|
formats = info.get("formats", [])
|
|
if not formats:
|
|
logger.error("No formats available for video %s", video_id)
|
|
return None
|
|
|
|
audio_url = None
|
|
stream_type = None
|
|
|
|
for fmt in formats:
|
|
protocol = fmt.get("protocol", "")
|
|
has_audio = fmt.get("acodec", "none") != "none"
|
|
|
|
if not has_audio:
|
|
continue
|
|
|
|
if protocol.startswith("m3u8"):
|
|
audio_url = fmt.get("url")
|
|
stream_type = "hls"
|
|
break
|
|
|
|
if not audio_url:
|
|
for fmt in formats:
|
|
protocol = fmt.get("protocol", "")
|
|
fmt_note = fmt.get("format_note", "")
|
|
has_audio = fmt.get("acodec", "none") != "none"
|
|
|
|
if not has_audio:
|
|
continue
|
|
|
|
if protocol.startswith("https") and "audio" in fmt_note.lower():
|
|
audio_url = fmt.get("url")
|
|
stream_type = "direct"
|
|
break
|
|
elif protocol.startswith("https") and has_audio:
|
|
audio_url = fmt.get("url")
|
|
stream_type = "direct"
|
|
|
|
if not audio_url:
|
|
logger.error("No audio stream found for video %s", video_id)
|
|
return None
|
|
|
|
return {
|
|
"videoId": video_id,
|
|
"url": audio_url,
|
|
"streamType": stream_type,
|
|
"title": info.get("title", "Unknown"),
|
|
"channel": info.get("channel", "Unknown"),
|
|
"duration": info.get("duration"),
|
|
"isLive": info.get("live_status") == "live",
|
|
}
|
|
|
|
except yt_dlp.utils.DownloadError as e:
|
|
logger.error("yt-dlp error for video %s: %s", video_id, e)
|
|
return None
|
|
except Exception as e:
|
|
logger.error("Unexpected error for video %s: %s", video_id, e)
|
|
return None
|