mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-14 08:17:13 +03:00
115 lines
4.5 KiB
Python
Executable File
115 lines
4.5 KiB
Python
Executable File
#!/usr/bin/python
|
|
#
|
|
# Command-line interface. Requires prior setup through the GUI.
|
|
#
|
|
|
|
import argparse
|
|
import sys
|
|
from time import time, sleep
|
|
|
|
import l10n
|
|
import companion
|
|
import bpc
|
|
import outfitting
|
|
import loadout
|
|
import coriolis
|
|
import shipyard
|
|
import eddb
|
|
import prefs
|
|
from config import appcmdname, appversion, config
|
|
|
|
|
|
l10n.Translations().install_dummy()
|
|
EDDB = eddb.EDDB()
|
|
|
|
SERVER_RETRY = 5 # retry pause for Companion servers [s]
|
|
EXIT_SUCCESS, EXIT_SERVER, EXIT_CREDENTIALS, EXIT_VERIFICATION, EXIT_NOT_DOCKED, EXIT_SYS_ERR = range(6)
|
|
|
|
try:
|
|
# arg parsing
|
|
parser = argparse.ArgumentParser(prog=appcmdname, description='Prints the current system and station (if docked) to stdout and optionally writes ship loadout and/or station data to file. Requires prior setup through the accompanying GUI app.')
|
|
parser.add_argument('-v', '--version', help='print program version and exit', action='store_const', const=True)
|
|
parser.add_argument('-c', metavar='FILE', help='write ship loadout to FILE in Coriolis json format')
|
|
parser.add_argument('-e', metavar='FILE', help='write ship loadout to FILE in E:D Shipyard format')
|
|
parser.add_argument('-m', metavar='FILE', help='write station commodity market data to FILE in CSV format')
|
|
parser.add_argument('-o', metavar='FILE', help='write station outfitting data to FILE in CSV format')
|
|
parser.add_argument('-s', metavar='FILE', help='write station shipyard data to FILE in CSV format')
|
|
args = parser.parse_args()
|
|
if args.version:
|
|
print '%.2f' % (float(''.join(appversion.split('.')[:3])) / 100) # just first three digits
|
|
sys.exit(EXIT_SUCCESS)
|
|
|
|
session = companion.Session()
|
|
session.login(config.get('username'), config.get('password'))
|
|
|
|
querytime = int(time())
|
|
data = session.query()
|
|
config.set('querytime', querytime)
|
|
|
|
# Validation
|
|
if not data.get('commander') or not data['commander'].get('name','').strip():
|
|
sys.stderr.write('Who are you?!\n')
|
|
sys.exit(EXIT_SERVER)
|
|
elif not data.get('lastSystem') or not data['lastSystem'].get('name','').strip() or not data.get('lastStarport') or not data['lastStarport'].get('name','').strip():
|
|
sys.stderr.write('Where are you?!\n') # Shouldn't happen
|
|
sys.exit(EXIT_SERVER)
|
|
elif not data.get('ship') or not data['ship'].get('modules') or not data['ship'].get('name','').strip():
|
|
sys.stderr.write('What are you flying?!\n') # Shouldn't happen
|
|
sys.exit(EXIT_SERVER)
|
|
elif (args.m or args.o or args.s) and not data['commander'].get('docked'):
|
|
print data['lastSystem']['name']
|
|
sys.stderr.write("You're not docked at a station!\n")
|
|
sys.exit(EXIT_NOT_DOCKED)
|
|
|
|
# stuff we can do when not docked
|
|
if args.c:
|
|
coriolis.export(data, args.c)
|
|
if args.e:
|
|
loadout.export(data, args.e)
|
|
|
|
# Finally - the data looks sane and we're docked at a station
|
|
print '%s,%s' % (data['lastSystem']['name'], data['lastStarport']['name'])
|
|
(station_id, has_shipyard, has_outfitting) = EDDB.station(data['lastSystem']['name'], data['lastStarport']['name'])
|
|
|
|
if not data['lastStarport'].get('commodities') and not has_outfitting and not has_shipyard:
|
|
sys.stderr.write("Station doesn't have anything!\n")
|
|
sys.exit(EXIT_SUCCESS)
|
|
|
|
if args.m:
|
|
if data['lastStarport'].get('commodities'):
|
|
# Fixup anomalies in the commodity data
|
|
session.fixup(data['lastStarport']['commodities'])
|
|
bpc.export(data, True, args.m)
|
|
else:
|
|
sys.stderr.write("Station doesn't have a market\n")
|
|
|
|
if args.o:
|
|
if has_outfitting:
|
|
outfitting.export(data, args.o)
|
|
else:
|
|
sys.stderr.write("Station doesn't supply outfitting\n")
|
|
|
|
if args.s:
|
|
if has_shipyard:
|
|
if not data['lastStarport'].get('ships'):
|
|
sleep(SERVER_RETRY)
|
|
data = session.query()
|
|
if data['lastStarport'].get('ships'):
|
|
shipyard.export(data, args.s)
|
|
else:
|
|
sys.stderr.write("Couldn't retrieve shipyard info\n")
|
|
else:
|
|
sys.stderr.write("Station doesn't have a shipyard\n")
|
|
|
|
sys.exit(EXIT_SUCCESS)
|
|
|
|
except companion.ServerError as e:
|
|
sys.stderr.write('Server is down\n')
|
|
sys.exit(EXIT_SERVER_DOWN)
|
|
except companion.CredentialsError as e:
|
|
sys.stderr.write('Invalid Credentials\n')
|
|
sys.exit(EXIT_CREDENTIALS)
|
|
except companion.VerificationRequired:
|
|
sys.stderr.write('Verification Required\n')
|
|
sys.exit(EXIT_VERIFICATION)
|