Oh Boy...

dsp_calc.py

@@ -0,0 +1,103 @@
+#!/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()