elia/dsp-calc
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

get tree json outpout working

Browse Source
This commit is contained in:
Schemen
2021-01-27 15:53:25 +01:00
parent c8b4aaa743
commit 881fd2cd22
3 changed files with 20 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.vscode/launch.json vendored
View File

@@ -10,7 +10,7 @@
"request": "launch", "request": "launch",
"program": "${file}", "program": "${file}",
"console": "integratedTerminal", "console": "integratedTerminal",
"args": ["tree", "hydrogen"] "args": ["tree", "gear"]
} }
] ]
} }

8
data copy.json → data.json
View File

@@ -12,6 +12,14 @@
"produced_with": "water_pump", "produced_with": "water_pump",
"requires": {} "requires": {}
}, },
"plant_fuel":{
"produced_with": "manual_labor",
"requires": {}
},
"log":{
"produced_with": "manual_labor",
"requires": {}
},
"fire_ice":{ "fire_ice":{
"produced_with": "orbital_harvester", "produced_with": "orbital_harvester",
"requires": {} "requires": {}

36
dsp_calc.py
View File

@@ -4,7 +4,7 @@ import json
import pprint import pprint
with open('data copy.json', 'r') as DATA_JSON: with open('data.json', 'r') as DATA_JSON:
data=DATA_JSON.read() data=DATA_JSON.read()
try: try:
@@ -44,38 +44,32 @@ def tree(item):
def get_required(item_name): def get_required(item_name):
print("Item_name: "+ item_name) """Function that returns the entire requirements of an item"""
dsp_data_combined = get_combined_data(dsp_data) dsp_data_combined = get_combined_data(dsp_data)
required_items = {} required_items = {}
item = dsp_data_combined[item_name] item = dsp_data_combined[item_name]
print("item: " + str(item))
if not item.get("requires") and not item.get("recipes"): if "requires" not in item and "recipes" not in item:
print("this is and enditem 1")
return {} return {}
if item.get("recipes"): if item.get("recipes"):
print(item["recipes"]) required_items["recipes"] = {}
for i in item["recipes"]: for i in item["recipes"]:
if not item.get("requires") and not item.get("recipes"):
print("this is and enditem 2")
return {}
for req_name in i["requires"].items(): for req_name, amount in i["requires"].items():
sub_req_name = req_name[0] sub_req = get_required(req_name)
sub_req = get_required(sub_req_name) required_items["recipes"][req_name] = {**dsp_data_combined[req_name], "amount": amount, "requires": sub_req }
required_items[sub_req_name] = {**dsp_data_combined[sub_req_name], "recipes": [ sub_req ] }
return required_items
for req_name, amount in item["requires"].items(): for req_name, amount in item["requires"].items():
sub_req = get_required(req_name) sub_req = get_required(req_name)
required_items[req_name] = {**dsp_data_combined[req_name], "amount": amount, "requires": sub_req} required_items[req_name] = {**dsp_data_combined[req_name], "amount": amount, "requires": sub_req}
return required_items return required_items
def get_item_requirements(item):
item["children"] = {}
return item.get("requires")
def get_combined_data(data): def get_combined_data(data):
combined = data["components"] combined = data["components"]
@@ -83,14 +77,6 @@ def get_combined_data(data):
return combined return combined
def get_item_type(item):
if dsp_data["components"].get(item):
return "components"
elif dsp_data["buildings"].get(item):
return "buildings"
else:
click.echo("Item %s not found!" % item)
def get_item(item): def get_item(item):
if dsp_data["components"].get(item): if dsp_data["components"].get(item):
return dsp_data["components"].get(item) return dsp_data["components"].get(item)