Jonathan
22 lines
757 B
Python
22 lines
757 B
Python
from difflib import SequenceMatcher
|
|
|
|
from lyra_worker.types import Candidate, MBTarget
|
|
|
|
|
|
def _sim(a: str, b: str) -> float:
|
|
return SequenceMatcher(None, a.strip().casefold(), b.strip().casefold()).ratio()
|
|
|
|
|
|
def score_confidence(target: MBTarget, c: Candidate) -> float:
|
|
"""0..1 confidence that candidate c is the requested target."""
|
|
artist_sim = _sim(target.artist, c.matched_artist)
|
|
album_sim = _sim(target.album, c.matched_album)
|
|
|
|
if target.track_count is None or target.track_count == c.track_count:
|
|
track_factor = 1.0
|
|
else:
|
|
diff = abs(target.track_count - c.track_count)
|
|
track_factor = max(0.0, 1.0 - diff / max(target.track_count, 1))
|
|
|
|
return 0.5 * artist_sim + 0.4 * album_sim + 0.1 * track_factor
|