elite-scripts/edts.py

#!/usr/bin/env python3
# PYTHON_ARGCOMPLETE_OK
import argcomplete, argparse
import json as JSON
import requests
import urllib
import sys

from pyEDSM.edsm.exception import ServerError, NotFoundError

# ===========================================================================

class ProcGenNameError(Exception):
  def __str__(self):
    return "Given system is not a valid proc gen name."

def getCoords(system):
  url = APIURLS["coords"] + urllib.parse.quote(system)
  response = requests.get(url)
  if response.status_code != 200:
    if response.status_code == 400:
      raise ProcGenNameError()
    else:
      raise ServerError(url, {})

  json = response.json()['result']

  ret = json['position']
  ret['uncertainty'] = json['uncertainty']

  return ret

# ===========================================================================

parser = argparse.ArgumentParser(description="Script for interfacing with "
    + "Alot’s hosted EDTS API.")
subparsers = parser.add_subparsers(title="subcommands", help="sub-command help",
    dest="subcommand", required=True)

parser_coords = subparsers.add_parser("coords",
    help="Searches for the approximate coordinates of a given procedurally "
      + "generated system name.")
parser_coords.add_argument("system",
    help="the system name to get coordinates for")
parser_coords.add_argument("--maxuncertainty", nargs="?", type=int,
    help="maximum accepted uncertainty, if any")

argcomplete.autocomplete(parser)
args = parser.parse_args()

# ===========================================================================

APIURLS = {
    "coords": "http://edts.thargoid.space/api/v1/system_position/"
  }

try:
  if args.subcommand == "coords":
    coords = getCoords(args.system)
    if args.maxuncertainty:
      if args.maxuncertainty < coords['uncertainty']:
        raise NotFoundError()
    out = "{},{},{}|{}".format(int(coords['x']), int(coords['y']),
        int(coords['z']), int(coords['uncertainty']))
except ServerError as e:
  print(e)
  sys.exit(1)
except ProcGenNameError as e:
  print(e)
  sys.exit(4)
except NotFoundError as e:
  print("Maximum uncertainty exceeded: " + str(int(coords['uncertainty']))
      + " > " + str(args.maxuncertainty))
  sys.exit(3)
else:
  print(out)
  sys.exit(0)