Fix ordering of blocks

1 files changed, 6 insertions, 0 deletions

diff --git a/codegen/lib/extract.py b/codegen/lib/extract.py
index 360c368c..40263779 100644
--- a/codegen/lib/extract.py
+++ b/codegen/lib/extract.py
@@ -21,11 +21,17 @@ def get_block_states_report(version_id: str):
     with open(get_dir_location(f'downloads/generated-{version_id}/reports/blocks.json'), 'r') as f:
         return json.load(f)
 
+
 def get_block_states_burger(version_id: str):
     burger_data = get_burger_data_for_version(version_id)
     return burger_data[0]['blocks']['block']
 
 
+def get_ordered_blocks_burger(version_id: str):
+    burger_data = get_burger_data_for_version(version_id)
+    return burger_data[0]['blocks']['ordered_blocks']
+
+
 def get_burger_data_for_version(version_id: str):
     if not os.path.exists(get_dir_location(f'downloads/burger-{version_id}.json')):
         get_burger()