1. import os
  2. import argparse
  3. def get_env_var(var_name, default=None):
  4. """Retrieves an environment variable, returning a default if it's not set."""
  5. try:
  6. return os.environ[var_name]
  7. except KeyError:
  8. return default
  9. def main():
  10. parser = argparse.ArgumentParser(description="A quick prototype using environment variables.")
  11. parser.add_argument("--name", type=str, help="The name to greet.")
  12. parser.add_argument("--debug", action="store_true", help="Enable debug mode.")
  13. args = parser.parse_args()
  14. # Get environment variables with defaults
  15. api_key = get_env_var("API_KEY", "default_api_key")
  16. log_level = get_env_var("LOG_LEVEL", "INFO")
  17. data_url = get_env_var("DATA_URL", "http://example.com/data")
  18. # Use the environment variables
  19. print(f"API Key: {api_key}")
  20. print(f"Log Level: {log_level}")
  21. print(f"Data URL: {data_url}")
  22. if args.name:
  23. print(f"Hello, {args.name}!")
  24. if args.debug:
  25. print("Debug mode enabled.")
  26. if __name__ == "__main__":
  27. main()

Add your comment