1. import requests
  2. import argparse
  3. import json
  5. def attach_metadata(url, headers=None, data=None, method='GET'):
  6. """
  7. Attaches metadata to an HTTP request.
  9. Args:
  10. url (str): The URL to make the request to.
  11. headers (dict, optional): HTTP headers to include. Defaults to None.
  12. data (dict, optional): Data to send in the request body. Defaults to None.
  13. method (str, optional): HTTP method (GET, POST, PUT, DELETE, etc.). Defaults to 'GET'.
  15. Returns:
  16. requests.Response: The response object from the request.
  17. """
  19. # Default headers with metadata
  20. default_headers = {
  21. 'X-Request-Source': 'local_utility',
  22. 'X-Request-Timestamp': requests.utils.get_timestamp(), # Timestamp
  23. 'X-Request-User-Agent': 'local_utility/1.0' # User agent
  24. }
  26. # Merge user-provided headers with default headers
  27. if headers:
  28. merged_headers = {**default_headers, **headers}
  29. else:
  30. merged_headers = default_headers
  32. try:
  33. if method.upper() == 'GET':
  34. response = requests.get(url, headers=merged_headers, params=data)
  35. elif method.upper() == 'POST':
  36. response = requests.post(url, headers=merged_headers, json=data)
  37. elif method.upper() == 'PUT':
  38. response = requests.put(url, headers=merged_headers, json=data)
  39. elif method.upper() == 'DELETE':
  40. response = requests.delete(url, headers=merged_headers)
  41. else:
  42. raise ValueError(f"Unsupported HTTP method: {method}")
  44. return response
  46. except requests.exceptions.RequestException as e:
  47. print(f"Request failed: {e}")
  48. return None
  52. if __name__ == '__main__':
  53. parser = argparse.ArgumentParser(description='Attach metadata to HTTP requests.')
  54. parser.add_argument('url', help='The URL to make the request to.')
  55. parser.add_argument('--headers', help='Additional HTTP headers (JSON format).')
  56. parser.add_argument('--data', help='Data to send in the request body (JSON format).')
  57. parser.add_argument('--method', default='GET', choices=['GET', 'POST', 'PUT', 'DELETE'], help='HTTP method.')
  59. args = parser.parse_args()
  61. headers = None
  62. if args.headers:
  63. try:
  64. headers = json.loads(args.headers)
  65. except json.JSONDecodeError:
  66. print("Invalid JSON format for headers.")
  67. exit(1)
  69. data = None
  70. if args.data:
  71. try:
  72. data = json.loads(args.data)
  73. except json.JSONDecodeError:
  74. print("Invalid JSON format for data.")
  75. exit(1)
  77. response = attach_metadata(args.url, headers=headers, data=data, method=args.method)
  79. if response:
  80. print("Status Code:", response.status_code)
  81. print("Response Headers:", response.headers)
  82. try:
  83. print("Response Body:", response.json()) #try to print as json
  84. except json.JSONDecodeError:
  85. print("Response Body:", response.text)

Add your comment