From 16cc5a44eb64d61edf5f71c89c62a3058235c1a4 Mon Sep 17 00:00:00 2001
From: odecif <odecif@gmail.com>
Date: Fri, 28 Oct 2022 15:25:58 +0200
Subject: [PATCH] Added handling for when artwork is missing

---
 api/modules/game.py | 26 +++++++++++++++++---------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/api/modules/game.py b/api/modules/game.py
index 63067af..c2c613c 100644
--- a/api/modules/game.py
+++ b/api/modules/game.py
@@ -23,15 +23,23 @@ def game(predefinednfo=False, return_dict=False, skip_artwork=False):
     # Add front cover artwork in medium size
     artpath = os.path.dirname(nfopath)+'/art/'
     i = 0
-    for art in nfo['game']['artwork']:
-        if skip_artwork is False:
-            nfo['game']['artwork'][i]['data'] = reduceartcv2(
-                    artpath+art['filename'], 'thumbnail')
-        if art['type'] == 'front':
-            nfo['game']['displayimage'] = reduceartcv2(
-                artpath+art['filename'], 'thumbnail')
-        i += 1
-
+    if 'artwork' in nfo['game']:
+        if nfo['game']['artwork']:
+            for art in nfo['game']['artwork']:
+                if skip_artwork is False:
+                    nfo['game']['artwork'][i]['data'] = reduceartcv2(
+                            artpath+art['filename'], 'thumbnail')
+                if art['type'] == 'front':
+                    nfo['game']['displayimage'] = reduceartcv2(
+                        artpath+art['filename'], 'thumbnail')
+                i += 1
+            if 'displayimage' not in nfo['game']:
+                for art in nfo['game']['artwork']:
+                    if art['type'] == 'cd':
+                        nfo['game']['displayimage'] = reduceartcv2(
+                            artpath+art['filename'], 'thumbnail')
+    else:
+        nfo['game']['displayimage'] = ""
     if return_dict is False:
         return jsonify(nfo)
     return nfo