79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from dataclasses import dataclass, field
|
||
|
|
from pathlib import Path
|
||
|
|
from typing import Optional, List, Dict
|
||
|
|
|
||
|
|
|
||
|
|
AUDIO_EXTENSIONS = {
|
||
|
|
".mp3", ".flac", ".m4a", ".aac", ".ogg", ".opus",
|
||
|
|
".wav", ".wma", ".aiff", ".ape",
|
||
|
|
}
|
||
|
|
IMAGE_EXTENSIONS = {".jpg", ".jpeg", ".png", ".webp", ".bmp", ".gif"}
|
||
|
|
TRACKLIST_EXTENSIONS = {".txt", ".htm", ".html", ".nfo"}
|
||
|
|
PLAYLIST_EXTENSIONS = {".m3u", ".m3u8", ".pls"}
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class ScannedFile:
|
||
|
|
path: Path
|
||
|
|
kind: str # "audio" | "image" | "tracklist" | "playlist" | "other"
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class AlbumScan:
|
||
|
|
album_dir: Path
|
||
|
|
audio_files: List[Path] = field(default_factory=list)
|
||
|
|
image_files: List[Path] = field(default_factory=list)
|
||
|
|
tracklist_files: List[Path] = field(default_factory=list)
|
||
|
|
other_files: List[Path] = field(default_factory=list)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class TrackHints:
|
||
|
|
path: Path
|
||
|
|
track_number: Optional[int] = None
|
||
|
|
disc_number: Optional[int] = None
|
||
|
|
title: Optional[str] = None
|
||
|
|
artist: Optional[str] = None
|
||
|
|
duration: Optional[float] = None
|
||
|
|
existing_tags: Dict[str, str] = field(default_factory=dict)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class AlbumHints:
|
||
|
|
album_dir: Path
|
||
|
|
dir_artist: Optional[str] = None
|
||
|
|
dir_album: Optional[str] = None
|
||
|
|
dir_year: Optional[str] = None
|
||
|
|
tracklist_text: Optional[str] = None # merged text from all tracklist files
|
||
|
|
cover_images: List[Path] = field(default_factory=list)
|
||
|
|
tracks: List[TrackHints] = field(default_factory=list)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class TrackProposal:
|
||
|
|
path: Path
|
||
|
|
title: str
|
||
|
|
artist: str
|
||
|
|
track_number: Optional[int]
|
||
|
|
disc_number: Optional[int]
|
||
|
|
new_filename: Optional[str] = None # only set when --rename is active
|
||
|
|
mbid: Optional[str] = None
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class AlbumProposal:
|
||
|
|
album_dir: Path
|
||
|
|
album: str
|
||
|
|
albumartist: str
|
||
|
|
date: Optional[str]
|
||
|
|
genre: Optional[str]
|
||
|
|
label: Optional[str]
|
||
|
|
mbid: Optional[str] # MusicBrainz release ID
|
||
|
|
cover_path: Optional[Path] # resolved local or downloaded cover
|
||
|
|
cover_source: Optional[str] # "local" | "musicbrainz" | "discogs"
|
||
|
|
tracks: List[TrackProposal]
|
||
|
|
confidence: float
|
||
|
|
sources: List[str] = field(default_factory=list)
|
||
|
|
notes: List[str] = field(default_factory=list)
|