Upload musicbrainz_client.py

src/musicbrainz_client.py

@@ -0,0 +1,64 @@
+import musicbrainzngs
+import string
+from modules.console_colors import ULTRASINGER_HEAD, blue_highlighted, red_highlighted
+
+
+def get_music_infos(search_string: str) -> tuple[str, str, str, str]:
+    print(f"{ULTRASINGER_HEAD} Searching song in {blue_highlighted('musicbrainz')}")
+    # https://python-musicbrainzngs.readthedocs.io/en/v0.7.1/usage/#searching
+
+    musicbrainzngs.set_useragent("UltraSinger", "0.1", "https://github.com/rakuri255/UltraSinger")
+
+    # search for artist and titel to get release on the first place
+    artist = None
+    artists = musicbrainzngs.search_artists(search_string)
+    if len(artists['artist-list']) != 0:
+        artist = artists['artist-list'][0]['name'].strip()
+    else:
+        print(f"{ULTRASINGER_HEAD} {red_highlighted('No match found')}")
+        return None, None, None, None
+
+    release = None
+    release_groups = musicbrainzngs.search_release_groups(search_string, artist=artist)
+    if len(release_groups['release-group-list']) != 0:
+        release = release_groups['release-group-list'][0]
+
+    if release is not None and 'artist-credit-phrase' in release:
+        artist = release['artist-credit-phrase'].strip()
+
+    title = None
+    if release is not None and 'title' in release:
+        clean_search_string = search_string.translate(str.maketrans('', '', string.punctuation)).lower().strip()
+        clean_release_title = release['title'].translate(str.maketrans('', '', string.punctuation)).lower().strip()
+        clean_artist = artist.translate(str.maketrans('', '', string.punctuation)).lower().strip()
+
+        # prepare search string when title and artist are the same
+        if clean_release_title == clean_artist:
+            # remove the first appearance of the artist
+            clean_search_string = clean_search_string.replace(clean_artist, "", 1)
+
+        if clean_release_title in clean_search_string:
+            title = release['title'].strip()
+        else:
+            print(
+                f"{ULTRASINGER_HEAD} cant find title {red_highlighted(clean_release_title)} in {red_highlighted(clean_search_string)}")
+
+    if title is None or artist is None:
+        print(f"{ULTRASINGER_HEAD} {red_highlighted('No match found')}")
+        return None, None, None, None
+
+    print(f"{ULTRASINGER_HEAD} Found Titel and Artist {blue_highlighted(title)} by {blue_highlighted(artist)}")
+
+    year = None
+    if 'first-release-date' in release:
+        year = release['first-release-date'].strip()
+        print(f"{ULTRASINGER_HEAD} Found release year: {blue_highlighted(year)}")
+
+    genres = None
+    if 'tag-list' in release:
+        genres = ""
+        for tag in release['tag-list']:
+            genres += f"{tag['name'].strip()},"
+        print(f"{ULTRASINGER_HEAD} Found genres: {blue_highlighted(genres)}")
+
+    return title, artist, year, genres