From c10b732510868819a652b76e31278906914d699d Mon Sep 17 00:00:00 2001 From: mat Date: Sat, 7 May 2022 18:26:19 -0500 Subject: improve data generator and add packet --- data-code-generator/main.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'data-code-generator/main.py') 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!') -- cgit v1.2.3