Fix inara updater to use the included EDMarketConnector submodule.
This commit is contained in:
parent
8f9a78ad7e
commit
d0147e0dbb
273
companion.py
273
companion.py
|
|
@ -1,273 +0,0 @@
|
|||
import requests
|
||||
from collections import defaultdict
|
||||
from cookielib import LWPCookieJar
|
||||
import numbers
|
||||
import os
|
||||
from os.path import dirname, join
|
||||
import sys
|
||||
from sys import platform
|
||||
import time
|
||||
|
||||
if __debug__:
|
||||
from traceback import print_exc
|
||||
|
||||
from config import config
|
||||
|
||||
holdoff = 60 # be nice
|
||||
timeout = 10 # requests timeout
|
||||
|
||||
URL_LOGIN = 'https://companion.orerve.net/user/login'
|
||||
URL_CONFIRM = 'https://companion.orerve.net/user/confirm'
|
||||
URL_QUERY = 'https://companion.orerve.net/profile'
|
||||
|
||||
|
||||
# Map values reported by the Companion interface to names displayed in-game
|
||||
|
||||
category_map = {
|
||||
'Narcotics' : 'Legal Drugs',
|
||||
'Slaves' : 'Slavery',
|
||||
'Waste ' : 'Waste',
|
||||
'NonMarketable' : False, # Don't report these
|
||||
}
|
||||
|
||||
commodity_map= {
|
||||
'Agricultural Medicines' : 'Agri-Medicines',
|
||||
'Ai Relics' : 'AI Relics',
|
||||
'Atmospheric Extractors' : 'Atmospheric Processors',
|
||||
'Auto Fabricators' : 'Auto-Fabricators',
|
||||
'Basic Narcotics' : 'Narcotics',
|
||||
'Bio Reducing Lichen' : 'Bioreducing Lichen',
|
||||
'Hazardous Environment Suits' : 'H.E. Suits',
|
||||
'Heliostatic Furnaces' : 'Microbial Furnaces',
|
||||
'Marine Supplies' : 'Marine Equipment',
|
||||
'Non Lethal Weapons' : 'Non-Lethal Weapons',
|
||||
'S A P8 Core Container' : 'SAP 8 Core Container',
|
||||
'Terrain Enrichment Systems' : 'Land Enrichment Systems',
|
||||
}
|
||||
|
||||
ship_map = {
|
||||
'adder' : 'Adder',
|
||||
'anaconda' : 'Anaconda',
|
||||
'asp' : 'Asp Explorer',
|
||||
'cobramkiii' : 'Cobra MkIII',
|
||||
'diamondback' : 'Diamondback Scout',
|
||||
'diamondbackxl' : 'Diamondback Explorer',
|
||||
'eagle' : 'Eagle',
|
||||
'empire_courier' : 'Imperial Courier',
|
||||
'empire_eagle' : 'Imperial Eagle',
|
||||
'empire_fighter' : 'Imperial Fighter',
|
||||
'empire_trader' : 'Imperial Clipper',
|
||||
'federation_dropship' : 'Federal Dropship',
|
||||
'federation_dropship_mkii' : 'Federal Assault Ship',
|
||||
'federation_gunship' : 'Federal Gunship',
|
||||
'federation_fighter' : 'F63 Condor',
|
||||
'ferdelance' : 'Fer-de-Lance',
|
||||
'hauler' : 'Hauler',
|
||||
'orca' : 'Orca',
|
||||
'python' : 'Python',
|
||||
'sidewinder' : 'Sidewinder',
|
||||
'type6' : 'Type-6 Transporter',
|
||||
'type7' : 'Type-7 Transporter',
|
||||
'type9' : 'Type-9 Heavy',
|
||||
'viper' : 'Viper MkIII',
|
||||
'vulture' : 'Vulture',
|
||||
}
|
||||
|
||||
|
||||
# Companion API sometimes returns an array as a json array, sometimes as a json object indexed by "int".
|
||||
# This seems to depend on whether the there are 'gaps' in the Cmdr's data - i.e. whether the array is sparse.
|
||||
# In practice these arrays aren't very sparse so just convert them to lists with any 'gaps' holding None.
|
||||
def listify(thing):
|
||||
if thing is None:
|
||||
return [] # data is not present
|
||||
elif isinstance(thing, list):
|
||||
return thing # array is not sparse
|
||||
elif isinstance(thing, dict):
|
||||
retval = []
|
||||
for k,v in thing.iteritems():
|
||||
idx = int(k)
|
||||
if idx >= len(retval):
|
||||
retval.extend([None] * (idx - len(retval)))
|
||||
retval.append(v)
|
||||
else:
|
||||
retval[idx] = v
|
||||
return retval
|
||||
else:
|
||||
assert False, thing # we expect an array or a sparse array
|
||||
return list(thing) # hope for the best
|
||||
|
||||
|
||||
class ServerError(Exception):
|
||||
def __unicode__(self):
|
||||
return _('Error: Server is down')
|
||||
def __str__(self):
|
||||
return unicode(self).encode('utf-8')
|
||||
|
||||
class CredentialsError(Exception):
|
||||
def __unicode__(self):
|
||||
return _('Error: Invalid Credentials')
|
||||
def __str__(self):
|
||||
return unicode(self).encode('utf-8')
|
||||
|
||||
class VerificationRequired(Exception):
|
||||
pass
|
||||
|
||||
# Server companion.orerve.net uses a session cookie ("CompanionApp") to tie together login, verification
|
||||
# and query. So route all requests through a single Session object which holds this state.
|
||||
|
||||
class Session:
|
||||
|
||||
STATE_NONE, STATE_INIT, STATE_AUTH, STATE_OK = range(4)
|
||||
|
||||
def __init__(self):
|
||||
self.state = Session.STATE_INIT
|
||||
self.credentials = None
|
||||
|
||||
# yuck suppress InsecurePlatformWarning
|
||||
try:
|
||||
from requests.packages import urllib3
|
||||
urllib3.disable_warnings()
|
||||
except:
|
||||
pass
|
||||
|
||||
if platform=='win32' and getattr(sys, 'frozen', False):
|
||||
os.environ['REQUESTS_CA_BUNDLE'] = join(dirname(sys.executable), 'cacert.pem')
|
||||
|
||||
self.session = requests.Session()
|
||||
self.session.headers['User-Agent'] = 'Mozilla/5.0 (iPhone; CPU iPhone OS 7_1_2 like Mac OS X) AppleWebKit/537.51.2 (KHTML, like Gecko) Mobile/11D257'
|
||||
self.session.cookies = LWPCookieJar(join(config.app_dir, 'cookies.txt'))
|
||||
try:
|
||||
self.session.cookies.load()
|
||||
except IOError:
|
||||
pass
|
||||
|
||||
def login(self, username=None, password=None):
|
||||
self.state = Session.STATE_INIT
|
||||
if (not username or not password):
|
||||
if not self.credentials:
|
||||
raise CredentialsError()
|
||||
else:
|
||||
self.credentials = { 'email' : username, 'password' : password }
|
||||
try:
|
||||
r = self.session.post(URL_LOGIN, data = self.credentials, timeout=timeout)
|
||||
except:
|
||||
if __debug__: print_exc()
|
||||
raise ServerError()
|
||||
|
||||
if r.status_code != requests.codes.ok:
|
||||
self.dump(r)
|
||||
r.raise_for_status()
|
||||
|
||||
if 'server error' in r.text:
|
||||
self.dump(r)
|
||||
raise ServerError()
|
||||
elif 'Password' in r.text:
|
||||
self.dump(r)
|
||||
raise CredentialsError()
|
||||
elif 'Verification Code' in r.text:
|
||||
self.state = Session.STATE_AUTH
|
||||
raise VerificationRequired()
|
||||
else:
|
||||
self.state = Session.STATE_OK
|
||||
return r.status_code
|
||||
|
||||
def verify(self, code):
|
||||
if not code:
|
||||
raise VerificationRequired()
|
||||
r = self.session.post(URL_CONFIRM, data = {'code' : code}, timeout=timeout)
|
||||
r.raise_for_status()
|
||||
# verification doesn't actually return a yes/no, so log in again to determine state
|
||||
try:
|
||||
self.login()
|
||||
except:
|
||||
pass
|
||||
|
||||
def query(self):
|
||||
if self.state == Session.STATE_NONE:
|
||||
raise Exception('General error') # Shouldn't happen - don't bother localizing
|
||||
elif self.state == Session.STATE_INIT:
|
||||
self.login()
|
||||
elif self.state == Session.STATE_AUTH:
|
||||
raise VerificationRequired()
|
||||
try:
|
||||
r = self.session.get(URL_QUERY, timeout=timeout)
|
||||
except:
|
||||
if __debug__: print_exc()
|
||||
raise ServerError()
|
||||
|
||||
if r.status_code != requests.codes.ok:
|
||||
self.dump(r)
|
||||
if r.status_code == requests.codes.forbidden or (r.history and r.url == URL_LOGIN):
|
||||
# Start again - maybe our session cookie expired?
|
||||
self.login()
|
||||
return self.query()
|
||||
|
||||
r.raise_for_status()
|
||||
try:
|
||||
data = r.json()
|
||||
except:
|
||||
self.dump(r)
|
||||
raise ServerError()
|
||||
|
||||
return data
|
||||
|
||||
def close(self):
|
||||
self.state = Session.STATE_NONE
|
||||
try:
|
||||
self.session.cookies.save()
|
||||
self.session.close()
|
||||
except:
|
||||
pass
|
||||
self.session = None
|
||||
|
||||
# Fixup in-place anomalies in the recieved commodity data
|
||||
def fixup(self, commodities):
|
||||
i=0
|
||||
while i<len(commodities):
|
||||
commodity = commodities[i]
|
||||
|
||||
# Check all required numeric fields are present and are numeric
|
||||
# Catches "demandBracket": "" for some phantom commodites in ED 1.3 - https://github.com/Marginal/EDMarketConnector/issues/2
|
||||
# But also see https://github.com/Marginal/EDMarketConnector/issues/32
|
||||
for thing in ['buyPrice', 'sellPrice', 'demand', 'demandBracket', 'stock', 'stockBracket']:
|
||||
if not isinstance(commodity.get(thing), numbers.Number):
|
||||
if __debug__: print 'Invalid "%s":"%s" (%s) for "%s"' % (thing, commodity.get(thing), type(commodity.get(thing)), commodity.get('name', ''))
|
||||
break
|
||||
else:
|
||||
if not category_map.get(commodity['categoryname'], True): # Check marketable
|
||||
pass
|
||||
elif not commodity.get('categoryname'):
|
||||
if __debug__: print 'Missing "categoryname" for "%s"' % commodity.get('name', '')
|
||||
elif not commodity.get('name'):
|
||||
if __debug__: print 'Missing "name" for a commodity in "%s"' % commodity.get('categoryname', '')
|
||||
elif not commodity['demandBracket'] in range(4):
|
||||
if __debug__: print 'Invalid "demandBracket":"%s" for "%s"' % (commodity['demandBracket'], commodity['name'])
|
||||
elif not commodity['stockBracket'] in range(4):
|
||||
if __debug__: print 'Invalid "stockBracket":"%s" for "%s"' % (commodity['stockBracket'], commodity['name'])
|
||||
else:
|
||||
# Rewrite text fields
|
||||
commodity['categoryname'] = category_map.get(commodity['categoryname'], commodity['categoryname'])
|
||||
commodity['name'] = commodity_map.get(commodity['name'], commodity['name'])
|
||||
|
||||
# Force demand and stock to zero if their corresponding bracket is zero
|
||||
# Fixes spurious "demand": 1 in ED 1.3
|
||||
if not commodity['demandBracket']:
|
||||
commodity['demand'] = 0
|
||||
if not commodity['stockBracket']:
|
||||
commodity['stock'] = 0
|
||||
|
||||
# We're good
|
||||
i+=1
|
||||
continue
|
||||
|
||||
# Skip the commodity
|
||||
commodities.pop(i)
|
||||
|
||||
return commodities
|
||||
|
||||
def dump(self, r):
|
||||
if __debug__:
|
||||
print 'Status\t%s' % r.status_code
|
||||
print 'URL\t%s' % r.url
|
||||
print 'Headers\t%s' % r.headers
|
||||
print ('Content:\n%s' % r.text).encode('utf-8')
|
||||
|
|
@ -1,6 +1,6 @@
|
|||
#!/usr/bin/python
|
||||
|
||||
import companion
|
||||
from EDMarketConnector import companion
|
||||
import inara
|
||||
import utils
|
||||
|
||||
|
|
|
|||
Reference in New Issue
Block a user