mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-13 15:57:14 +03:00
ad8c63c64c
Also refactors data.py to edmc_data.py as I'm having weird issues with PyCharm debugger not starting, and this seems to be to do with module name clashes.
219 lines
11 KiB
Python
219 lines
11 KiB
Python
import pickle
|
|
import time
|
|
from collections import OrderedDict
|
|
from os.path import join
|
|
|
|
from config import config
|
|
from edmc_data import outfitting_armour_map as armour_map
|
|
from edmc_data import outfitting_cabin_map as cabin_map
|
|
from edmc_data import outfitting_corrosion_rating_map as corrosion_rating_map
|
|
from edmc_data import outfitting_countermeasure_map as countermeasure_map
|
|
from edmc_data import outfitting_fighter_rating_map as fighter_rating_map
|
|
from edmc_data import outfitting_internal_map as internal_map
|
|
from edmc_data import outfitting_misc_internal_map as misc_internal_map
|
|
from edmc_data import outfitting_missiletype_map as missiletype_map
|
|
from edmc_data import outfitting_planet_rating_map as planet_rating_map
|
|
from edmc_data import outfitting_rating_map as rating_map
|
|
from edmc_data import outfitting_standard_map as standard_map
|
|
from edmc_data import outfitting_utility_map as utility_map
|
|
from edmc_data import outfitting_weapon_map as weapon_map
|
|
from edmc_data import outfitting_weaponclass_map as weaponclass_map
|
|
from edmc_data import outfitting_weaponmount_map as weaponmount_map
|
|
from edmc_data import outfitting_weaponoldvariant_map as weaponoldvariant_map
|
|
from edmc_data import outfitting_weaponrating_map as weaponrating_map
|
|
from edmc_data import ship_name_map
|
|
|
|
# Module mass, FSD data etc
|
|
moduledata = OrderedDict()
|
|
|
|
# Given a module description from the Companion API returns a description of the module in the form of a
|
|
# dict { category, name, [mount], [guidance], [ship], rating, class } using the same terms found in the
|
|
# English langauge game. For fitted modules, dict also includes { enabled, priority }.
|
|
# ship_name_map tells us what ship names to use for Armour - i.e. EDDN schema names or in-game names.
|
|
#
|
|
# Returns None if the module is user-specific (i.e. decal, paintjob, kit) or PP-specific in station outfitting.
|
|
# (Given the ad-hocery in this implementation a big lookup table might have been simpler and clearer).
|
|
def lookup(module, ship_map, entitled=False):
|
|
|
|
# Lazily populate
|
|
if not moduledata:
|
|
moduledata.update(pickle.load(open(join(config.respath_path, 'modules.p'), 'rb')))
|
|
|
|
# if not module.get('category'): raise AssertionError('%s: Missing category' % module['id']) # only present post 1.3, and not present in ship loadout
|
|
if not module.get('name'): raise AssertionError('%s: Missing name' % module['id'])
|
|
|
|
name = module['name'].lower().split('_')
|
|
new = { 'id': module['id'], 'symbol': module['name'] }
|
|
|
|
# Armour - e.g. Federation_Dropship_Armour_Grade2
|
|
if name[-2] == 'armour':
|
|
name = module['name'].lower().rsplit('_', 2) # Armour is ship-specific, and ship names can have underscores
|
|
new['category'] = 'standard'
|
|
new['name'] = armour_map[name[2]]
|
|
new['ship'] = ship_map[name[0]] # Generate error on unknown ship
|
|
new['class'] = '1'
|
|
new['rating'] = 'I'
|
|
|
|
# Skip uninteresting stuff - some no longer present in ED 3.1 cAPI data
|
|
elif name[0] in ['bobble', 'decal', 'nameplate', 'paintjob', 'enginecustomisation', 'voicepack', 'weaponcustomisation'] or name[1].startswith('shipkit') :
|
|
return None
|
|
|
|
# Shouldn't be listing player-specific paid stuff or broker/powerplay-specific modules in outfitting, other than Horizons
|
|
elif not entitled and module.get('sku') and module['sku'] != 'ELITE_HORIZONS_V_PLANETARY_LANDINGS':
|
|
return None
|
|
|
|
# Don't report Planetary Approach Suite in outfitting
|
|
elif not entitled and name[1] == 'planetapproachsuite':
|
|
return None
|
|
|
|
# Countermeasures - e.g. Hpt_PlasmaPointDefence_Turret_Tiny
|
|
elif name[0]=='hpt' and name[1] in countermeasure_map:
|
|
new['category'] = 'utility'
|
|
new['name'], new['rating'] = countermeasure_map[len(name)>4 and (name[1],name[4]) or name[1]]
|
|
new['class'] = weaponclass_map[name[-1]]
|
|
|
|
# Utility - e.g. Hpt_CargoScanner_Size0_Class1
|
|
elif name[0]=='hpt' and name[1] in utility_map:
|
|
new['category'] = 'utility'
|
|
new['name'] = utility_map[len(name)>4 and (name[1],name[4]) or name[1]]
|
|
if not name[2].startswith('size') or not name[3].startswith('class'): raise AssertionError('%s: Unknown class/rating "%s/%s"' % (module['id'], name[2], name[3]))
|
|
new['class'] = str(name[2][4:])
|
|
new['rating'] = rating_map[name[3][5:]]
|
|
|
|
# Hardpoints - e.g. Hpt_Slugshot_Fixed_Medium
|
|
elif name[0]=='hpt':
|
|
# Hack 'Guardian' and 'Mining' prefixes
|
|
if len(name) > 3 and name[3] in weaponmount_map:
|
|
prefix = name.pop(1)
|
|
name[1] = '%s_%s' % (prefix, name[1])
|
|
if name[1] not in weapon_map: raise AssertionError('%s: Unknown weapon "%s"' % (module['id'], name[0]))
|
|
if name[2] not in weaponmount_map: raise AssertionError('%s: Unknown weapon mount "%s"' % (module['id'], name[2]))
|
|
if name[3] not in weaponclass_map: raise AssertionError('%s: Unknown weapon class "%s"' % (module['id'], name[3]))
|
|
|
|
new['category'] = 'hardpoint'
|
|
if len(name)>4:
|
|
if name[4] in weaponoldvariant_map: # Old variants e.g. Hpt_PulseLaserBurst_Turret_Large_OC
|
|
new['name'] = weapon_map[name[1]] + ' ' + weaponoldvariant_map[name[4]]
|
|
new['rating'] = '?'
|
|
elif '_'.join(name[:4]) not in weaponrating_map:
|
|
raise AssertionError('%s: Unknown weapon rating "%s"' % (module['id'], module['name']))
|
|
else: # PP faction-specific weapons e.g. Hpt_Slugshot_Fixed_Large_Range
|
|
new['name'] = weapon_map[(name[1],name[4])]
|
|
new['rating'] = weaponrating_map['_'.join(name[:4])] # assumes same rating as base weapon
|
|
elif module['name'].lower() not in weaponrating_map:
|
|
raise AssertionError('%s: Unknown weapon rating "%s"' % (module['id'], module['name']))
|
|
else:
|
|
new['name'] = weapon_map[name[1]]
|
|
new['rating'] = weaponrating_map[module['name'].lower()] # no obvious rule - needs lookup table
|
|
new['mount'] = weaponmount_map[name[2]]
|
|
if name[1] in missiletype_map: # e.g. Hpt_DumbfireMissileRack_Fixed_Small
|
|
new['guidance'] = missiletype_map[name[1]]
|
|
new['class'] = weaponclass_map[name[3]]
|
|
|
|
elif name[0]!='int':
|
|
raise AssertionError('%s: Unknown prefix "%s"' % (module['id'], name[0]))
|
|
|
|
# Miscellaneous Class 1 - e.g. Int_PlanetApproachSuite, Int_StellarBodyDiscoveryScanner_Advanced, Int_DockingComputer_Standard
|
|
elif name[1] in misc_internal_map:
|
|
new['category'] = 'internal'
|
|
new['name'], new['rating'] = misc_internal_map[name[1]]
|
|
new['class'] = '1'
|
|
elif len(name) > 2 and (name[1],name[2]) in misc_internal_map:
|
|
# Reported category is not necessarily helpful. e.g. "Int_DockingComputer_Standard" has category "utility"
|
|
new['category'] = 'internal'
|
|
new['name'], new['rating'] = misc_internal_map[(name[1],name[2])]
|
|
new['class'] = '1'
|
|
|
|
# Standard & Internal
|
|
else:
|
|
if name[1] == 'dronecontrol': # e.g. Int_DroneControl_Collection_Size1_Class1
|
|
name.pop(0)
|
|
elif name[-1] == 'free': # Starter Sidewinder or Freagle modules - just treat them like vanilla modules
|
|
name.pop()
|
|
|
|
if name[1] in standard_map: # e.g. Int_Engine_Size2_Class1, Int_ShieldGenerator_Size8_Class5_Strong
|
|
new['category'] = 'standard'
|
|
new['name'] = standard_map[len(name)>4 and (name[1],name[4]) or name[1]]
|
|
elif name[1] in internal_map: # e.g. Int_CargoRack_Size8_Class1
|
|
new['category'] = 'internal'
|
|
if name[1] == 'passengercabin':
|
|
new['name'] = cabin_map[name[3][5:]]
|
|
else:
|
|
new['name'] = internal_map[len(name)>4 and (name[1],name[4]) or name[1]]
|
|
else:
|
|
raise AssertionError('%s: Unknown module "%s"' % (module['id'], name[1]))
|
|
|
|
if len(name) < 4 and name[1] == 'unkvesselresearch': # Hack! No size or class.
|
|
(new['class'], new['rating']) = ('1', 'E')
|
|
elif len(name) < 4 and name[1] == 'resourcesiphon': # Hack! 128066402 has no size or class.
|
|
(new['class'], new['rating']) = ('1', 'I')
|
|
elif len(name) < 4 and name[1] in ['guardianpowerdistributor', 'guardianpowerplant']: # Hack! No class.
|
|
(new['class'], new['rating']) = (str(name[2][4:]), 'A')
|
|
elif len(name) < 4 and name[1] in ['guardianfsdbooster']: # Hack! No class.
|
|
(new['class'], new['rating']) = (str(name[2][4:]), 'H')
|
|
else:
|
|
if len(name) < 3: raise AssertionError('%s: length < 3]' % (name))
|
|
if not name[2].startswith('size') or not name[3].startswith('class'): raise AssertionError('%s: Unknown class/rating "%s/%s"' % (module['id'], name[2], name[3]))
|
|
new['class'] = str(name[2][4:])
|
|
new['rating'] = (name[1]=='buggybay' and planet_rating_map or
|
|
name[1]=='fighterbay' and fighter_rating_map or
|
|
name[1]=='corrosionproofcargorack' and corrosion_rating_map or
|
|
rating_map)[name[3][5:]]
|
|
|
|
# Disposition of fitted modules
|
|
if 'on' in module and 'priority' in module:
|
|
new['enabled'], new['priority'] = module['on'], module['priority'] # priority is zero-based
|
|
|
|
# Entitlements
|
|
if not module.get('sku'):
|
|
pass
|
|
else:
|
|
new['entitlement'] = module['sku']
|
|
|
|
# Extra module data
|
|
if module['name'].endswith('_free'):
|
|
key = module['name'][:-5].lower() # starter modules - treated like vanilla modules
|
|
else:
|
|
key = module['name'].lower()
|
|
if __debug__:
|
|
m = moduledata.get(key, {})
|
|
if not m:
|
|
print('No data for module %s' % key)
|
|
elif new['name'] == 'Frame Shift Drive':
|
|
assert 'mass' in m and 'optmass' in m and 'maxfuel' in m and 'fuelmul' in m and 'fuelpower' in m, m
|
|
else:
|
|
assert 'mass' in m, m
|
|
new.update(moduledata.get(module['name'].lower(), {}))
|
|
|
|
# check we've filled out mandatory fields
|
|
for thing in ['id', 'symbol', 'category', 'name', 'class', 'rating']: # Don't consider mass etc as mandatory
|
|
if not new.get(thing): raise AssertionError('%s: failed to set %s' % (module['id'], thing))
|
|
if new['category'] == 'hardpoint' and not new.get('mount'):
|
|
raise AssertionError('%s: failed to set %s' % (module['id'], 'mount'))
|
|
|
|
return new
|
|
|
|
|
|
def export(data, filename):
|
|
|
|
querytime = config.get_int('querytime', default=int(time.time()))
|
|
|
|
assert data['lastSystem'].get('name')
|
|
assert data['lastStarport'].get('name')
|
|
|
|
header = 'System,Station,Category,Name,Mount,Guidance,Ship,Class,Rating,FDevID,Date\n'
|
|
rowheader = '%s,%s' % (data['lastSystem']['name'], data['lastStarport']['name'])
|
|
|
|
h = open(filename, 'wt')
|
|
h.write(header)
|
|
for v in list(data['lastStarport'].get('modules', {}).values()):
|
|
try:
|
|
m = lookup(v, ship_name_map)
|
|
if m:
|
|
h.write('%s,%s,%s,%s,%s,%s,%s,%s,%s,%s\n' % (rowheader, m['category'], m['name'], m.get('mount',''), m.get('guidance',''), m.get('ship',''), m['class'], m['rating'], m['id'], data['timestamp']))
|
|
except AssertionError as e:
|
|
if __debug__: print('Outfitting: %s' % e) # Silently skip unrecognized modules
|
|
except:
|
|
if __debug__: raise
|
|
h.close()
|