mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-16 01:00:42 +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.
93 lines
3.5 KiB
Python
Executable File
93 lines
3.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# build ship and module databases from https://github.com/EDCD/coriolis-data/
|
|
#
|
|
|
|
import csv
|
|
import json
|
|
import pickle
|
|
import subprocess
|
|
import sys
|
|
from collections import OrderedDict
|
|
from traceback import print_exc
|
|
|
|
import outfitting
|
|
from edmc_data import coriolis_ship_map, ship_name_map
|
|
|
|
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
|
|
|
|
# Regenerate coriolis-data distribution
|
|
subprocess.check_call('npm install', cwd='coriolis-data', shell=True, stdout=sys.stdout, stderr=sys.stderr)
|
|
|
|
data = json.load(open('coriolis-data/dist/index.json'))
|
|
|
|
# Symbolic name from in-game name
|
|
reverse_ship_map = {v: k for k, v in list(ship_name_map.items())}
|
|
|
|
bulkheads = list(outfitting.armour_map.keys())
|
|
|
|
ships = {}
|
|
modules = {}
|
|
|
|
# Ship and armour masses
|
|
for m in list(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
|
|
pickle.dump(ships, open('ships.p', 'wb'))
|
|
|
|
# Module masses
|
|
for cat in list(data['Modules'].values()):
|
|
for grp, mlist in list(cat.items()):
|
|
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
|
|
|
|
# Pre 3.3 modules
|
|
add(modules, 'int_stellarbodydiscoveryscanner_standard', { 'mass': 2 })
|
|
add(modules, 'int_stellarbodydiscoveryscanner_intermediate', { 'mass': 2 })
|
|
add(modules, 'int_stellarbodydiscoveryscanner_advanced', { 'mass': 2 })
|
|
|
|
# Missing
|
|
add(modules, 'hpt_dumbfiremissilerack_fixed_small_advanced', { 'mass': 2 })
|
|
add(modules, 'hpt_dumbfiremissilerack_fixed_medium_advanced',{ 'mass': 4 })
|
|
add(modules, 'hpt_multicannon_fixed_small_advanced', { 'mass': 2 })
|
|
add(modules, 'hpt_multicannon_fixed_medium_advanced', { 'mass': 4 })
|
|
|
|
modules = OrderedDict([(k,modules[k]) for k in sorted(modules)]) # sort for easier diffing
|
|
pickle.dump(modules, open('modules.p', 'wb'))
|
|
|
|
# Check data is present for all modules
|
|
with open('outfitting.csv') as csvfile:
|
|
reader = csv.DictReader(csvfile, restval='')
|
|
for row in reader:
|
|
try:
|
|
module = outfitting.lookup({ 'id': row['id'], 'name': row['symbol'] }, ship_name_map)
|
|
except:
|
|
print(row['symbol'])
|
|
print_exc()
|