fix list issues on ratings imports

1 files changed, 6 insertions, 3 deletions

diff --git a/clerk_helper b/clerk_helper
index f9d7d1f..f12e95d 100755
--- a/clerk_helper
+++ b/clerk_helper
@@ -310,7 +310,7 @@ def importTrackRatings(args):
             artist = x['artist']
             title = x['title']
             album = x['album']
-            if isinstance(x, list):
+            if isinstance(x['track'], list):
                 track = x['track'][0]
             else:
                 track = x['track']
@@ -329,12 +329,15 @@ def importAlbumRatings(args):
             artist = x['albumartist']
             album = x['album']
             date = x['date']
-            if isinstance(x, list):
+            if isinstance(x['track'], list):
                 track = x['track'][0]
             else:
                 track = x['track']
             if 'disc' in x:
-                disc = x['disc']
+                if isinstance(x['disc'], list):
+                    disc = x['disc'][0]
+                else:
+                    disc = x['disc']
             else:
                 disc = ""
         entry = {'albumartist': artist, 'track': track, 'album': album, 'date': date, 'disc': disc, 'rating': rating}