1. import os
  2. import logging
  4. class EnvVarWrapper:
  5. """
  6. Wraps environment variable access for testing purposes,
  7. providing a consistent interface and support for older Python versions.
  8. """
  10. def __init__(self, var_name, default_value=None):
  11. """
  12. Initializes the wrapper for a specific environment variable.
  14. Args:
  15. var_name (str): The name of the environment variable.
  16. default_value (any, optional): The default value to use if the environment variable is not set. Defaults to None.
  17. """
  18. self.var_name = var_name
  19. self.default_value = default_value
  20. self.value = os.environ.get(var_name, self.default_value) # Get env var, use default if missing
  21. self.is_set = bool(self.value) #track if variable is set
  23. def get(self):
  24. """
  25. Returns the value of the environment variable.
  27. Returns:
  28. any: The value of the environment variable.
  29. """
  30. return self.value
  32. def is_enabled(self):
  33. """
  34. Returns True if the environment variable is set and not empty, False otherwise.
  35. """
  36. return self.is_set
  38. def set_default(self, new_default):
  39. """
  40. Updates the default value of the environment variable.
  41. """
  42. self.default_value = new_default
  43. self.value = os.environ.get(self.var_name, self.default_value)
  44. self.is_set = bool(self.value)
  47. if __name__ == '__main__':
  48. # Example Usage
  49. logging.basicConfig(level=logging.INFO)
  51. # Create a wrapper for a variable
  52. db_host = EnvVarWrapper("DB_HOST", "localhost")
  53. db_port = EnvVarWrapper("DB_PORT", "5432")
  54. debug_mode = EnvVarWrapper("DEBUG", False)
  56. # Get the values
  57. host = db_host.get()
  58. port = db_port.get()
  59. debug = debug_mode.get()
  61. print(f"DB Host: {host}")
  62. print(f"DB Port: {port}")
  63. print(f"Debug Mode: {debug}")
  65. #Check if variable is set
  66. print(f"DB Host is enabled: {db_host.is_enabled()}")
  67. print(f"Debug Mode is enabled: {debug_mode.is_enabled()}")
  69. #Set new default value
  70. db_port.set_default(5433)
  71. print(f"DB Port after setting new default: {db_port.get()}")

Add your comment