40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""
|
|
Author: Uyanide pywang0608@foxmail.com
|
|
Date: 2026-03-31 06:09:11
|
|
Description: Metadata enrichment pipeline
|
|
"""
|
|
|
|
from loguru import logger
|
|
|
|
from .base import BaseEnricher
|
|
from .audio_tag import AudioTagEnricher
|
|
from .file_name import FileNameEnricher
|
|
from ..models import TrackMeta
|
|
|
|
# Enrichers run in order; earlier ones have higher priority.
|
|
_ENRICHERS: list[BaseEnricher] = [
|
|
AudioTagEnricher(),
|
|
FileNameEnricher(),
|
|
]
|
|
|
|
|
|
def enrich_track(track: TrackMeta) -> TrackMeta:
|
|
"""Run all enrichers and return a track with missing fields filled in.
|
|
|
|
Each enricher sees the cumulative state (earlier enrichers' results
|
|
are already applied). A field is only set if it is currently None.
|
|
"""
|
|
for enricher in _ENRICHERS:
|
|
try:
|
|
result = enricher.enrich(track)
|
|
except Exception as e:
|
|
logger.warning(f"Enricher {enricher.name} failed: {e}")
|
|
continue
|
|
if not result:
|
|
continue
|
|
# Only apply fields that are still None
|
|
updates = {k: v for k, v in result.items() if getattr(track, k, None) is None}
|
|
if updates:
|
|
track = track.model_copy(update=updates)
|
|
return track
|