luciano
e7fde842f2
Tab Upgrade: - Checkbox "Scelta automatica" per skippare il modal quando ci sono multi-match (usa bitrate DESC) - Anteprima audio nel picker: play button sul source + ogni candidato via bridge read_audio_data_url (WKWebView blocca file://) - Bottone "Annulla tutto" nel picker → interrompe l'intera coda - Fix mtime: shutil.copy (non copy2) così il file upgradato ha data odierna nel Finder - Fix stop audio → non triggera più toast di errore - Modal picker più largo (620-780px) con footer wrap Extra: campo release_date estratto anche in Beatport (publish_date) e YouTube (upload_date) — completato scaffold sort colonna Data
84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
"""Ricerca su YouTube via yt-dlp (subprocess, flat playlist metadata)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import subprocess
|
|
|
|
from core.paths import find_ytdlp, subprocess_flags
|
|
|
|
|
|
_TIMEOUT_SEC = 30
|
|
|
|
|
|
def search_youtube(query: str, limit: int = 50) -> list:
|
|
"""Cerca su YouTube (ytsearchN:query) e ritorna metadata "flat" (senza scaricare).
|
|
|
|
Args:
|
|
query: testo di ricerca libero
|
|
limit: numero massimo di risultati (default 50)
|
|
|
|
Returns:
|
|
list[dict] con {id, url, title, channel, duration_sec}. Vuota se nessun match.
|
|
|
|
Raises:
|
|
RuntimeError: se yt-dlp non disponibile, timeout, o exit non-zero.
|
|
"""
|
|
ytdlp = find_ytdlp()
|
|
if not ytdlp:
|
|
raise RuntimeError("yt-dlp non trovato nel bundle / PATH")
|
|
|
|
cmd = [
|
|
ytdlp,
|
|
"--flat-playlist",
|
|
"--dump-single-json",
|
|
"--no-warnings",
|
|
f"ytsearch{limit}:{query}",
|
|
]
|
|
try:
|
|
proc = subprocess.run(
|
|
cmd,
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=_TIMEOUT_SEC,
|
|
**subprocess_flags(),
|
|
)
|
|
except subprocess.TimeoutExpired as e:
|
|
raise RuntimeError(f"Timeout ricerca YouTube ({_TIMEOUT_SEC}s)") from e
|
|
|
|
if proc.returncode != 0:
|
|
tail = (proc.stderr or "").strip().splitlines()[-3:]
|
|
raise RuntimeError(f"yt-dlp error: {' | '.join(tail) or 'unknown'}")
|
|
|
|
try:
|
|
data = json.loads(proc.stdout)
|
|
except json.JSONDecodeError as e:
|
|
raise RuntimeError(f"yt-dlp output non-JSON: {e}") from e
|
|
|
|
entries = data.get("entries") or []
|
|
result: list = []
|
|
for e in entries:
|
|
if not isinstance(e, dict):
|
|
continue
|
|
video_id = e.get("id") or ""
|
|
url = e.get("url") or (f"https://www.youtube.com/watch?v={video_id}" if video_id else "")
|
|
image_url = f"https://i.ytimg.com/vi/{video_id}/mqdefault.jpg" if video_id else ""
|
|
# yt-dlp `upload_date` è YYYYMMDD (stringa). Lo converto in ISO YYYY-MM-DD.
|
|
# Con --flat-playlist può essere assente; in tal caso resta stringa vuota.
|
|
raw_date = str(e.get("upload_date") or "").strip()
|
|
release_date = ""
|
|
if len(raw_date) == 8 and raw_date.isdigit():
|
|
release_date = f"{raw_date[0:4]}-{raw_date[4:6]}-{raw_date[6:8]}"
|
|
elif raw_date:
|
|
release_date = raw_date # forma sconosciuta, passa così
|
|
result.append({
|
|
"id": video_id,
|
|
"url": url,
|
|
"title": e.get("title") or "",
|
|
"channel": e.get("uploader") or e.get("channel") or "",
|
|
"duration_sec": int(e.get("duration") or 0),
|
|
"image_url": image_url,
|
|
"release_date": release_date,
|
|
})
|
|
return result
|