From 3eb993e3c4e513e93d6e9a6a9303b0351e49cff4 Mon Sep 17 00:00:00 2001
From: 20xd6 <20xd6@airmail.cc>
Date: Fri, 20 Sep 2024 13:17:35 -0400
Subject: [PATCH] mb_confidence now working as a variable.

The value from config.ini is returned as a string. It needs to be cast to an int before use.
---
 api_calls.py      |  4 +++-
 get_artist_art.py | 10 ++++++----
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/api_calls.py b/api_calls.py
index 1015e68..0ecdfa7 100644
--- a/api_calls.py
+++ b/api_calls.py
@@ -8,13 +8,15 @@ def get_mb_id(artist_name, mb_confidence):
         mb_data = response.json()
         if mb_data['count'] > 0:
             if mb_data['artists'][0]['score'] > mb_confidence:
-                return mb_data['artists'][0]['id']
+                return True, mb_data['artists'][0]['id']
             else:
                 print("No artist found of hight enough confidance.")
         else:
             print("No artist found.")
+            return False, ""
     else:
         print(f"Error: {response.status_code}")
+        return False, ""
 
 def get_image(mb_id, ftv_api_key, artist_path):
     ftv_api_url = f'https://webservice.fanart.tv/v3/music/{mb_id}?api_key={ftv_api_key}'
diff --git a/get_artist_art.py b/get_artist_art.py
index df50094..8ff2724 100755
--- a/get_artist_art.py
+++ b/get_artist_art.py
@@ -10,7 +10,7 @@ config.read('config.ini')
 
 music_path = config['music']['dir']
 ftv_api_key = config['fanart_tv']['api_key']
-mb_confidence = config['musicbrainz']['confidence']
+mb_confidence = int(config['musicbrainz']['confidence'])
 
 count = 1
 dir_list = dir_activities.get_all(music_path)
@@ -20,10 +20,12 @@ for artist in dir_list:
         print(dir_activities.has_artist_art(artist))
         print(str(count) + ": " + artist)
         try:
-            mb_id = api_calls.get_mb_id(artist, mb_confidence)
+            found_status, mb_id = api_calls.get_mb_id(artist, mb_confidence)
             # print("Getting ", artist_image)
-            artist_image = api_calls.get_image(mb_id, ftv_api_key, os.path.join(music_path, artist))
-            
+            if found_status:
+                artist_image = api_calls.get_image(mb_id, ftv_api_key, os.path.join(music_path, artist))
+            else:
+                print(f"{artist} returned no results.")
             # api_requests.get_art(artist_image, artist, music_path)
         except Exception as e:
             print("Artist or art not found.")