gpx2json.py

#!/usr/bin/python
from xml.etree import ElementTree as ET
import json
import os
import argparse
import logging
import random
import gpxpy.geo

vcount = {2: logging.DEBUG,
		  1: logging.INFO,
		  None: logging.WARNING}

#argument parser
parser = argparse.ArgumentParser(description='Converts c:geo gpx to geojson.')
parser.add_argument("-j", "--json", help="Outputs formatted GeoJSON.", action="store_true")
parser.add_argument("-o", "--original", help="Exports location of 'Original Coordinates' waypoint instead of changed cache location.", action="store_true")
parser.add_argument("-v", "--verbose", help="Verbose output.", action="count")
parser.add_argument("-p", "--premium", help="Protect premium caches by adding random noise to degrees.", action="store_true")
parser.add_argument('gpx', nargs='+', help="GPX files exported from c:geo")
args = parser.parse_args()

logging.basicConfig(level=vcount[args.verbose])

#geocache variables
ns = {"tg": "http://www.topografix.com/GPX/1/0",
	  "gsak": "http://www.gsak.net/xmlv1/6"}
features = []
unknowns_detected = []
names = []
colors = {"Geocache|Traditional Cache": "#02874d",
		  "Geocache|Unknown Cache": "#12508c",
		  "Geocache|Multi-cache": "#e98300",
		  "Geocache|Virtual Cache": "#ffffff",
		  "Geocache|Earthcache": "#009bbb",
		  "Geocache|Wherigo Cache": "#1562a6",
		  "Geocache|Letterbox hybrid": "#f1f2f3",
		  "fallback": "#ff0000"}

for filename in args.gpx:
	logging.warning("====================")
	logging.warning("Parsing %s", filename)
	#XML parsing
	tree = ET.parse(filename)
	gpx = tree.getroot()
	for wpt in gpx:
		if wpt.find('tg:sym', ns).text == "Geocache Found":
			gc_name = wpt.find('tg:name', ns).text
			gc_desc = wpt.find('tg:desc', ns).text
			gc_type = wpt.find('tg:type', ns).text
			gc_lat = float(wpt.attrib['lat'])
			gc_lon = float(wpt.attrib['lon'])
			if wpt.find('gsak:wptExtension', ns).find('gsak:IsPremium', ns).text == "true":
				if args.premium:
					gc_lat_orig = gc_lat
					gc_lon_orig = gc_lon
					gc_lat = gc_lat + round(random.uniform(-0.0002,+0.0002), 5)
					gc_lon = gc_lon + round(random.uniform(-0.0002,+0.0002), 5)
					logging.info("Premium found: %s - %s. Adjusted distance is %fm", gc_name, gc_desc, gpxpy.geo.haversine_distance(gc_lat_orig, gc_lon_orig, gc_lat, gc_lon))
			info_orig = ""
			if args.original:
				path = "tg:wpt/gsak:wptExtension/[gsak:Parent='" + gc_name + "']/../[tg:sym='Original Coordinates']"
				orig_wpt = gpx.find(path, ns)
				try:
					logging.debug("Waypoints: %s, name: %s", str(orig_wpt), orig_wpt.find('tg:name', ns).text)
					gc_lat = float(orig_wpt.attrib['lat'])
					gc_lon = float(orig_wpt.attrib['lon'])
					info_orig = " (using original wpt)"
				except AttributeError:
					pass
			try:
				gc_color = colors[gc_type]
			except KeyError:
				gc_color = colors["fallback"]
				unknowns_detected.append(gc_type)
			logging.info("%s %s %s %f %f%s", gc_name, gc_type, gc_desc, gc_lat, gc_lon, info_orig)
			properties = {"title": gc_name,
				 "description": gc_desc,
				 "type": gc_type,
				 "marker-color": gc_color,
				 "stroke-width": 1}
			geometry = {"type": "Point",
			   "coordinates": [gc_lon, gc_lat]}
			feature = {"type": "Feature",
			  "properties": properties,
			  "geometry": geometry}
			features.append(feature)
			names.append(gc_name)

def find_dups(l):
	return list(set([x for x in l if l.count(x) > 1]))
duplicates = find_dups(names)

logging.warning("====================")
logging.warning("Unidentified types: %i", len(unknowns_detected))
if len(unknowns_detected) != 0:
	logging.warning("Detected unidentified cache types: %s", unknowns_detected)
logging.warning("Duplicate caches: %i", len(duplicates))
if len(duplicates) != 0:
	logging.warning("List of duplicates: %s", duplicates)
logging.warning("Total caches added: %i", len(features))
logging.warning("====================")
geojson = {"type": "FeatureCollection", "features": features}
if args.json:
	print(json.JSONEncoder(indent=2, ensure_ascii=False).encode(geojson))