Initial implementation of Music Metadata Enricher
AI-powered per-album pipeline: scan → local hints → MusicBrainz/Discogs/Claude resolve → cover art → interactive or auto review → tag write + rename + report. All external dependencies optional; 17/17 unit tests passing. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
parent
b273052f68
commit
f7cf520dbe
8 changed files with 1748 additions and 0 deletions
171
cover_handler.py
Normal file
171
cover_handler.py
Normal file
|
|
@ -0,0 +1,171 @@
|
|||
from __future__ import annotations
|
||||
|
||||
import sys
|
||||
import tempfile
|
||||
import time
|
||||
from pathlib import Path
|
||||
from typing import Optional, List
|
||||
|
||||
try:
|
||||
from PIL import Image
|
||||
HAS_PIL = True
|
||||
except ImportError:
|
||||
HAS_PIL = False
|
||||
|
||||
try:
|
||||
import requests
|
||||
HAS_REQUESTS = True
|
||||
except ImportError:
|
||||
HAS_REQUESTS = False
|
||||
|
||||
try:
|
||||
import musicbrainzngs as mb
|
||||
HAS_MB = True
|
||||
except ImportError:
|
||||
HAS_MB = False
|
||||
|
||||
try:
|
||||
from mutagen.id3 import ID3, APIC, error as ID3Error
|
||||
from mutagen.flac import FLAC, Picture
|
||||
from mutagen.mp4 import MP4, MP4Cover
|
||||
HAS_MUTAGEN = True
|
||||
except ImportError:
|
||||
HAS_MUTAGEN = False
|
||||
|
||||
_MIN_COVER_SIZE = 200 # pixels
|
||||
|
||||
|
||||
def _image_ok(path: Path) -> bool:
|
||||
if not HAS_PIL:
|
||||
return path.stat().st_size > 5000
|
||||
try:
|
||||
with Image.open(path) as img:
|
||||
w, h = img.size
|
||||
return w >= _MIN_COVER_SIZE and h >= _MIN_COVER_SIZE
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
|
||||
def find_local_cover(image_files: List[Path]) -> Optional[Path]:
|
||||
priority = ("front", "folder", "cover", "album")
|
||||
# Sort by priority keyword, then size descending
|
||||
def key(p: Path):
|
||||
name = p.name.lower()
|
||||
score = next((i for i, kw in enumerate(priority) if kw in name), len(priority))
|
||||
size = p.stat().st_size if p.exists() else 0
|
||||
return (score, -size)
|
||||
|
||||
for p in sorted(image_files, key=key):
|
||||
if _image_ok(p):
|
||||
return p
|
||||
return None
|
||||
|
||||
|
||||
def _mb_cover_url(release_mbid: str) -> Optional[str]:
|
||||
url = f"https://coverartarchive.org/release/{release_mbid}/front"
|
||||
if not HAS_REQUESTS:
|
||||
return None
|
||||
try:
|
||||
r = requests.head(url, timeout=5, allow_redirects=True)
|
||||
if r.status_code == 200:
|
||||
return url
|
||||
except Exception:
|
||||
pass
|
||||
return None
|
||||
|
||||
|
||||
def download_cover(release_mbid: Optional[str], dest_dir: Path) -> Optional[Path]:
|
||||
if not release_mbid or not HAS_REQUESTS:
|
||||
return None
|
||||
url = _mb_cover_url(release_mbid)
|
||||
if not url:
|
||||
return None
|
||||
try:
|
||||
r = requests.get(url, timeout=15)
|
||||
if r.status_code == 200:
|
||||
ext = ".jpg"
|
||||
ct = r.headers.get("content-type", "")
|
||||
if "png" in ct:
|
||||
ext = ".png"
|
||||
dest = dest_dir / f"_cover_download{ext}"
|
||||
dest.write_bytes(r.content)
|
||||
if _image_ok(dest):
|
||||
return dest
|
||||
dest.unlink(missing_ok=True)
|
||||
except Exception as e:
|
||||
print(f" ⚠️ Cover-Download-Fehler: {e}", file=sys.stderr)
|
||||
return None
|
||||
|
||||
|
||||
def embed_cover(audio_path: Path, cover_path: Path) -> bool:
|
||||
if not HAS_MUTAGEN:
|
||||
return False
|
||||
try:
|
||||
img_data = cover_path.read_bytes()
|
||||
mime = "image/jpeg" if cover_path.suffix.lower() in (".jpg", ".jpeg") else "image/png"
|
||||
ext = audio_path.suffix.lower()
|
||||
|
||||
if ext == ".mp3":
|
||||
try:
|
||||
tags = ID3(str(audio_path))
|
||||
except ID3Error:
|
||||
tags = ID3()
|
||||
tags.delall("APIC")
|
||||
tags.add(APIC(encoding=3, mime=mime, type=3, desc="Cover", data=img_data))
|
||||
tags.save(str(audio_path), v2_version=4)
|
||||
return True
|
||||
|
||||
elif ext == ".flac":
|
||||
audio = FLAC(str(audio_path))
|
||||
audio.clear_pictures()
|
||||
pic = Picture()
|
||||
pic.type = 3
|
||||
pic.mime = mime
|
||||
pic.desc = "Cover"
|
||||
pic.data = img_data
|
||||
audio.add_picture(pic)
|
||||
audio.save()
|
||||
return True
|
||||
|
||||
elif ext == ".m4a":
|
||||
audio = MP4(str(audio_path))
|
||||
fmt = MP4Cover.FORMAT_JPEG if mime == "image/jpeg" else MP4Cover.FORMAT_PNG
|
||||
audio.tags["covr"] = [MP4Cover(img_data, imageformat=fmt)]
|
||||
audio.save()
|
||||
return True
|
||||
|
||||
else:
|
||||
# Generic mutagen fallback
|
||||
from mutagen import File as MutagenFile
|
||||
audio = MutagenFile(str(audio_path), easy=False)
|
||||
if audio is not None:
|
||||
if audio.tags is None:
|
||||
audio.add_tags()
|
||||
if hasattr(audio.tags, "add"):
|
||||
audio.tags.add(
|
||||
APIC(encoding=3, mime=mime, type=3, desc="Cover", data=img_data)
|
||||
)
|
||||
audio.save()
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
print(f" ⚠️ Cover-Einbettungsfehler {audio_path.name}: {e}", file=sys.stderr)
|
||||
return False
|
||||
|
||||
|
||||
def resolve_cover(
|
||||
image_files: List[Path],
|
||||
release_mbid: Optional[str],
|
||||
album_dir: Path,
|
||||
) -> tuple[Optional[Path], Optional[str]]:
|
||||
"""Returns (cover_path, source_label)."""
|
||||
local = find_local_cover(image_files)
|
||||
if local:
|
||||
return local, "local"
|
||||
|
||||
if release_mbid:
|
||||
downloaded = download_cover(release_mbid, album_dir)
|
||||
if downloaded:
|
||||
return downloaded, "musicbrainz"
|
||||
|
||||
return None, None
|
||||
Loading…
Add table
Add a link
Reference in a new issue