Reordered scripts folder

This commit is contained in:
Pax1601
2023-08-15 15:02:49 +02:00
parent e4b34b1dbc
commit 9db07b2ff8
13 changed files with 2830 additions and 2831 deletions

View File

@@ -0,0 +1,75 @@
from slpp import slpp as lua
import os
import json
import logging
SEARCH_FOLDER = "D:\\Eagle Dynamics\\DCS World OpenBeta"
def dump_lua(data):
if type(data) is str:
return f'"{data}"'
if type(data) in (int, float):
return f'{data}'
if type(data) is bool:
return data and "true" or "false"
if type(data) is list:
l = "{"
l += ",\n ".join([dump_lua(item) for item in data])
l += "}"
return l
if type(data) is dict:
t = "{"
t += ",\n ".join([f'[{k}] = {dump_lua(v)}' if type(k) == int else f'["{k}"]={dump_lua(v)}' for k,v in data.items()])
t += "}"
return t
logging.error(f"Unknown type {type(data)}")
filenames = [os.path.join(dp, f) for dp, dn, filenames in os.walk(SEARCH_FOLDER) for f in filenames if os.path.splitext(f)[1] == '.lua']
for filename in list(filenames):
with open(filename, 'r') as f:
try:
if f.read().find("unitPayloads") == -1:
filenames.remove(filename)
except:
filenames.remove(filename)
logging.warning(f"{filename} skipped...")
pass
names = {}
payloads = {}
for filename in filenames:
with open(os.path.join(os.getcwd(), filename), 'r') as f: # open in readonly mode
try:
lines = f.read()
searchString = "local unitPayloads = {"
start = lines.find(searchString)
end = lines.rfind("}")
tmp = lua.decode(lines[start + len(searchString) - 1: end + 1])
if type(tmp['payloads']) == dict:
src = tmp['payloads'].values()
else:
src = tmp['payloads']
names[tmp['unitType']] = []
payloads[tmp['unitType']] = {}
for payload in src:
names[tmp['unitType']].append(payload['name'])
if type(payload['pylons']) == dict:
payloads[tmp['unitType']][payload['name']] = {payload['pylons'][key]['num']: {"CLSID" : payload['pylons'][key]['CLSID']} for key in payload['pylons']}
else:
payloads[tmp['unitType']][payload['name']] = {payload['pylons'][key]['num']: {"CLSID" : payload['pylons'][key]['CLSID']} for key in range(len(payload['pylons']))}
except:
pass
with open('payloadNames.js', 'w') as f:
f.write("payloadNames = ")
json.dump(names, f, ensure_ascii = False)
with open('unitPayloads.lua', 'w') as f:
f.write("Olympus.unitPayloads = " + dump_lua(payloads))

View File

@@ -0,0 +1,61 @@
import pandas as pd
import json
# Load data from an Excel file
df = pd.read_excel('data.xlsx')
# Group by 'Name', 'Fuel', 'Loadout Name', 'Role', and 'Code' and aggregate 'Items - Name' and 'Items - Quantity'
grouped = df.groupby(['Name', 'Fuel', 'Loadout Name', 'Role', 'Code'])['Items - Name', 'Items - Quantity'].agg(lambda x: list(x)).reset_index()
# Convert the grouped data into the desired format
result = {}
for index, row in grouped.iterrows():
name = row['Name']
if name not in result:
result[name] = {
"name": row['Name'],
"label": row['Name'],
"loadouts": [
{
"fuel": row['Fuel'],
"items": [
{
"name": item,
"quantity": quantity
} for item, quantity in zip(row['Items - Name'], row['Items - Quantity'])
],
"roles": [row['Role']],
"code": row['Code'],
"loadout_name": row['Loadout Name']
}
]
}
else:
found = False
for loadout in result[name]["loadouts"]:
if loadout["fuel"] == row['Fuel'] and loadout["code"] == row['Code'] and loadout["loadout_name"] == row['Loadout Name']:
loadout["items"].extend([
{
"name": item,
"quantity": quantity
} for item, quantity in zip(row['Items - Name'], row['Items - Quantity'])
])
loadout["roles"].append(row['Role'])
found = True
break
if not found:
result[name]["loadouts"].append({
"fuel": row['Fuel'],
"items": [
{
"name": item,
"quantity": quantity
} for item, quantity in zip(row['Items - Name'], row['Items - Quantity'])
],
"roles": [row['Role']],
"code": row['Code'],
"loadout_name": row['Loadout Name']
})
# Print the result with the correct indents
print(json.dumps(result, indent=2))