mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-12 15:27:14 +03:00
61 lines
2.5 KiB
Python
61 lines
2.5 KiB
Python
# Export various CSV formats
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from os.path import join
|
|
import hashlib
|
|
import codecs
|
|
import numbers
|
|
import time
|
|
|
|
from config import config
|
|
|
|
bracketmap = { 0: '',
|
|
1: 'Low',
|
|
2: 'Med',
|
|
3: 'High', }
|
|
|
|
(COMMODITY_DEFAULT, COMMODITY_BPC, COMMODITY_CSV) = range(3)
|
|
|
|
def export(data, kind=COMMODITY_DEFAULT, filename=None):
|
|
|
|
querytime = config.getint('querytime') or int(time.time())
|
|
|
|
if not filename:
|
|
filename = join(config.get('outdir'), '%s.%s.%s.%s' % (data['lastSystem']['name'].strip(), data['lastStarport']['name'].strip(), time.strftime('%Y-%m-%dT%H.%M.%S', time.localtime(querytime)), kind==COMMODITY_BPC and 'bpc' or 'csv'))
|
|
|
|
if kind == COMMODITY_CSV:
|
|
sep = ';'
|
|
header = sep.join(['System','Station','Commodity','Sell','Buy','Demand','','Supply','','Date','\n'])
|
|
rowheader = sep.join([data['lastSystem']['name'], data['lastStarport']['name']])
|
|
elif kind == COMMODITY_BPC:
|
|
sep = ';'
|
|
header = sep.join(['userID', 'System','Station','Commodity','Sell','Buy','Demand','','Supply','','Date\n'])
|
|
cmdr = data['commander']['name'].strip()
|
|
rowheader = sep.join([(config.getint('anonymous') and hashlib.md5(cmdr.encode('utf-8')).hexdigest()) or (sep in cmdr and '"%s"' % cmdr) or cmdr, 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']])
|
|
|
|
h = open(filename, 'wt') # 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)
|
|
|
|
h.close()
|