155 lines
4.9 KiB
Python
155 lines
4.9 KiB
Python
from __future__ import annotations
|
|
|
|
from copy import copy
|
|
from dataclasses import dataclass, field
|
|
from typing import Dict, List, Optional, Set, Tuple, Type, TypedDict, Union
|
|
|
|
from ..utils import create_dataclass_instance, custom_hash
|
|
from ..utils.config import main_settings
|
|
from ..utils.enums import PictureType
|
|
from ..utils.string_processing import hash_url, unify
|
|
from .collection import Collection
|
|
from .metadata import ID3Timestamp
|
|
from .metadata import Mapping as id3Mapping
|
|
from .metadata import Metadata
|
|
from .parents import OuterProxy as Base
|
|
|
|
|
|
@dataclass
|
|
class ArtworkVariant:
|
|
url: str
|
|
width: Optional[int] = None
|
|
height: Optional[int] = None
|
|
image_format: Optional[str] = ""
|
|
|
|
def __hash__(self) -> int:
|
|
return custom_hash(self.url)
|
|
|
|
def __eq__(self, other: ArtworkVariant) -> bool:
|
|
return hash(self) == hash(other)
|
|
|
|
def __contains__(self, other: str) -> bool:
|
|
return custom_hash(other) == hash(self.url)
|
|
|
|
|
|
@dataclass
|
|
class Artwork:
|
|
variants: List[ArtworkVariant] = field(default_factory=list)
|
|
|
|
artwork_type: PictureType = PictureType.OTHER
|
|
|
|
def search_variant(self, url: str) -> Optional[ArtworkVariant]:
|
|
if url is None:
|
|
return None
|
|
|
|
for variant in self.variants:
|
|
if url in variant:
|
|
return variant
|
|
|
|
return None
|
|
|
|
def __contains__(self, other: str) -> bool:
|
|
return self.search_variant(other) is not None
|
|
|
|
def add_data(self, **kwargs) -> None:
|
|
variant = self.search_variant(kwargs.get("url"))
|
|
|
|
if variant is None:
|
|
variant, kwargs = create_dataclass_instance(ArtworkVariant, **kwargs)
|
|
self.variants.append(variant)
|
|
|
|
variant.url = url
|
|
variant.__dict__.update(kwargs)
|
|
|
|
|
|
class ArtworkCollection:
|
|
"""
|
|
Stores all the images/artworks for one data object.
|
|
|
|
There could be duplicates before calling ArtworkCollection.compile()
|
|
_this is called before one object is downloaded automatically._
|
|
"""
|
|
|
|
artwork_type: PictureType = PictureType.OTHER
|
|
|
|
def __init__(
|
|
self,
|
|
*data: List[Union[Artwork, ArtworkVariant, dict]],
|
|
parent_artworks: Set[ArtworkCollection] = None,
|
|
crop_images: bool = True
|
|
) -> None:
|
|
# this is used for the song artwork, to fall back to the song artwork
|
|
self.parent_artworks: Set[ArtworkCollection] = parent_artworks or set()
|
|
self.crop_images: bool = crop_images
|
|
|
|
self._data = []
|
|
|
|
def search_artwork(self, url: str) -> Optional[ArtworkVariant]:
|
|
for artwork in self._data:
|
|
if url in artwork:
|
|
return artwork
|
|
|
|
return None
|
|
|
|
def __contains__(self, other: str) -> bool:
|
|
return self.search_artwork(other) is not None
|
|
|
|
def _create_new_artwork(self, **kwargs) -> Tuple[Artwork, dict]:
|
|
kwargs["artwork_type"] = kwargs.get("artwork_type", self.artwork_type)
|
|
|
|
return create_dataclass_instance(Artwork, **kwargs)
|
|
|
|
def add_data(self, url: str, **kwargs) -> None:
|
|
kwargs["url"] = url
|
|
|
|
artwork = self.search_artwork(url)
|
|
|
|
if artwork is None:
|
|
artwork, kwargs = self._create_new_artwork(url=url, **kwargs)
|
|
self._data.append(artwork)
|
|
|
|
artwork.add_data(url, **kwargs)
|
|
|
|
def append(self, value: Union[Artwork, ArtworkVariant, dict], **kwargs):
|
|
if isinstance(value, dict):
|
|
kwargs.update(value)
|
|
value, kwargs = create_dataclass_instance(ArtworkVariant, kwargs)
|
|
|
|
if isinstance(value, ArtworkVariant):
|
|
kwargs["variants"] = [value]
|
|
value, kwargs = create_dataclass_instance(Artwork, kwargs)
|
|
|
|
if isinstance(value, Artwork):
|
|
self._data.append(value)
|
|
return
|
|
|
|
@property
|
|
def flat_empty(self) -> bool:
|
|
return len(self._variant_mapping.keys()) <= 0
|
|
|
|
def _get_best_from_list(self, artwork_variants: List[ArtworkVariant]) -> Optional[ArtworkVariant]:
|
|
return min(artwork_variants, key=lambda x: x["deviation"])
|
|
|
|
@property
|
|
def best_variant(self) -> ArtworkVariant:
|
|
if self.flat_empty:
|
|
return self._get_best_from_list([parent.best_variant for parent in self.parent_artworks])
|
|
return self._get_best_from_list(self._variant_mapping.values())
|
|
|
|
def get_variant_name(self, variant: ArtworkVariant) -> str:
|
|
return f"artwork_{variant['width']}x{variant['height']}_{hash_url(variant['url']).replace('/', '_')}"
|
|
|
|
def __merge__(self, other: ArtworkCollection, **kwargs) -> None:
|
|
self.parent_artworks.update(other.parent_artworks)
|
|
|
|
for key, value in other._variant_mapping.items():
|
|
if key not in self._variant_mapping:
|
|
self._variant_mapping[key] = value
|
|
|
|
def __hash__(self) -> int:
|
|
return id(self)
|
|
|
|
def __iter__(self) -> Generator[ArtworkVariant, None, None]:
|
|
yield from self._variant_mapping.values()
|
|
|
|
|