1. import re
  2. from collections import defaultdict
  4. def group_response_headers(headers):
  5. """
  6. Groups response headers for isolated environments and performs basic sanity checks.
  8. Args:
  9. headers (dict): A dictionary of response headers.
  11. Returns:
  12. dict: A dictionary where keys are environment names and values are lists of headers
  13. specific to that environment. Returns an empty dictionary if input is invalid.
  14. """
  15. if not isinstance(headers, dict):
  16. return {}
  18. environments = defaultdict(list)
  20. # Define environment-specific header patterns. Can be extended.
  21. environment_patterns = {
  22. "dev": r"X-Dev-.*", #headers starting with X-Dev-
  23. "staging": r"X-Staging-.*", # headers starting with X-Staging-
  24. "prod": r"X-Prod-.*", #headers starting with X-Prod-
  25. }
  27. for key, value in headers.items():
  28. if isinstance(value, str): #process only string header values
  29. for env, pattern in environment_patterns.items():
  30. if re.match(pattern, key):
  31. environments[env].append(key)
  32. break # Only assign to one environment
  34. # Sanity check: Ensure each environment has at least one header.
  35. filtered_environments = {k: v for k, v in environments.items() if v}
  37. return filtered_environments
  39. if __name__ == '__main__':
  40. #Example Usage
  41. headers1 = {
  42. "Content-Type": "application/json",
  43. "X-Dev-Debug": "enabled",
  44. "X-Staging-Feature": "new",
  45. "Server": "nginx",
  46. }
  47. grouped_headers1 = group_response_headers(headers1)
  48. print(grouped_headers1)
  50. headers2 = {
  51. "Content-Type": "application/json",
  52. "X-Prod-Security": "high"
  53. }
  54. grouped_headers2 = group_response_headers(headers2)
  55. print(grouped_headers2)
  57. headers3 = {}
  58. grouped_headers3 = group_response_headers(headers3)
  59. print(grouped_headers3)
  61. headers4 = "not a dictionary"
  62. grouped_headers4 = group_response_headers(headers4)
  63. print(grouped_headers4)

Add your comment