aboutsummaryrefslogtreecommitdiff
path: root/data-code-generator/main.py
diff options
context:
space:
mode:
authormat <github@matdoes.dev>2022-05-07 18:30:44 -0500
committermat <github@matdoes.dev>2022-05-07 18:30:44 -0500
commit2ddbc5cfc00897ed644f29456a52c4f07f94844f (patch)
treeeb0316b2a88fca19fc85e21653ab0dcdbda2beeb /data-code-generator/main.py
parentc10b732510868819a652b76e31278906914d699d (diff)
downloadazalea-drasl-2ddbc5cfc00897ed644f29456a52c4f07f94844f.tar.xz
improve usage
Diffstat (limited to 'data-code-generator/main.py')
-rw-r--r--data-code-generator/main.py8
1 files changed, 5 insertions, 3 deletions
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')