81 lines
2.9 KiB
Python
Executable file
81 lines
2.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
||
# PYTHON_ARGCOMPLETE_OK
|
||
import argcomplete, argparse
|
||
import math
|
||
import sys
|
||
|
||
from pyEDSM.edsm.exception import ServerError, NotFoundError
|
||
from pyEDSM.edsm.models import System, Commander
|
||
|
||
# ===========================================================================
|
||
|
||
def distanceBetween(system1, system2):
|
||
coords1 = System(system1).coords
|
||
coords2 = System(system2).coords
|
||
return int(round(math.sqrt( (coords1['x']-coords2['x'])**2
|
||
+ (coords1['y']-coords2['y'])**2
|
||
+ (coords1['z']-coords2['z'])**2 ),0))
|
||
|
||
def findCommander(name, apikey):
|
||
return Commander(name, apikey).currentSystem
|
||
|
||
def getProfileUrl(name, apikey):
|
||
return Commander(name, apikey).profileUrl
|
||
|
||
def getBodyCount(system):
|
||
return System(system).bodyCount
|
||
|
||
# ===========================================================================
|
||
|
||
parser = argparse.ArgumentParser(description="A collection of tools useful for "
|
||
+ "exploration.")
|
||
subparsers = parser.add_subparsers(title="subcommands", help="sub-command help",
|
||
dest="subCommand")
|
||
|
||
parser_bodycount = subparsers.add_parser("bodycount",
|
||
help="Returns the number of bodies in a system. Will exit with code 1 on "
|
||
+ "server error and code 2 if the system could not be found in EDSM.")
|
||
parser_bodycount.add_argument("system", nargs=1, help="system to query")
|
||
|
||
parser_distance = subparsers.add_parser("distancebetween",
|
||
help="Calculates the distance between two systems. Will exit with code 1 "
|
||
+ "on server error and code 2 if (one of) the systems could not be found "
|
||
+ "on EDSM.")
|
||
parser_distance.add_argument("system", nargs=2, help="the systems to measure")
|
||
|
||
parser_find = subparsers.add_parser("findcommander",
|
||
help="Attempts to find a CMDR’s last known position. Will exit with code 1 "
|
||
+ "on server error and code 2 if the CMDR could not be found on EDSM.")
|
||
group = parser_find.add_mutually_exclusive_group(required=False)
|
||
group.add_argument('--system', action='store_true',
|
||
help='output the commander’s last known system (default)')
|
||
group.add_argument('--url', action='store_true',
|
||
help='output the commander’s profile URL')
|
||
parser_find.add_argument("name", help="the commander in question")
|
||
parser_find.add_argument("apikey", default="", nargs="?",
|
||
help="the commander’s EDSM API key. Can be empty for public profiles.")
|
||
|
||
argcomplete.autocomplete(parser)
|
||
args = parser.parse_args()
|
||
|
||
# ===========================================================================
|
||
|
||
try:
|
||
if args.subCommand == "bodycount":
|
||
out = getBodyCount(args.system[0])
|
||
elif args.subCommand == "distancebetween":
|
||
out = distanceBetween(args.system[0], args.system[1])
|
||
elif args.subCommand == "findcommander":
|
||
if args.url:
|
||
out = getProfileUrl(args.name, args.apikey)
|
||
else:
|
||
out = findCommander(args.name, args.apikey)
|
||
except ServerError as e:
|
||
print(e)
|
||
sys.exit(1)
|
||
except NotFoundError as e:
|
||
print(e)
|
||
sys.exit(2)
|
||
else:
|
||
print(out)
|
||
sys.exit(0)
|