1. import json
  2. from typing import Any, Dict, List, Union
  4. def nest_api_response(data: Dict[str, Any], dry_run: bool = False) -> Union[Dict[str, Any], List[Dict[str, Any]]]:
  5. """
  6. Nests API response structures with optional dry-run mode.
  8. Args:
  9. data: The API response data (dict or list).
  10. dry_run: If True, only prints the structure without modifying it.
  12. Returns:
  13. Nested data structure.
  14. """
  16. if dry_run:
  17. print("Dry run mode: Printing structure only.")
  18. print(json.dumps(data, indent=4))
  19. return data # Return original data for consistency
  21. if isinstance(data, dict):
  22. nested_data = {}
  23. for key, value in data.items():
  24. nested_data[key] = nest_api_response(value, dry_run)
  25. return nested_data
  26. elif isinstance(data, list):
  27. nested_data = []
  28. for item in data:
  29. nested_data.append(nest_api_response(item, dry_run))
  30. return nested_data
  31. else:
  32. return data
  34. if __name__ == '__main__':
  35. # Example usage
  36. api_response = {
  37. "status": "success",
  38. "code": 200,
  39. "data": {
  40. "user": {
  41. "id": 123,
  42. "name": "John Doe",
  43. "email": "john.doe@example.com"
  44. },
  45. "posts": [
  46. {"id": 1, "title": "First Post"},
  47. {"id": 2, "title": "Second Post"}
  48. ]
  49. }
  50. }
  52. print("Original API Response:")
  53. print(json.dumps(api_response, indent=4))
  55. nested_response = nest_api_response(api_response)
  56. print("\nNested API Response:")
  57. print(json.dumps(nested_response, indent=4))
  59. # Dry run example
  60. nested_response_dry = nest_api_response(api_response, dry_run=True)
  61. print("\nDry Run Output:")
  62. print(nested_response_dry)

Add your comment