1. import requests
  2. import json
  3. import argparse
  4. import sys
  6. def api_request(url, method='GET', data=None, headers=None):
  7. """
  8. Makes an API request to the specified URL.
  10. Args:
  11. url (str): The URL to request.
  12. method (str): The HTTP method (GET, POST, PUT, DELETE, etc.). Defaults to GET.
  13. data (dict): The data to send in the request body (for POST, PUT, etc.). Defaults to None.
  14. headers (dict): The HTTP headers to include in the request. Defaults to None.
  16. Returns:
  17. dict: The JSON response from the API, or None if an error occurred.
  18. """
  19. try:
  20. if method.upper() == 'GET':
  21. response = requests.get(url, headers=headers)
  22. elif method.upper() == 'POST':
  23. response = requests.post(url, headers=headers, json=data)
  24. elif method.upper() == 'PUT':
  25. response = requests.put(url, headers=headers, json=data)
  26. elif method.upper() == 'DELETE':
  27. response = requests.delete(url, headers=headers)
  28. else:
  29. print(f"Unsupported HTTP method: {method}")
  30. return None
  32. response.raise_for_status() # Raise HTTPError for bad responses (4xx or 5xx)
  33. return response.json()
  34. except requests.exceptions.RequestException as e:
  35. print(f"API request failed: {e}", file=sys.stderr)
  36. return None
  37. except json.JSONDecodeError as e:
  38. print(f"Failed to decode JSON response: {e}", file=sys.stderr)
  39. return None
  42. def main():
  43. """
  44. Main function to handle command-line arguments and API requests.
  45. """
  46. parser = argparse.ArgumentParser(description="Local utility with API integration.")
  47. parser.add_argument("api_url", help="The base URL of the API.")
  48. parser.add_argument("endpoint", help="The API endpoint to call.")
  49. parser.add_argument("--method", default="GET", help="HTTP method (GET, POST, PUT, DELETE).")
  50. parser.add_argument("--data", help="JSON data to send in the request body.")
  51. parser.add_argument("--headers", help="JSON data to send in the request headers.")
  52. args = parser.parse_args()
  54. url = args.api_url + args.endpoint
  55. method = args.method.upper()
  57. headers = {}
  58. if args.headers:
  59. headers = json.loads(args.headers)
  61. data = None
  62. if args.data:
  63. data = json.loads(args.data)
  65. response = api_request(url, method, data, headers)
  67. if response:
  68. print(json.dumps(response, indent=4)) #pretty print the json
  69. else:
  70. sys.exit(1) #exit with error code if API request fails
  73. if __name__ == "__main__":
  74. main()

Add your comment