mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-08 05:20:03 +03:00
ad8c63c64c
Also refactors data.py to edmc_data.py as I'm having weird issues with PyCharm debugger not starting, and this seems to be to do with module name clashes.
65 lines
2.7 KiB
Python
65 lines
2.7 KiB
Python
# Export to Trade Dangerous
|
|
|
|
from os.path import join
|
|
from collections import defaultdict
|
|
import codecs
|
|
import numbers
|
|
from operator import itemgetter
|
|
from platform import system
|
|
from sys import platform
|
|
import time
|
|
|
|
from config import applongname, appversion, config
|
|
|
|
# These are specific to Trade Dangerous, so don't move to edmc_data.py
|
|
demandbracketmap = { 0: '?',
|
|
1: 'L',
|
|
2: 'M',
|
|
3: 'H', }
|
|
stockbracketmap = { 0: '-',
|
|
1: 'L',
|
|
2: 'M',
|
|
3: 'H', }
|
|
|
|
def export(data):
|
|
|
|
querytime = config.get_int('querytime', default=int(time.time()))
|
|
|
|
filename = join(config.get_str('outdir'), '%s.%s.%s.prices' % (data['lastSystem']['name'].strip(), data['lastStarport']['name'].strip(), time.strftime('%Y-%m-%dT%H.%M.%S', time.localtime(querytime))))
|
|
|
|
timestamp = time.strftime('%Y-%m-%d %H:%M:%S', time.strptime(data['timestamp'], '%Y-%m-%dT%H:%M:%SZ'))
|
|
|
|
# Format described here: https://bitbucket.org/kfsone/tradedangerous/wiki/Price%20Data
|
|
h = open(filename, 'wb') # codecs can't automatically handle line endings, so encode manually where required
|
|
h.write('#! trade.py import -\n# Created by {appname} {appversion} on {platform} for Cmdr {cmdr}.\n'
|
|
'#\n# <item name> <sellCR> <buyCR> <demand> <stock> <timestamp>\n\n'
|
|
'@ {system}/{starport}\n'.format(
|
|
appname=applongname,
|
|
appversion=appversion(),
|
|
platform=platform == 'darwin' and "Mac OS" or system(),
|
|
cmdr=data['commander']['name'].strip(),
|
|
system=data['lastSystem']['name'].strip(),
|
|
starport=data['lastStarport']['name'].strip()
|
|
).encode('utf-8'))
|
|
|
|
# sort commodities by category
|
|
bycategory = defaultdict(list)
|
|
for commodity in data['lastStarport']['commodities']:
|
|
bycategory[commodity['categoryname']].append(commodity)
|
|
|
|
for category in sorted(bycategory):
|
|
h.write(' + {}\n'.format(category).encode('utf-8'))
|
|
# corrections to commodity names can change the sort order
|
|
for commodity in sorted(bycategory[category], key=itemgetter('name')):
|
|
h.write(' {:<23} {:7d} {:7d} {:9}{:1} {:8}{:1} {}\n'.format(
|
|
commodity['name'],
|
|
int(commodity['sellPrice']),
|
|
int(commodity['buyPrice']),
|
|
int(commodity['demand']) if commodity['demandBracket'] else '',
|
|
demandbracketmap[commodity['demandBracket']],
|
|
int(commodity['stock']) if commodity['stockBracket'] else '',
|
|
stockbracketmap[commodity['stockBracket']],
|
|
timestamp).encode('utf-8'))
|
|
|
|
h.close()
|