104 lines
2.5 KiB
Python
Executable File
104 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import click
|
|
import json
|
|
import pprint
|
|
|
|
|
|
with open('data copy.json', 'r') as DATA_JSON:
|
|
data=DATA_JSON.read()
|
|
|
|
try:
|
|
dsp_data = json.loads(data)
|
|
except ValueError as e:
|
|
click.echo("Error: %s"% e)
|
|
|
|
|
|
|
|
@click.group()
|
|
def cli():
|
|
pass
|
|
|
|
@cli.command()
|
|
def verify():
|
|
"""verify your data.json"""
|
|
try:
|
|
dsp_data = json.loads(data)
|
|
except ValueError as e:
|
|
click.echo("Error: %s"% e)
|
|
else:
|
|
click.echo("Your data seems to be good.")
|
|
|
|
@cli.command()
|
|
@click.argument('item')
|
|
def tree(item):
|
|
"""
|
|
Show the requirement tree of an item.
|
|
|
|
Pass the item down lowercase, space replaced with a "_" underscore and MKII and mk3.
|
|
|
|
Example:
|
|
$dsp_calc tree sorter_mk3
|
|
|
|
"""
|
|
click.echo(json.dumps(get_required(item), indent=4))
|
|
|
|
|
|
|
|
def get_required(item_name):
|
|
print("Item_name: "+ item_name)
|
|
dsp_data_combined = get_combined_data(dsp_data)
|
|
required_items = {}
|
|
item = dsp_data_combined[item_name]
|
|
print("item: " + str(item))
|
|
|
|
if not item.get("requires") and not item.get("recipes"):
|
|
print("this is and enditem 1")
|
|
return {}
|
|
|
|
if item.get("recipes"):
|
|
print(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():
|
|
sub_req_name = req_name[0]
|
|
sub_req = get_required(sub_req_name)
|
|
required_items[sub_req_name] = {**dsp_data_combined[sub_req_name], "recipes": [ sub_req ] }
|
|
|
|
for req_name, amount in item["requires"].items():
|
|
sub_req = get_required(req_name)
|
|
required_items[req_name] = {**dsp_data_combined[req_name], "amount": amount, "requires": sub_req}
|
|
return required_items
|
|
|
|
def get_item_requirements(item):
|
|
item["children"] = {}
|
|
return item.get("requires")
|
|
|
|
|
|
def get_combined_data(data):
|
|
combined = data["components"]
|
|
combined.update(data["buildings"])
|
|
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):
|
|
if dsp_data["components"].get(item):
|
|
return dsp_data["components"].get(item)
|
|
elif dsp_data["buildings"].get(item):
|
|
return dsp_data["buildings"].get(item)
|
|
else:
|
|
click.echo("Item %s not found!" % item)
|
|
|
|
if __name__ == '__main__':
|
|
cli()
|