Added some error handling

victordlp8 · Oct 28, 2022 · f0aaa81 · f0aaa81
1 parent e74f263
commit f0aaa81
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/spotifyVisualizer.py b/spotifyVisualizer.py
@@ -24,8 +24,15 @@ def main():
     DATE_FORMAT = 'Week %W of %Y'
     paths = Paths()
 
-    fileList = [f for f in os.listdir(
-        paths.data) if os.path.isfile(os.path.join(paths.data, f))]
+    if os.path.exists(paths.data):    
+        fileList = [f for f in os.listdir(
+            paths.data) if os.path.isfile(os.path.join(paths.data, f))]
+        if len(fileList) == 0:
+            print('[ERROR] Unable to find StreamingHistory files.')
+            exit()
+    else:
+        print('[ERROR] Please make sure the directory \'data\' exists.')
+        exit()
 
     songs = []
     for file in fileList: