1. import argparse
  2. import json
  4. def decode_api_endpoint(api_endpoint, dry_run=False):
  5. """
  6. Decodes an API endpoint string, handling staging environment variables and dry-run mode.
  8. Args:
  9. api_endpoint (str): The API endpoint string.
  10. dry_run (bool): If True, only print the decoded endpoint without executing.
  12. Returns:
  13. str: The decoded API endpoint.
  14. """
  15. # Define staging environment variables (replace with your actual variables)
  16. staging_env = {
  17. "API_BASE_URL": "https://staging.example.com",
  18. "API_VERSION": "v2",
  19. "AUTH_TOKEN": "staging_token"
  20. }
  22. decoded_endpoint = api_endpoint
  24. # Replace placeholders with values from the staging environment
  25. for key, value in staging_env.items():
  26. decoded_endpoint = decoded_endpoint.replace(f"{{{{{key}}}", value)
  28. # Add API version if not already present
  29. if "v" not in decoded_endpoint and "api" in decoded_endpoint.lower():
  30. decoded_endpoint = f"{decoded_endpoint} {staging_env['API_VERSION']}"
  32. if dry_run:
  33. print(f"Dry Run: Decoded endpoint: {decoded_endpoint}")
  34. return decoded_endpoint # Return for proper usage
  35. else:
  36. print(f"Decoded endpoint: {decoded_endpoint}")
  37. return decoded_endpoint
  40. if __name__ == "__main__":
  41. parser = argparse.ArgumentParser(description="Decode API endpoints for staging environments.")
  42. parser.add_argument("endpoint", help="The API endpoint to decode.")
  43. parser.add_argument("--dry_run", action="store_true", help="Perform a dry run (print decoded endpoint without executing).")
  44. args = parser.parse_args()
  46. decode_api_endpoint(args.endpoint, args.dry_run)

Add your comment