graph: allow multiple target arguments

This commit is contained in:
Danny Robson 2019-07-14 13:25:19 +10:00
parent 27743d9e6d
commit 25e57f272c

View File

@ -42,14 +42,14 @@ if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--data', type=str, default=recipe_root) parser.add_argument('--data', type=str, default=recipe_root)
parser.add_argument('--target', type=str, required=False) parser.add_argument('--target', type=str, nargs="*")
args = parser.parse_args() args = parser.parse_args()
recipes = satisfactory.Cookbook(args.data) recipes = satisfactory.Cookbook(args.data)
if args.target: if args.target:
graph_one(recipes, args.target) graph(recipes, args.target)
else: else:
graph(recipes, recipes.components()) graph(recipes, recipes.components())