1. import os
  3. def truncate_runtime_data(data, environment):
  4. """
  5. Truncates runtime environment specific data for staging environments.
  7. Args:
  8. data (dict): The data dictionary to truncate.
  9. environment (str): The current environment (e.g., "staging").
  11. Returns:
  12. dict: The truncated data dictionary.
  13. """
  14. if environment == "staging":
  15. # Truncate environment-specific data. Example: remove sensitive keys.
  16. if "api_key" in data:
  17. del data["api_key"]
  18. if "secret_key" in data:
  19. del data["secret_key"]
  20. if "database_password" in data:
  21. del data["database_password"]
  22. #Truncate other environment specific values
  23. for key in list(data.keys()): # Iterate over copy to allow deletion
  24. if key.startswith("staging_"):
  25. del data[key]
  27. return data
  29. if __name__ == '__main__':
  30. #Example Usage
  31. test_data = {
  32. "api_key": "staging_api_key",
  33. "secret_key": "staging_secret",
  34. "database_password": "staging_password",
  35. "name": "Test Data",
  36. "staging_setting": "value"
  37. }
  39. current_environment = os.environ.get("ENVIRONMENT", "production") #Default to production if not set
  40. truncated_data = truncate_runtime_data(test_data, current_environment)
  41. print(truncated_data)

Add your comment