70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
"""
|
|
Author: Uyanide pywang0608@foxmail.com
|
|
Date: 2026-04-05 02:13:49
|
|
Description: Musixmatch metadata enricher (matcher.track.get by Spotify track ID)
|
|
"""
|
|
|
|
from typing import Optional
|
|
|
|
from loguru import logger
|
|
|
|
from .base import BaseEnricher
|
|
from ..authenticators.musixmatch import MusixmatchAuthenticator
|
|
from ..models import TrackMeta
|
|
from ..config import MUSIXMATCH_TRACK_MATCH_URL
|
|
|
|
|
|
class MusixmatchSpotifyEnricher(BaseEnricher):
|
|
"""Fill title, artist, album, and length from Musixmatch using Spotify track ID."""
|
|
|
|
def __init__(self, auth: MusixmatchAuthenticator) -> None:
|
|
self.auth = auth
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
return "musixmatch"
|
|
|
|
@property
|
|
def provides(self) -> set[str]:
|
|
return {"title", "artist", "album", "length"}
|
|
|
|
async def enrich(self, track: TrackMeta) -> Optional[dict]:
|
|
if not track.trackid:
|
|
return None
|
|
|
|
logger.debug(f"Musixmatch enricher: looking up trackid={track.trackid}")
|
|
|
|
try:
|
|
data = await self.auth.get_json(
|
|
MUSIXMATCH_TRACK_MATCH_URL,
|
|
{"track_spotify_id": track.trackid},
|
|
)
|
|
except Exception as e:
|
|
logger.warning(f"Musixmatch enricher: request failed: {e}")
|
|
return None
|
|
|
|
if data is None:
|
|
return None
|
|
|
|
body = data.get("message", {}).get("body")
|
|
t = body.get("track") if isinstance(body, dict) else None
|
|
if not isinstance(t, dict):
|
|
logger.debug(
|
|
f"Musixmatch enricher: no track data for trackid={track.trackid}"
|
|
)
|
|
return None
|
|
|
|
updates: dict = {}
|
|
if isinstance(t.get("track_name"), str) and t["track_name"]:
|
|
updates["title"] = t["track_name"]
|
|
if isinstance(t.get("artist_name"), str) and t["artist_name"]:
|
|
updates["artist"] = t["artist_name"]
|
|
if isinstance(t.get("album_name"), str) and t["album_name"]:
|
|
updates["album"] = t["album_name"]
|
|
if isinstance(t.get("track_length"), int) and t["track_length"] > 0:
|
|
updates["length"] = t["track_length"] * 1000
|
|
|
|
if updates:
|
|
logger.debug(f"Musixmatch enricher: filled {list(updates.keys())}")
|
|
return updates or None
|