Add support for saving JSON to file

.gitignore

@@ -6,4 +6,7 @@ venv/
 .idea
 
 credentials.sh
-.cache-*
+.cache*
+output/
+.DS_Store
+

src/display_utils.py

@@ -6,6 +6,7 @@
 # Written by Mark Koh
 # 2/3/2018
 import json
+import os
 
 def print_header(message, length=30):
     """
@@ -87,6 +88,20 @@ def print_audio_analysis_for_track(track, track_analysis):
     print(json.dumps(track_analysis, indent=2))
 
 
+def save_audio_analysis_for_track(track, track_analysis):
+    """
+    Given a track and a analysis response, print out the analysis JSON
+    :param track:
+    :param track_analysis:
+    :return:
+    """
+    if not os.path.exists('output'):
+        os.mkdir('output')
+    with open('output/{}.json'.format(track_string(track)), 'w') as f:
+        print('\nJSON saved to \'output/{}.json\''.format(track_string(track)))
+        f.write(json.dumps(track_analysis, ensure_ascii=False, indent=2))
+
+
 def choose_tracks(tracks):
     """
     Given a list of tracks, list them on the console and let the user choose a

src/main.py

@@ -10,6 +10,7 @@
     print_header,
     track_string,
     print_audio_features_for_track,
+    save_audio_analysis_for_track,
     print_audio_analysis_for_track,
     choose_tracks
     )
@@ -131,15 +132,27 @@ def get_audio_analysis(spotify, tracks, pretty_print=False):
     # can be really big
     tracks_analysis = {}
 
-    print_header('Getting Audio Audio Analysis...')
+
+    try:
+        print("""
+How would you like you output?:
+  1.) Save to a file
+  2.) Print to terminal""")
+        program_choice = input('Choice: ')
+    except ValueError as e:
+        print('Error: Invalid input.')
+
     for track_id in track_map.keys():
         analysis = spotify.audio_analysis(track_id)
         tracks_analysis[track_id] = analysis
 
         # Print out the track info and audio features
         if pretty_print:
             track = track_map.get(track_id)
-            print_audio_analysis_for_track(track, analysis)
+            if program_choice == '1':
+                save_audio_analysis_for_track(track, analysis)
+            elif program_choice == '2':
+                print_audio_analysis_for_track(track, analysis)
 
     return tracks_analysis