some refactoring

This commit is contained in:
alterNERDtive 2020-01-08 08:49:09 +01:00
parent ce5817faab
commit 3def9ad3c3

View file

@ -9,6 +9,9 @@ from pyEDSM.edsm.models import System, Commander
# =========================================================================== # ===========================================================================
def getBodyCount(system):
return System(system).bodyCount
def distanceBetween(system1, system2): def distanceBetween(system1, system2):
coords1 = System(system1).coords coords1 = System(system1).coords
coords2 = System(system2).coords coords2 = System(system2).coords
@ -16,14 +19,11 @@ def distanceBetween(system1, system2):
+ (coords1['y']-coords2['y'])**2 + (coords1['y']-coords2['y'])**2
+ (coords1['z']-coords2['z'])**2 ),0)) + (coords1['z']-coords2['z'])**2 ),0))
def findCommander(name, apikey): def getCommanderProfileUrl(name, apikey):
return Commander(name, apikey).currentSystem
def getProfileUrl(name, apikey):
return Commander(name, apikey).profileUrl return Commander(name, apikey).profileUrl
def getBodyCount(system): def getCommanderSystem(name, apikey):
return System(system).bodyCount return Commander(name, apikey).currentSystem
# =========================================================================== # ===========================================================================
@ -67,9 +67,9 @@ try:
out = distanceBetween(args.system[0], args.system[1]) out = distanceBetween(args.system[0], args.system[1])
elif args.subCommand == "findcommander": elif args.subCommand == "findcommander":
if args.url: if args.url:
out = getProfileUrl(args.name, args.apikey) out = getCommanderProfileUrl(args.name, args.apikey)
else: else:
out = findCommander(args.name, args.apikey) out = getCommanderSystem(args.name, args.apikey)
except ServerError as e: except ServerError as e:
print(e) print(e)
sys.exit(1) sys.exit(1)