mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-07 13:00:04 +03:00
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://github.com/eyeonus/Trade-Dangerous/wiki/Price-Data
|
|
h.write('#! trade.py import -\n'.encode('utf-8'))
|
|
this_platform = 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')
|
|
)
|