feat: add domain types and quality ranking

This commit is contained in:
Jonathan
2026-07-10 18:22:47 +02:00
parent b62b4145da
commit 1c013dfdd1
3 changed files with 97 additions and 0 deletions
+23
View File
@@ -0,0 +1,23 @@
from lyra_worker.types import Candidate, Quality
_SOURCE_TIERS = {"qobuz": 0, "soulseek": 1, "youtube": 2}
def quality_class(q: Quality) -> int:
"""3 = hi-res lossless, 2 = CD lossless, 1 = lossy."""
if q.lossless:
hires = (q.bit_depth is not None and q.bit_depth > 16) or (
q.sample_rate is not None and q.sample_rate > 44100
)
return 3 if hires else 2
return 1
def source_tier(source: str) -> int:
"""Lower is better. Unknown sources rank worst."""
return _SOURCE_TIERS.get(source, 99)
def rank_key(c: Candidate) -> tuple[int, int]:
"""Sort key; larger is better. Quality class first, then source tier."""
return (quality_class(c.quality), -source_tier(c.source))