DCSOlympus/scripts/python/importDatabases.py
2023-10-11 18:01:27 +02:00

60 lines
1.7 KiB
Python

import csv
import sys
import json
# Read CSV file
with open("data.csv") as fp:
reader = csv.reader(fp, delimiter=",", quotechar='"')
data_read = [row for row in reader]
base = '..\\..\\client\\public\\databases\\units\\'
filenames = [f'{base}aircraftdatabase.json', f'{base}helicopterdatabase.json', f'{base}groundunitdatabase.json', f'{base}navyunitdatabase.json']
lines = []
for filename in filenames:
# Loads the database
with open(filename, encoding="utf8") as f:
database = json.load(f)
for row in data_read:
if row[0] in database:
name = row[0]
if row[1] == "yes":
database[name]["enabled"] = True
else:
database[name]["enabled"] = False
database[name]["type"] = row[3]
database[name]["label"] = row[4]
database[name]["shortLabel"] = row[5]
database[name]["coalition"] = row[6]
database[name]["era"] = row[7]
if row[8] == "yes":
database[name]["canTargetPoint"] = True
else:
database[name]["canTargetPoint"] = False
if row[9] == "yes":
database[name]["canRearm"] = True
else:
database[name]["canRearm"] = False
if row[10] != "":
database[name]["acquisitionRange"] = int(row[10])
if row[11] != "":
database[name]["engagementRange"] = int(row[11])
database[name]["description"] = row[12]
database[name]["abilities"] = row[13]
# Dump everything in the database
with open(filename, "w") as f:
json.dump(database, f, indent=2)
# Done!
print("Done!")