Skip to content
Snippets Groups Projects
Commit bee4dba4 authored by Ernesto Rico Schmidt's avatar Ernesto Rico Schmidt
Browse files

added support for multi-shows in JSON.

parent f0b1c87d
No related branches found
No related tags found
No related merge requests found
...@@ -2,6 +2,7 @@ from django.conf import settings ...@@ -2,6 +2,7 @@ from django.conf import settings
import json import json
import urllib import urllib
import bisect
from os.path import join from os.path import join
from datetime import datetime, date, timedelta from datetime import datetime, date, timedelta
...@@ -15,17 +16,27 @@ def get_automation_id_choices(): ...@@ -15,17 +16,27 @@ def get_automation_id_choices():
try: try:
shows_json = urllib.urlopen(base_url).read() shows_json = urllib.urlopen(base_url).read()
shows_list = json.loads(shows_json)['shows'] shows_list = json.loads(shows_json)['shows']
multi_shows_list = json.loads(shows_json)['multi-shows']
except IOError: except IOError:
try: try:
with open(cached_shows) as cache: with open(cached_shows) as cache:
shows_list = json.loads(cache.read())['shows'] shows_list = json.loads(cache.read())['shows']
multi_shows_list = json.loads(cache.read())['multi-shows']
except IOError: except IOError:
shows_list = [] shows_list = []
multi_shows_list = []
else: else:
with open(cached_shows, 'w') as cache: with open(cached_shows, 'w') as cache:
cache.write(shows_json) cache.write(shows_json)
shows = [(s['id'], '%d | %s' % (s['id'], s['title'])) for s in shows_list] shows = [(s['id'], '%d | %s' % (s['id'], s['title']), s['title']) for s in shows_list if not s.get('multi')]
[bisect.insort(shows, (s['id'], '%05d | %s' % (s['id'], s['title']), s['title'])) for s in multi_shows_list]
shows.sort(key=lambda show: show[2].lower())
shows = [(s[0], s[1]) for s in shows]
return shows return shows
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment