diff options
Diffstat (limited to 'data-code-generator/main.py')
| -rw-r--r-- | data-code-generator/main.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/data-code-generator/main.py b/data-code-generator/main.py index e1668369..07a591e4 100644 --- a/data-code-generator/main.py +++ b/data-code-generator/main.py @@ -2,6 +2,7 @@ from mappings import Mappings import packetcodegen import requests import json +import sys import os # enable this if you already have the burger.json and don't want to wait @@ -34,11 +35,14 @@ if not SKIP_BURGER: client_mappings_url = package_data['downloads']['client_mappings']['url'] mappings = Mappings.parse(requests.get(client_mappings_url).text) - with open('burger.json', 'r') as f: burger_data = json.load(f) burger_packets_data = burger_data[0]['packets']['packet'] -packetcodegen.generate(burger_packets_data, mappings) +packet_ids = list(map(int, sys.argv[1:])) +print(packet_ids) +packetcodegen.generate(burger_packets_data, mappings, packet_ids) os.system('cd .. && cargo fmt') + +print('Done!') |