llm-review/json_filter.py
2025-03-24 12:03:55 -04:00

52 lines
1.4 KiB
Python

import argparse
import json
import logging
from lib import filter_schematic_json
logger = logging.getLogger(__name__)
handler = logging.StreamHandler()
formatter = logging.Formatter("%(asctime)s [%(levelname)s] %(message)s")
handler.setFormatter(formatter)
logger.addHandler(handler)
def main():
parser = argparse.ArgumentParser(
description="Filter graphical details from a schematic JSON file"
)
parser.add_argument("--input", "-i", required=True, help="Path to input JSON file")
parser.add_argument(
"--output", "-o", required=True, help="Path to output filtered JSON file"
)
parser.add_argument("--log-level", default="INFO", help="Logging level")
args = parser.parse_args()
logger.setLevel(args.log_level)
logger.info(f"Reading JSON from {args.input}")
try:
with open(args.input, "r") as f:
input_json = json.load(f)
except Exception as e:
logger.error(f"Error reading input file: {e}")
return 1
logger.info("Filtering JSON...")
filtered_json = filter_schematic_json(input_json)
logger.info(f"Writing filtered JSON to {args.output}")
try:
with open(args.output, "w") as f:
json.dump(filtered_json, f)
except Exception as e:
logger.error(f"Error writing output file: {e}")
return 1
logger.info("JSON filtering complete")
return 0
if __name__ == "__main__":
main()