feat: bandcamp artist artwork

music_kraken/download/page_attributes.py

@@ -4,6 +4,8 @@ from pathlib import Path
 import re
 import logging
 
+from PIL import Image
+
 from . import FetchOptions, DownloadOptions
 from .results import SearchResults
 from ..objects import (
@@ -17,6 +19,7 @@ from ..objects import (
     Artist,
     Label,
 )
+from ..objects.artwork import ArtworkVariant
 from ..audio import write_metadata_to_target, correct_codec
 from ..utils import output, BColors
 from ..utils.string_processing import fit_to_file_system
@@ -29,9 +32,11 @@ from ..utils.support_classes.download_result import DownloadResult
 from ..utils.exception import MKMissingNameException
 from ..utils.exception.download import UrlNotFoundException
 from ..utils.shared import DEBUG_PAGES
+from ..connection import Connection
 
 from ..pages import Page, EncyclopaediaMetallum, Musify, YouTube, YoutubeMusic, Bandcamp, Genius, INDEPENDENT_DB_OBJECTS
 
+artwork_connection: Connection = Connection()
 
 ALL_PAGES: Set[Type[Page]] = {
     # EncyclopaediaMetallum,
@@ -162,11 +167,64 @@ class Pages:
         
         return False
 
+    def download_artwork_variant_to_target(self, artwork_variant: ArtworkVariant, target: Target):
+        
+        r = artwork_connection.get(
+            url=artwork_variant["url"],
+            name=artwork_variant["url"],
+        )
+
+        temp_target: Target = Target.temp()
+        with temp_target.open("wb") as f:
+            f.write(r.content)
+
+        converted_target: Target = Target.temp(file_extension=main_settings["image_format"])
+        with Image.open(temp_target.file_path) as img:
+            # crop the image if it isn't square in the middle with minimum data loss
+            width, height = img.size
+            if width != height:
+                if width > height:
+                    img = img.crop((width // 2 - height // 2, 0, width // 2 + height // 2, height))
+                else:
+                    img = img.crop((0, height // 2 - width // 2, width, height // 2 + width // 2))
+
+            # resize the image to the preferred resolution
+            img.thumbnail((main_settings["preferred_artwork_resolution"], main_settings["preferred_artwork_resolution"]))
+
+            # https://stackoverflow.com/a/59476938/16804841
+            if img.mode != 'RGB':
+                img = img.convert('RGB')
+
+            img.save(converted_target.file_path, main_settings["image_format"])
+
+
+
+    def _fetch_artist_artwork(self, artist: Artist, naming: dict):
+        naming: Dict[str, List[str]] = defaultdict(list, naming)
+        naming["artist"].append(artist.name)
+        naming["label"].extend([l.title_value for l in artist.label_collection])
+        # removing duplicates from the naming, and process the strings
+        for key, value in naming.items():
+            # https://stackoverflow.com/a/17016257
+            naming[key] = list(dict.fromkeys(value))
+
+        artwork: Artwork = artist.artwork
+        for image_number, variant in enumerate(artwork):
+            naming["image_number"] = [str(image_number)]
+
+            url: str = variant["url"]
+
+            target = Target(
+                relative_to_music_dir=True,
+                file_path=Path(self._parse_path_template(main_settings["artist_artwork_path"], naming=naming))
+            )
+            self.download_artwork_variant_to_target(variant, target)
+
     def download(self, data_object: DataObject, genre: str, **kwargs) -> DownloadResult:
         # fetch the given object
         self.fetch_details(data_object)
         output(f"\nDownloading {data_object.option_string}...", color=BColors.BOLD)
-        
+
         # fetching all parent objects (e.g. if you only download a song)
         if not kwargs.get("fetched_upwards", False):
             to_fetch: List[DataObject] = [data_object]
@@ -185,7 +243,17 @@ class Pages:
                 to_fetch = new_to_fetch
             
             kwargs["fetched_upwards"] = True
-        
+
+        naming = kwargs.get("naming", {
+            "genre": [genre],
+            "audio_format": [main_settings["audio_format"]],
+            "image_format": [main_settings["image_format"]]
+        })
+
+        # download artist artwork
+        if isinstance(data_object, Artist):
+            self._fetch_artist_artwork(artist=data_object, naming=naming)
+
         # download all children
         download_result: DownloadResult = DownloadResult()
         for c in data_object.get_child_collections():
@@ -203,10 +271,7 @@ class Pages:
             I am able to do that, because duplicate values are removed later on.
             """
 
-            self._download_song(data_object, naming={
-                "genre": [genre],
-                "audio_format": [main_settings["audio_format"]],
-            })
+            self._download_song(data_object, naming=naming)
 
         return download_result
 
@@ -325,4 +390,6 @@ class Pages:
         
         _actual_page = self._source_to_page[source.source_type]
         
-        return _actual_page, self._page_instances[_actual_page].fetch_object_from_source(source=source, stop_at_level=stop_at_level)
+        return _actual_page, self._page_instances[_actual_page].fetch_object_from_source(source=source, stop_at_level=stop_at_level)
+
+