mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-04 19:40:02 +03:00
76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
"""Export various CSV formats."""
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import time
|
|
from pathlib import Path
|
|
|
|
from config import config
|
|
from edmc_data import commodity_bracketmap as bracketmap
|
|
|
|
# DEFAULT means semi-colon separation
|
|
# CSV means comma separation
|
|
(COMMODITY_DEFAULT, COMMODITY_CSV) = range(2)
|
|
|
|
|
|
def export(data, kind=COMMODITY_DEFAULT, filename=None) -> None:
|
|
"""
|
|
Export commodity data from the given CAPI data.
|
|
|
|
:param data: CAPI data.
|
|
:param kind: The type of file to write.
|
|
:param filename: Filename to write to, or None for a standard format name.
|
|
:return:
|
|
"""
|
|
querytime = config.get_int('querytime', default=int(time.time()))
|
|
|
|
if not filename:
|
|
filename_system = data['lastSystem']['name'].strip()
|
|
filename_starport = data['lastStarport']['name'].strip()
|
|
filename_time = time.strftime('%Y-%m-%dT%H.%M.%S', time.localtime(querytime))
|
|
filename_kind = 'csv'
|
|
filename = f'{filename_system}.{filename_starport}.{filename_time}.{filename_kind}'
|
|
filename = Path(config.get_str('outdir')) / filename
|
|
|
|
if kind == COMMODITY_CSV:
|
|
sep = ';' # BUG: for fixing later after cleanup
|
|
header = sep.join(('System', 'Station', 'Commodity', 'Sell', 'Buy', 'Demand', '', 'Supply', '', 'Date', '\n'))
|
|
rowheader = sep.join((data['lastSystem']['name'], data['lastStarport']['name']))
|
|
|
|
else:
|
|
sep = ','
|
|
header = sep.join(
|
|
('System', 'Station', 'Commodity', 'Sell', 'Buy', 'Demand', '', 'Supply', '', 'Average', 'FDevID', 'Date\n')
|
|
)
|
|
|
|
rowheader = sep.join((data['lastSystem']['name'], data['lastStarport']['name']))
|
|
|
|
with open(filename, 'wt') as h: # codecs can't automatically handle line endings, so encode manually where required
|
|
h.write(header)
|
|
|
|
for commodity in data['lastStarport']['commodities']:
|
|
line = sep.join((
|
|
rowheader,
|
|
commodity['name'],
|
|
commodity['sellPrice'] and str(int(commodity['sellPrice'])) or '',
|
|
commodity['buyPrice'] and str(int(commodity['buyPrice'])) or '',
|
|
str(int(commodity['demand'])) if commodity['demandBracket'] else '',
|
|
bracketmap[commodity['demandBracket']],
|
|
str(int(commodity['stock'])) if commodity['stockBracket'] else '',
|
|
bracketmap[commodity['stockBracket']]
|
|
))
|
|
|
|
if kind == COMMODITY_DEFAULT:
|
|
line = sep.join(
|
|
(
|
|
line,
|
|
str(int(commodity['meanPrice'])),
|
|
str(commodity['id']),
|
|
data['timestamp'] + '\n'
|
|
)
|
|
)
|
|
|
|
else:
|
|
line = sep.join((line, data['timestamp'] + '\n'))
|
|
|
|
h.write(line)
|