1. import json
  2. import argparse
  4. def format_api_output(data, overrides=None):
  5. """
  6. Formats API output with manual overrides for internal tooling.
  8. Args:
  9. data (dict): The raw API response data.
  10. overrides (dict, optional): A dictionary of key-value pairs to override
  11. in the output. Defaults to None.
  13. Returns:
  14. str: Formatted JSON string.
  15. """
  17. formatted_data = {}
  18. for key, value in data.items():
  19. if key in overrides:
  20. formatted_data[key] = overrides[key] # Apply override if available
  21. else:
  22. formatted_data[key] = value
  24. return json.dumps(formatted_data, indent=4)
  26. if __name__ == '__main__':
  27. parser = argparse.ArgumentParser(description='Format API output.')
  28. parser.add_argument('--input_data', type=str, required=True, help='JSON input data')
  29. parser.add_argument('--overrides', type=str, help='JSON override data')
  31. args = parser.parse_args()
  33. try:
  34. data = json.loads(args.input_data)
  35. except json.JSONDecodeError as e:
  36. print(f"Error decoding input JSON: {e}")
  37. exit(1)
  39. overrides = None
  40. if args.overrides:
  41. try:
  42. overrides = json.loads(args.overrides)
  43. except json.JSONDecodeError as e:
  44. print(f"Error decoding overrides JSON: {e}")
  45. exit(1)
  47. formatted_output = format_api_output(data, overrides)
  48. print(formatted_output)

Add your comment