mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-14 08:17:13 +03:00
106 lines
4.6 KiB
Python
Executable File
106 lines
4.6 KiB
Python
Executable File
#!/usr/bin/python
|
|
#
|
|
# build ship and module databases from https://github.com/EDCD/coriolis-data/
|
|
#
|
|
|
|
import base64
|
|
from collections import OrderedDict
|
|
import cPickle
|
|
import json
|
|
|
|
from config import config
|
|
import outfitting
|
|
import companion
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
def add(modules, name, attributes):
|
|
assert name not in modules or modules[name] == attributes, '%s: %s!=%s' % (name, modules.get(name), attributes)
|
|
assert name not in modules, name
|
|
modules[name] = attributes
|
|
|
|
|
|
data = json.load(open('coriolis-data/dist/index.json'))
|
|
|
|
# Map Coriolis's names to names displayed in the in-game shipyard
|
|
coriolis_ship_map = {
|
|
'Cobra Mk III' : 'Cobra MkIII',
|
|
'Cobra Mk IV' : 'Cobra MkIV',
|
|
'Krait Mk II' : 'Krait MkII',
|
|
'Viper' : 'Viper MkIII',
|
|
'Viper Mk IV' : 'Viper MkIV',
|
|
}
|
|
|
|
# Symbolic name from in-game name
|
|
reverse_ship_map = {v: k for k, v in companion.ship_map.iteritems()}
|
|
|
|
bulkheads = outfitting.armour_map.keys()
|
|
|
|
ships = {}
|
|
modules = {}
|
|
|
|
# Ship and armour masses
|
|
for m in data['Ships'].values():
|
|
name = coriolis_ship_map.get(m['properties']['name'], str(m['properties']['name']))
|
|
assert name in reverse_ship_map, name
|
|
ships[name] = { 'hullMass' : m['properties']['hullMass'] }
|
|
for i in range(len(bulkheads)):
|
|
modules['_'.join([reverse_ship_map[name], 'armour', bulkheads[i]])] = { 'mass': m['bulkheads'][i]['mass'] }
|
|
|
|
ships = OrderedDict([(k,ships[k]) for k in sorted(ships)]) # sort for easier diffing
|
|
cPickle.dump(ships, open('ships.p', 'wb'))
|
|
|
|
# Module masses
|
|
for cat in data['Modules'].values():
|
|
for grp, mlist in cat.iteritems():
|
|
for m in mlist:
|
|
assert 'symbol' in m, m
|
|
key = str(m['symbol'].lower())
|
|
if grp == 'fsd':
|
|
modules[key] = {
|
|
'mass' : m['mass'],
|
|
'optmass' : m['optmass'],
|
|
'maxfuel' : m['maxfuel'],
|
|
'fuelmul' : m['fuelmul'],
|
|
'fuelpower' : m['fuelpower'],
|
|
}
|
|
elif grp == 'gfsb':
|
|
modules[key] = {
|
|
'mass' : m['mass'],
|
|
'jumpboost' : m['jumpboost'],
|
|
}
|
|
else:
|
|
modules[key] = { 'mass': m.get('mass', 0) } # Some modules don't have mass
|
|
|
|
# 3.0 / 3.1 additions not yet present in coriolis-data
|
|
add(modules, 'hpt_causticmissile_fixed_medium', {'mass': 4})
|
|
add(modules, 'hpt_flechettelauncher_fixed_medium', {'mass': 4})
|
|
add(modules, 'hpt_flechettelauncher_turret_medium', {'mass': 4})
|
|
add(modules, 'hpt_guardian_plasmalauncher_fixed_large', {'mass': 8})
|
|
add(modules, 'hpt_guardian_plasmalauncher_turret_large', {'mass': 8})
|
|
add(modules, 'hpt_guardian_shardcannon_turret_medium', {'mass': 4})
|
|
add(modules, 'hpt_guardian_shardcannon_turret_large', {'mass': 8})
|
|
add(modules, 'hpt_plasmashockcannon_fixed_medium', {'mass': 4})
|
|
add(modules, 'hpt_plasmashockcannon_gimbal_large', {'mass': 8}) # ???
|
|
add(modules, 'hpt_plasmashockcannon_turret_medium', {'mass': 4})
|
|
add(modules, 'hpt_plasmashockcannon_turret_large', {'mass': 8}) # ???
|
|
add(modules, 'int_dronecontrol_decontamination_size1_class1', {'mass': 1.3})
|
|
add(modules, 'int_dronecontrol_decontamination_size3_class1', {'mass': 2})
|
|
add(modules, 'int_dronecontrol_decontamination_size5_class1', {'mass': 20})
|
|
add(modules, 'int_dronecontrol_decontamination_size7_class1', {'mass': 128})
|
|
add(modules, 'int_dronecontrol_unkvesselresearch', {'mass': 1.3})
|
|
add(modules, 'int_metaalloyhullreinforcement_size1_class1', { 'mass': 2 })
|
|
add(modules, 'int_metaalloyhullreinforcement_size1_class2', { 'mass': 2 }) # anomaly
|
|
add(modules, 'int_metaalloyhullreinforcement_size2_class1', { 'mass': 4 })
|
|
add(modules, 'int_metaalloyhullreinforcement_size2_class2', { 'mass': 2 })
|
|
add(modules, 'int_metaalloyhullreinforcement_size3_class1', { 'mass': 8 })
|
|
add(modules, 'int_metaalloyhullreinforcement_size3_class2', { 'mass': 4 })
|
|
add(modules, 'int_metaalloyhullreinforcement_size4_class1', { 'mass': 16 })
|
|
add(modules, 'int_metaalloyhullreinforcement_size4_class2', { 'mass': 8 })
|
|
add(modules, 'int_metaalloyhullreinforcement_size5_class1', { 'mass': 32 })
|
|
add(modules, 'int_metaalloyhullreinforcement_size5_class2', { 'mass': 16 })
|
|
|
|
modules = OrderedDict([(k,modules[k]) for k in sorted(modules)]) # sort for easier diffing
|
|
cPickle.dump(modules, open('modules.p', 'wb'))
|