mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-08 05:20:03 +03:00
c9555cce66
The output of this was confirmed the 'same' as from before this work: 1. Generate the output. 2. `cut -c 1-68` to a copy. 3. diff those files. This is necessary so as to not have 'every' line be different due to the timestamp on it.
66 lines
2.9 KiB
Python
66 lines
2.9 KiB
Python
"""Export data for Trade Dangerous."""
|
|
|
|
import pathlib
|
|
import time
|
|
from collections import defaultdict
|
|
from operator import itemgetter
|
|
from platform import system
|
|
|
|
from companion import CAPIData
|
|
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: CAPIData) -> None:
|
|
"""Export market data in TD format."""
|
|
data_path = pathlib.Path(config.get_str('outdir'))
|
|
timestamp = time.strftime('%Y-%m-%dT%H.%M.%S', time.strptime(data['timestamp'], '%Y-%m-%dT%H:%M:%SZ'))
|
|
data_filename = f"{data['lastSystem']['name'].strip()}.{data['lastStarport']['name'].strip()}.{timestamp}.prices"
|
|
|
|
# codecs can't automatically handle line endings, so encode manually where
|
|
# required
|
|
with open(data_path / data_filename, 'wb') as h:
|
|
# Format described here: https://bitbucket.org/kfsone/tradedangerous/wiki/Price%20Data
|
|
h.write('#! trade.py import -\n'.encode('utf-8'))
|
|
this_platform = 'darwin' and "Mac OS" or system()
|
|
cmdr_name = data['commander']['name'].strip()
|
|
h.write(
|
|
f'# Created by {applongname} {appversion()} on {this_platform} for Cmdr {cmdr_name}.\n'.encode('utf-8')
|
|
)
|
|
h.write(
|
|
'#\n# <item name> <sellCR> <buyCR> <demand> <stock> <timestamp>\n\n'.encode('utf-8')
|
|
)
|
|
system_name = data['lastSystem']['name'].strip()
|
|
starport_name = data['lastStarport']['name'].strip()
|
|
h.write(f'@ {system_name}/{starport_name}\n'.encode('utf-8'))
|
|
|
|
# sort commodities by category
|
|
by_category = defaultdict(list)
|
|
for commodity in data['lastStarport']['commodities']:
|
|
by_category[commodity['categoryname']].append(commodity)
|
|
|
|
timestamp = time.strftime('%Y-%m-%d %H:%M:%S', time.strptime(data['timestamp'], '%Y-%m-%dT%H:%M:%SZ'))
|
|
for category in sorted(by_category):
|
|
h.write(f' + {format(category)}\n'.encode('utf-8'))
|
|
# corrections to commodity names can change the sort order
|
|
for commodity in sorted(by_category[category], key=itemgetter('name')):
|
|
h.write(
|
|
f" {commodity['name']:<23}"
|
|
f" {int(commodity['sellPrice']):7d}"
|
|
f" {int(commodity['buyPrice']):7d}"
|
|
f" {int(commodity['demand']) if commodity['demandBracket'] else '':9}"
|
|
f"{demandbracketmap[commodity['demandBracket']]:1}"
|
|
f" {int(commodity['stock']) if commodity['stockBracket'] else '':8}"
|
|
f"{stockbracketmap[commodity['stockBracket']]:1}"
|
|
f" {timestamp}\n".encode('utf-8')
|
|
)
|