1. import json
  3. def initialize_api_response_components(api_response_data):
  4. """
  5. Initializes components of API responses for exploratory work.
  7. Args:
  8. api_response_data (dict): The raw API response data (usually a dictionary).
  10. Returns:
  11. dict: A dictionary containing initialized components. Returns an empty dict if input is invalid.
  12. """
  14. if not isinstance(api_response_data, dict):
  15. print("Error: Invalid input. Expected a dictionary.")
  16. return {}
  18. initialized_components = {}
  20. # Basic metadata
  21. initialized_components['status_code'] = api_response_data.get('status_code', 'N/A') # Get status code, default to 'N/A'
  22. initialized_components['request_url'] = api_response_data.get('request_url', 'N/A') # Get request URL, default to 'N/A'
  23. initialized_components['timestamp'] = api_response_data.get('timestamp', 'N/A') # Get timestamp, default to 'N/A'
  25. # Response headers
  26. initialized_components['response_headers'] = api_response_data.get('headers', {}) # Get headers, default to empty dict
  28. # Response body - Handling different content types
  29. response_body = api_response_data.get('body', '')
  30. content_type = api_response_data.get('content_type', 'application/json') # Default content type
  32. if content_type == 'application/json':
  33. try:
  34. initialized_components['response_data'] = json.loads(response_body) # Parse JSON
  35. except json.JSONDecodeError:
  36. initialized_components['response_data'] = response_body # Store as string if JSON parsing fails
  37. elif content_type == 'text/plain':
  38. initialized_components['response_data'] = response_body # Store as plain text
  39. else:
  40. initialized_components['response_data'] = response_body #Store as raw string if other type
  44. #Error/message handling
  45. initialized_components['error_message'] = api_response_data.get('error', None) #Get error message
  47. return initialized_components

Add your comment