Catch JSONDecodeError (#17)

* Catch JSONDecodeError * review
sleepyfran · Jul 12, 2024 · 39e0c4d · 39e0c4d
1 parent 4ae6fc1
commit 39e0c4d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/itunespy/__init__.py b/itunespy/__init__.py
@@ -22,6 +22,7 @@
 
 import pycountry
 import requests
+from json.decoder import JSONDecodeError
 from urllib.parse import urlencode
 from itunespy import artist
 from itunespy import music_artist
@@ -61,6 +62,8 @@ def search(
     try:
         json = r.json()['results']
         result_count = r.json()['resultCount']
+    except JSONDecodeError:
+        raise RuntimeError(general_no_connection)
     except:
         raise ConnectionError(general_no_connection)
 
@@ -105,6 +108,8 @@ def lookup(
     try:
         json = r.json()['results']
         result_count = r.json()['resultCount']
+    except JSONDecodeError:
+        raise RuntimeError(general_no_connection)
     except KeyError:
         raise ConnectionError(general_no_connection)