handle errors better

1 files changed, 16 insertions, 1 deletions

diff --git a/program/utils.py b/program/utils.py
index cdc6a09..9a086da 100644
--- a/program/utils.py
+++ b/program/utils.py
@@ -2,12 +2,27 @@ from django.conf import settings
 
 import json
 import urllib
+from os.path import join
 
 
 def get_automation_id_choices():
     base_url = getattr(settings, 'AUTOMATION_BASE_URL', None)
+    cache_dir = getattr(settings, 'AUTOMATION_CACHE_DIR', 'cache')
+    cached_shows = join(cache_dir, 'shows.json')
     shows = []
     if base_url:
-        shows_list = json.load(urllib.urlopen(base_url))['shows']
+        try:
+            shows_json = urllib.urlopen(base_url).read()
+            shows_list = json.loads(shows_json)['shows']
+        except IOError:
+            try:
+                with open(cached_shows) as cache:
+                    shows_list = json.loads(cache.read())['shows']
+            except IOError:
+                shows_list = []
+        else:
+            with open(cached_shows, 'w') as cache:
+                cache.write(shows_json)
+
         shows = [(s['id'], s['title']) for s in shows_list]
     return shows