IDer
This commit is contained in:
240
linedance-app/local/acoustid_worker.py
Normal file
240
linedance-app/local/acoustid_worker.py
Normal file
@@ -0,0 +1,240 @@
|
||||
"""
|
||||
acoustid_worker.py — Baggrunds-fingerprinting via AcoustID/Chromaprint.
|
||||
|
||||
Finder sange uden MBID, kører fpcalc på dem, spørger AcoustID API
|
||||
og gemmer MBID + acoustid lokalt. Kører som lavprioritets baggrundstråd.
|
||||
|
||||
Kræver fpcalc (Chromaprint) installeret:
|
||||
Linux: sudo apt install libchromaprint-tools
|
||||
Windows: download fra https://acoustid.org/chromaprint
|
||||
"""
|
||||
|
||||
import json
|
||||
import logging
|
||||
import subprocess
|
||||
import threading
|
||||
import time
|
||||
from pathlib import Path
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# AcoustID API nøgle — gratis til open-source apps
|
||||
# https://acoustid.org/api-key
|
||||
ACOUSTID_API_KEY = "8XaBELgH" # Offentlig test-nøgle, skift til din egen ved produktion
|
||||
ACOUSTID_API_URL = "https://api.acoustid.org/v2/lookup"
|
||||
|
||||
# Pause mellem API-kald (AcoustID tillader 3/sek)
|
||||
API_DELAY = 0.4
|
||||
|
||||
|
||||
def find_fpcalc() -> str | None:
|
||||
"""Find fpcalc på systemet."""
|
||||
import shutil, sys
|
||||
|
||||
# Tjek PATH
|
||||
path = shutil.which("fpcalc")
|
||||
if path:
|
||||
return path
|
||||
|
||||
# Windows: tjek ved siden af exe
|
||||
if sys.platform == "win32":
|
||||
candidates = [
|
||||
Path(sys.executable).parent / "fpcalc.exe",
|
||||
Path("C:/Program Files/Chromaprint/fpcalc.exe"),
|
||||
Path("C:/Program Files (x86)/Chromaprint/fpcalc.exe"),
|
||||
]
|
||||
for c in candidates:
|
||||
if c.exists():
|
||||
return str(c)
|
||||
|
||||
return None
|
||||
|
||||
|
||||
def fingerprint_file(path: str, fpcalc: str) -> tuple[str, int] | None:
|
||||
"""
|
||||
Kør fpcalc på en fil og returnér (fingerprint, duration).
|
||||
Returnerer None ved fejl.
|
||||
"""
|
||||
try:
|
||||
result = subprocess.run(
|
||||
[fpcalc, "-json", path],
|
||||
capture_output=True, text=True, timeout=30
|
||||
)
|
||||
if result.returncode != 0:
|
||||
return None
|
||||
data = json.loads(result.stdout)
|
||||
fp = data.get("fingerprint", "")
|
||||
dur = int(data.get("duration", 0))
|
||||
if fp and dur:
|
||||
return fp, dur
|
||||
except Exception as e:
|
||||
logger.warning(f"fpcalc fejl for {path}: {e}")
|
||||
return None
|
||||
|
||||
|
||||
def lookup_acoustid(fingerprint: str, duration: int) -> dict | None:
|
||||
"""
|
||||
Spørg AcoustID API om MBID for et fingerprint.
|
||||
Returnerer dict med 'mbid' og 'acoustid' eller None.
|
||||
"""
|
||||
try:
|
||||
import urllib.request, urllib.parse
|
||||
|
||||
params = urllib.parse.urlencode({
|
||||
"client": ACOUSTID_API_KEY,
|
||||
"fingerprint": fingerprint,
|
||||
"duration": duration,
|
||||
"meta": "recordings",
|
||||
})
|
||||
url = f"{ACOUSTID_API_URL}?{params}"
|
||||
req = urllib.request.Request(url)
|
||||
req.add_header("User-Agent", "LineDancePlayer/1.0")
|
||||
|
||||
with urllib.request.urlopen(req, timeout=10) as resp:
|
||||
data = json.loads(resp.read())
|
||||
|
||||
if data.get("status") != "ok":
|
||||
return None
|
||||
|
||||
results = data.get("results", [])
|
||||
if not results:
|
||||
return None
|
||||
|
||||
# Tag det bedste resultat (højeste score)
|
||||
best = max(results, key=lambda r: r.get("score", 0))
|
||||
if best.get("score", 0) < 0.85:
|
||||
return None # For usikkert
|
||||
|
||||
acoustid = best.get("id", "")
|
||||
recordings = best.get("recordings", [])
|
||||
mbid = recordings[0].get("id", "") if recordings else ""
|
||||
|
||||
return {"acoustid": acoustid, "mbid": mbid} if acoustid else None
|
||||
|
||||
except Exception as e:
|
||||
logger.warning(f"AcoustID API fejl: {e}")
|
||||
return None
|
||||
|
||||
|
||||
def run_acoustid_scan(db_path: str, on_progress=None, stop_event: threading.Event = None):
|
||||
"""
|
||||
Gennemgå alle sange uden MBID og forsøg AcoustID fingerprinting.
|
||||
Kører som baggrundsjob.
|
||||
"""
|
||||
import sqlite3
|
||||
|
||||
fpcalc = find_fpcalc()
|
||||
if not fpcalc:
|
||||
logger.info("AcoustID: fpcalc ikke fundet — spring over")
|
||||
if on_progress:
|
||||
on_progress(0, 0, "fpcalc ikke installeret")
|
||||
return
|
||||
|
||||
logger.info(f"AcoustID: fpcalc fundet: {fpcalc}")
|
||||
|
||||
conn = sqlite3.connect(db_path)
|
||||
conn.row_factory = sqlite3.Row
|
||||
|
||||
# Find sange uden MBID der har en lokal fil
|
||||
rows = conn.execute("""
|
||||
SELECT id, local_path, title, artist
|
||||
FROM songs
|
||||
WHERE (mbid IS NULL OR mbid = '')
|
||||
AND file_missing = 0
|
||||
AND local_path IS NOT NULL AND local_path != ''
|
||||
ORDER BY RANDOM()
|
||||
LIMIT 500
|
||||
""").fetchall()
|
||||
|
||||
total = len(rows)
|
||||
done = 0
|
||||
found = 0
|
||||
|
||||
logger.info(f"AcoustID: {total} sange uden MBID")
|
||||
|
||||
for row in rows:
|
||||
if stop_event and stop_event.is_set():
|
||||
logger.info("AcoustID: stoppet af bruger")
|
||||
break
|
||||
|
||||
path = row["local_path"]
|
||||
if not Path(path).exists():
|
||||
done += 1
|
||||
continue
|
||||
|
||||
if on_progress:
|
||||
on_progress(done, total, row["title"] or path)
|
||||
|
||||
# Fingerprint filen
|
||||
fp_result = fingerprint_file(path, fpcalc)
|
||||
if not fp_result:
|
||||
done += 1
|
||||
time.sleep(0.05)
|
||||
continue
|
||||
|
||||
fingerprint, duration = fp_result
|
||||
|
||||
# Spørg AcoustID
|
||||
result = lookup_acoustid(fingerprint, duration)
|
||||
time.sleep(API_DELAY) # Respektér rate limit
|
||||
|
||||
if result:
|
||||
mbid = result.get("mbid", "")
|
||||
acoustid = result.get("acoustid", "")
|
||||
conn.execute(
|
||||
"UPDATE songs SET mbid=?, acoustid=? WHERE id=?",
|
||||
(mbid or None, acoustid or None, row["id"])
|
||||
)
|
||||
conn.commit()
|
||||
found += 1
|
||||
logger.info(
|
||||
f"AcoustID: {row['title']} → MBID={mbid[:8] if mbid else '—'}"
|
||||
)
|
||||
|
||||
done += 1
|
||||
|
||||
conn.close()
|
||||
logger.info(f"AcoustID: færdig — {found}/{total} sange fik MBID")
|
||||
if on_progress:
|
||||
on_progress(total, total, f"Færdig — {found} sange fik MBID")
|
||||
|
||||
|
||||
class AcoustIDWorker:
|
||||
"""
|
||||
Baggrunds-worker der kører AcoustID fingerprinting
|
||||
som lavprioritets tråd.
|
||||
"""
|
||||
|
||||
def __init__(self, db_path: str):
|
||||
self._db_path = db_path
|
||||
self._thread = None
|
||||
self._stop_event = threading.Event()
|
||||
self._running = False
|
||||
|
||||
def start(self, on_progress=None):
|
||||
"""Start baggrunds-fingerprinting. Gør ingenting hvis allerede kører."""
|
||||
if self._running:
|
||||
return
|
||||
self._stop_event.clear()
|
||||
self._running = True
|
||||
|
||||
def _run():
|
||||
try:
|
||||
run_acoustid_scan(
|
||||
self._db_path,
|
||||
on_progress=on_progress,
|
||||
stop_event=self._stop_event,
|
||||
)
|
||||
finally:
|
||||
self._running = False
|
||||
|
||||
self._thread = threading.Thread(target=_run, daemon=True, name="acoustid")
|
||||
self._thread.start()
|
||||
logger.info("AcoustID: baggrunds-worker startet")
|
||||
|
||||
def stop(self):
|
||||
"""Stop baggrunds-fingerprinting."""
|
||||
self._stop_event.set()
|
||||
|
||||
def is_running(self) -> bool:
|
||||
return self._running
|
||||
Reference in New Issue
Block a user