NRL-1884 WIP PR comments

anjalitrace2-nhs · anjalitrace2-nhs · commit 23cb0216ca2f · 2026-01-14T10:57:22.000Z
diff --git a/Makefile b/Makefile
@@ -270,7 +270,7 @@ perftest-prepare:	## Prepare input files for producer & consumer perf tests
 	mkdir -p "${DIST_PATH}/nft"
 	aws s3 cp "s3://nhsd-nrlf--${ENV}-metadata/performance/seed-pointers-extract-${PERFTEST_TABLE_NAME}.zip" "${DIST_PATH}/pointer_extract-${PERFTEST_TABLE_NAME}.zip"
 	unzip "${DIST_PATH}/pointer_extract-${PERFTEST_TABLE_NAME}.zip"
-	cp "${DIST_PATH}/nft/seed-pointers-extract-${PERFTEST_TABLE_NAME}.csv" "${DIST_PATH}/seed-pointers-extract.csv"
+# 	cp "${DIST_PATH}/nft/seed-pointers-extract-${PERFTEST_TABLE_NAME}.csv" "${DIST_PATH}/seed-pointers-extract.csv"
 	PYTHONPATH=. poetry run python ./tests/performance/generate_producer_distributions.py
 
 perftest-producer:	## Run producer perf tests
diff --git a/scripts/seed_nft_tables.py b/scripts/seed_nft_tables.py
@@ -1,4 +1,5 @@
 import csv
+import json
 import os
 from datetime import datetime, timedelta, timezone
 from itertools import cycle
@@ -89,7 +90,7 @@ def _make_seed_pointer(
 
 
 def _write_pointer_extract_to_file(table_name, pointer_data):
-    local_csv_out = f"{nft_dist_path}/seed-pointers-extract-{table_name}.csv"
+    local_csv_out = f"{nft_dist_path}/seed-pointers-extract.csv"
     local_meta_out = f"{nft_dist_path}/info.json"
 
     print(f"writing pointer extract to files {local_csv_out} {local_meta_out}")
@@ -100,7 +101,15 @@ def _write_pointer_extract_to_file(table_name, pointer_data):
         writer.writerows(pointer_data)
     print(f"Pointer data saved to {local_csv_out}")
 
-    os.system(f"./scripts/get-current-info.sh > {local_meta_out}")
+    os.system(f"./scripts/get-current-info.sh > {local_meta_out}.temp")
+    with open(f"{local_meta_out}.temp", "w+") as temp_f:
+        metadata = json.load(temp_f)
+        metadata["table_name"] = table_name
+        with open(f"{local_meta_out}", "w+") as f:
+            json.dump(metadata, f)
+
+    os.system(f"rm {local_meta_out}.temp")
+
     print(f"Pointer extract metadata saved to {local_meta_out}")
 
 
