From fd1b7e50bec5c28a9f6dcef7a147dd79b891c20c Mon Sep 17 00:00:00 2001 From: Melody Horn Date: Sat, 6 Jun 2020 10:11:12 -0600 Subject: persist coordinates alongside camera info --- dev/utah/convert.py | 27 +++++++++++---------------- 1 file changed, 11 insertions(+), 16 deletions(-) (limited to 'dev/utah/convert.py') diff --git a/dev/utah/convert.py b/dev/utah/convert.py index cbad592..b02fc5c 100644 --- a/dev/utah/convert.py +++ b/dev/utah/convert.py @@ -3,27 +3,22 @@ from collections import defaultdict import xml.etree.ElementTree as ET from pprint import pprint -map_areas = [[29, "Statewide"], [44, "I-15 / I-70 Junction"], [30, "Farmington"], [8, "Ogden"], [9, "Park City"], - [14, "Provo / Orem"], [43, "Richfield"], [7, "Salt Lake City"], [31, "Sandy"], [27, "St. George"], - [41, "US-6"], [45, "I-15 Technology Corridor Project"]] -map_areas = dict(map_areas) - -names = dict() +info = dict() for place in ET.parse('UtahKML.xml').findall('.//{*}Placemark'): cam_id = int(place.attrib['id']) cam_name = place.find(".//{*}SimpleData[@name='DisplayName']").text - names[cam_id] = cam_name + coords = [float(x) for x in place.find(".//{*}coordinates").text.split(',')] + info[cam_id] = {'name': cam_name, 'coords': coords} -sources = defaultdict(list) +sources = [] with open('cameras.json', 'r') as f: places = json.load(f) for place in places: - area = place['mapAreaId'] - if area in map_areas: - area = map_areas[area] - cam_id = place['entityId'] - url = place['url'] - name = names[cam_id] - sources[area].append({'id': cam_id, 'url': url, 'name': name}) + cam_id = place['entityId'] + url = place['url'] + this_info = info[cam_id] + name = this_info['name'] + coords = this_info['coords'] + sources.append({'coords': coords, 'cams': [{'id': cam_id, 'url': url, 'name': name}]}) with open('sources.json', 'w') as f: - json.dump(dict(sources), f) + json.dump(sources, f) -- cgit v1.2.3