From 2ddbc5cfc00897ed644f29456a52c4f07f94844f Mon Sep 17 00:00:00 2001 From: mat Date: Sat, 7 May 2022 18:30:44 -0500 Subject: improve usage --- data-code-generator/main.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'data-code-generator/main.py') diff --git a/data-code-generator/main.py b/data-code-generator/main.py index 07a591e4..c95bd080 100644 --- a/data-code-generator/main.py +++ b/data-code-generator/main.py @@ -39,9 +39,11 @@ with open('burger.json', 'r') as f: burger_data = json.load(f) burger_packets_data = burger_data[0]['packets']['packet'] -packet_ids = list(map(int, sys.argv[1:])) -print(packet_ids) -packetcodegen.generate(burger_packets_data, mappings, packet_ids) +packet_id, direction, state = int(sys.argv[1]), sys.argv[2], sys.argv[3] +print( + f'Generating code for packet id: {packet_id} with direction {direction} and state {state}') +packetcodegen.generate(burger_packets_data, mappings, + packet_id, direction, state) os.system('cd .. && cargo fmt') -- cgit v1.2.3