57 lines
1.2 KiB
Python
Executable File
57 lines
1.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import satisfactory
|
|
|
|
import os
|
|
|
|
from typing import Iterable
|
|
|
|
|
|
def graph(cookbook: satisfactory.Cookbook, targets: Iterable[str]):
|
|
print("digraph G {")
|
|
|
|
seen = set()
|
|
remain = set(targets)
|
|
|
|
while remain:
|
|
output = remain.pop()
|
|
|
|
for need in cookbook.recipes(output)[0]['input']:
|
|
print(f"{need} -> {output}")
|
|
if need not in seen:
|
|
remain.add(need)
|
|
|
|
seen.add(output)
|
|
print("}")
|
|
|
|
|
|
def graph_all(recipes: dict) -> None:
|
|
graph(recipes, recipes.all())
|
|
|
|
|
|
def graph_one(recipes: dict, target: str) -> None:
|
|
graph(recipes, [target])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
def main():
|
|
import argparse
|
|
|
|
root = os.path.dirname(__file__)
|
|
recipe_root = os.path.join(root, 'data', 'recipes')
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--data', type=str, default=recipe_root)
|
|
parser.add_argument('--target', type=str, required=False)
|
|
|
|
args = parser.parse_args()
|
|
|
|
recipes = satisfactory.Cookbook(args.data)
|
|
|
|
if args.target:
|
|
graph_one(recipes, args.target)
|
|
else:
|
|
graph(recipes, recipes.components())
|
|
|
|
main()
|