mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-15 00:30:33 +03:00
d6bb4f90ca
"Ion Distributor" and "Trinkets Of Hidden Fortune" - see https://github.com/EDCD/FDevIDs/pull/6
106 lines
3.2 KiB
Python
106 lines
3.2 KiB
Python
# Export poor man's flight log
|
|
|
|
from collections import defaultdict
|
|
import errno
|
|
import os
|
|
from os.path import join
|
|
from sys import platform
|
|
import time
|
|
|
|
from config import config
|
|
from companion import ship_map
|
|
|
|
|
|
logfile = None
|
|
last_timestamp = last_system = last_ship = None
|
|
last_commodities = {}
|
|
|
|
def openlog():
|
|
|
|
global logfile
|
|
if logfile: return
|
|
|
|
try:
|
|
logfile = open(join(config.get('outdir'), 'Flight Log.csv'), 'a+b')
|
|
if platform != 'win32': # open for writing is automatically exclusive on Windows
|
|
from fcntl import lockf, LOCK_SH, LOCK_NB
|
|
lockf(logfile, LOCK_SH|LOCK_NB)
|
|
logfile.seek(0, os.SEEK_END)
|
|
if not logfile.tell():
|
|
logfile.write('Date,Time,System,Station,Ship,Cargo\r\n')
|
|
except EnvironmentError as e:
|
|
logfile = None
|
|
if e.errno in [errno.EACCES, errno.EAGAIN]:
|
|
raise Exception('Can\'t write "Flight Log.csv". Are you editing it in another app?')
|
|
else:
|
|
raise
|
|
except:
|
|
logfile = None
|
|
raise
|
|
|
|
|
|
def export(data):
|
|
|
|
querytime = config.getint('querytime') or int(time.time())
|
|
|
|
commodities = defaultdict(int)
|
|
for item in data['ship'].get('cargo',{}).get('items',[]):
|
|
if item['commodity'] != 'drones':
|
|
commodities[item['commodity']] += item['qty']
|
|
|
|
writelog(querytime,
|
|
data['lastSystem']['name'],
|
|
data['commander']['docked'] and data['lastStarport']['name'],
|
|
ship_map.get(data['ship']['name'].lower(), data['ship']['name']),
|
|
commodities)
|
|
|
|
|
|
def writelog(timestamp, system, station=None, ship=None, commodities={}):
|
|
|
|
global last_timestamp, last_system, last_ship, last_commodities
|
|
if last_system and last_system != system:
|
|
_writelog(last_timestamp, last_system, None, last_ship, last_commodities)
|
|
|
|
if not station:
|
|
# If not docked, hold off writing log entry until docked or until system changes
|
|
last_timestamp, last_system, last_ship, last_commodities = timestamp, system, ship, commodities
|
|
else:
|
|
last_system = None
|
|
_writelog(timestamp, system, station, ship, commodities)
|
|
|
|
|
|
def _writelog(timestamp, system, station=None, ship=None, commodities={}):
|
|
|
|
openlog()
|
|
|
|
logfile.write('%s,%s,%s,%s,%s,%s\r\n' % (
|
|
time.strftime('%Y-%m-%d', time.localtime(timestamp)),
|
|
time.strftime('%H:%M:%S', time.localtime(timestamp)),
|
|
system,
|
|
station or '',
|
|
ship or '',
|
|
','.join([('%d %s' % (commodities[k], k)) for k in sorted(commodities)])))
|
|
|
|
logfile.flush()
|
|
|
|
|
|
def close():
|
|
global last_timestamp, last_system
|
|
|
|
if last_system:
|
|
_writelog(last_timestamp, last_system, None, last_ship, last_commodities)
|
|
last_system = None
|
|
|
|
|
|
# return log as list of (timestamp, system_name)
|
|
def logs():
|
|
entries = []
|
|
with open(join(config.get('outdir'), 'Flight Log.csv'), 'rU') as f:
|
|
f.readline() # Assume header
|
|
for line in f:
|
|
if not line.strip(): continue
|
|
cols = line.split(',')
|
|
assert len(cols) >= 3, cols
|
|
entries.append((time.mktime(time.strptime('%sT%s' % (cols[0], cols[1]), '%Y-%m-%dT%H:%M:%S')), cols[2])) # Convert from local time to UTC
|
|
return entries
|