feat: wire the mutagen tagger into the worker loop

This commit is contained in:
Jonathan
2026-07-10 23:59:28 +02:00
parent d1fcfd544f
commit 1596bab23e
3 changed files with 59 additions and 2 deletions
+3 -2
View File
@@ -4,7 +4,7 @@ from lyra_worker.claim import claim_next
from lyra_worker.config import get_config
from lyra_worker.db import wait_for_db
from lyra_worker.pipeline import run_pipeline
from lyra_worker.registry import build_adapters, build_resolver
from lyra_worker.registry import build_adapters, build_resolver, build_tagger
IDLE_SLEEP = 2.0
@@ -13,6 +13,7 @@ def run_forever() -> None:
conn = wait_for_db()
adapters = build_adapters(get_config(conn))
resolver = build_resolver()
tagger = build_tagger()
print(f"worker: {len(adapters)} adapter(s) enabled: {[a.name for a in adapters]}", flush=True)
print("worker: waiting for jobs", flush=True)
try:
@@ -22,7 +23,7 @@ def run_forever() -> None:
time.sleep(IDLE_SLEEP)
continue
print(f"worker: claimed job {job_id}", flush=True)
run_pipeline(conn, job_id, adapters, resolver=resolver)
run_pipeline(conn, job_id, adapters, resolver=resolver, tagger=tagger)
print(f"worker: finished job {job_id}", flush=True)
finally:
conn.close()
+7
View File
@@ -1,4 +1,5 @@
from lyra_worker._musicbrainz import MusicBrainzResolver
from lyra_worker._mutagen import MutagenTagger
from lyra_worker.adapters._slskd import SlskdClient
from lyra_worker.adapters._streamrip import StreamripClient
from lyra_worker.adapters._ytdlp import YtDlpClient
@@ -7,6 +8,7 @@ from lyra_worker.adapters.qobuz import QobuzAdapter
from lyra_worker.adapters.soulseek import SoulseekAdapter
from lyra_worker.adapters.youtube import YouTubeAdapter
from lyra_worker.resolver import MbResolver
from lyra_worker.tagger import Tagger
def build_adapters(config: dict) -> list[SourceAdapter]:
@@ -26,3 +28,8 @@ def build_adapters(config: dict) -> list[SourceAdapter]:
def build_resolver() -> MbResolver:
"""The MusicBrainz resolver used to canonicalize requests in intake."""
return MusicBrainzResolver()
def build_tagger() -> Tagger:
"""The tagger that writes canonical metadata onto downloaded files."""
return MutagenTagger()