Type musicbrainz

beetsplug/mbpseudo.py

@@ -43,6 +43,12 @@
     from beets.library import Item
     from beetsplug._typing import JSONDict
 
+    from ._utils.musicbrainz import (
+        Release,
+        ReleaseRelation,
+        ReleaseRelationRelease,
+    )
+
 _STATUS_PSEUDO = "Pseudo-Release"
 
 
@@ -133,7 +139,7 @@
                     yield album_info
 
     @override
-    def album_info(self, release: JSONDict) -> AlbumInfo:
+    def album_info(self, release: Release) -> AlbumInfo:
         official_release = super().album_info(release)
 
         if release.get("status") == _STATUS_PSEUDO:
@@ -147,7 +153,7 @@
 
             if self.config["custom_tags_only"].get(bool):
                 self._replace_artist_with_alias(
                     raw_pseudo_release, pseudo_release
                 )
                 self._add_custom_tags(official_release, pseudo_release)
                 return official_release
@@ -161,30 +167,30 @@
         else:
             return official_release
 
-    def _intercept_mb_release(self, data: JSONDict) -> list[str]:
+    def _intercept_mb_release(self, data: Release) -> list[str]:
         album_id = data["id"] if "id" in data else None
         if self._has_desired_script(data) or not isinstance(album_id, str):
             return []
 
         return [
             pr_id
-            for rel in data.get("release-relations", [])
+            for rel in data.get("release_relations", [])
             if (pr_id := self._wanted_pseudo_release_id(album_id, rel))
             is not None
         ]
 
-    def _has_desired_script(self, release: JSONDict) -> bool:
+    def _has_desired_script(self, release: Release) -> bool:
         if len(self._scripts) == 0:
             return False
-        elif script := release.get("text-representation", {}).get("script"):
+        elif script := release.get("text_representation", {}).get("script"):
             return script in self._scripts
         else:
             return False
 
     def _wanted_pseudo_release_id(
         self,
         album_id: str,
-        relation: JSONDict,
+        relation: ReleaseRelation,
     ) -> str | None:
         if (
             len(self._scripts) == 0
@@ -195,7 +201,7 @@
             return None
 
         release = relation["release"]
         if "id" in release and self._has_desired_script(release):
             self._log.debug(
                 "Adding pseudo-release {0} for main release {1}",
                 release["id"],
@@ -216,9 +222,9 @@
         if len(config["import"]["languages"].as_str_seq()) > 0:
             return
 
-        lang = raw_pseudo_release.get("text-representation", {}).get("language")
-        artist_credits = raw_pseudo_release.get("release-group", {}).get(
-            "artist-credit", []
+        lang = raw_pseudo_release.get("text_representation", {}).get("language")
+        artist_credits = raw_pseudo_release.get("release_group", {}).get(
+            "artist_credit", []
         )
         aliases = [
             artist_credit.get("artist", {}).get("aliases", [])