mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-14 08:17:13 +03:00
Fix collation from JSON dump
This commit is contained in:
parent
d6bb4f90ca
commit
2f0468333a
19
EDMC.py
19
EDMC.py
@ -7,6 +7,7 @@ import argparse
|
||||
import json
|
||||
import requests
|
||||
import sys
|
||||
from os.path import getmtime
|
||||
from time import time, sleep
|
||||
from xml.etree import ElementTree
|
||||
|
||||
@ -69,14 +70,12 @@ try:
|
||||
print '%.2f%s' % (float(''.join(appversion.split('.')[:3])) / 100, latest) # just first three digits
|
||||
sys.exit(EXIT_SUCCESS)
|
||||
|
||||
session = companion.Session()
|
||||
if args.j:
|
||||
# Import and collate from JSON dump
|
||||
data = json.load(open(args.j))
|
||||
collate.addcommodities(data)
|
||||
collate.addmodules(data)
|
||||
collate.addships(data)
|
||||
config.set('querytime', getmtime(args.j))
|
||||
else:
|
||||
session = companion.Session()
|
||||
session.login(config.get('username'), config.get('password'))
|
||||
querytime = int(time())
|
||||
data = session.query()
|
||||
@ -93,6 +92,10 @@ try:
|
||||
sys.stderr.write('What are you flying?!\n') # Shouldn't happen
|
||||
sys.exit(EXIT_SERVER)
|
||||
|
||||
if data['lastStarport'].get('commodities'):
|
||||
# Fixup anomalies in the commodity data
|
||||
session.fixup(data['lastStarport']['commodities'])
|
||||
|
||||
# stuff we can do when not docked
|
||||
if args.d:
|
||||
with open(args.d, 'wt') as h:
|
||||
@ -122,10 +125,14 @@ try:
|
||||
sys.stderr.write("Station doesn't have anything!\n")
|
||||
sys.exit(EXIT_SUCCESS)
|
||||
|
||||
if args.j:
|
||||
# Collate from JSON dump
|
||||
collate.addcommodities(data)
|
||||
collate.addmodules(data)
|
||||
collate.addships(data)
|
||||
|
||||
if args.m:
|
||||
if data['lastStarport'].get('commodities'):
|
||||
# Fixup anomalies in the commodity data
|
||||
session.fixup(data['lastStarport']['commodities'])
|
||||
commodity.export(data, COMMODITY_DEFAULT, args.m)
|
||||
else:
|
||||
sys.stderr.write("Station doesn't have a market\n")
|
||||
|
Loading…
x
Reference in New Issue
Block a user