Merge pull request #18 from color/mpcusack-override

Allow clearing of runtime overrides

Author: mpcusack-color

clrenv/lazy_env.py

@@ -8,6 +8,7 @@
 import shlex
 import socket
 import sys
+import traceback
 
 from botocore.exceptions import EndpointConnectionError
 
@@ -24,11 +25,13 @@
 OFFLINE_FLAG = 'CLRENV_OFFLINE_DEV'
 OFFLINE_VALUE = 'CLRENV_OFFLINE_PLACEHOLDER'
 
+DEBUG_MODE = environ.get('CLRENV_DEBUG', '').lower() in ('true', '1')
 
 class LazyEnv(object):
     def __init__(self):
         self.__mode = tuple(shlex.split(environ.get('CLRENV_MODE', '')))
         self.__env = None
+        self.__runtime_overrides = {}
 
     def is_set(self):
         return self.__env is not None
@@ -40,18 +43,35 @@ def set_mode(self, *mode):
     def get_mode(self):
         return self.__mode
 
+    def clear_runtime_overrides(self):
+        self.__runtime_overrides.clear()
+
     def __getattr__(self, key):
         if self.__env is None:
             self.__env = get_env(*self.__mode)
 
-        try:
-            return getattr(self.__env, key)
-        except AttributeError:
-            return None
+        if key in self.__runtime_overrides:
+            return self.__runtime_overrides[key]
+        return getattr(self.__env, key, None)
 
     def __getitem__(self, key):
         return self.__getattr__(key)
 
+    def __setattr__(self, key, value):
+        # Internal fields are prefixed with a _
+        if key.startswith('_'):
+            return object.__setattr__(self, key, value)
+
+        # Ideally we wouldn't be overriding global state like this at all, but at least
+        # make it loud.
+        logger.warning(f'Manually overriding env.{key} to {value}.')
+        if DEBUG_MODE:
+            # Get stack and remove this frame.
+            tb = traceback.extract_stack()[:-1]
+            logger.warning("".join(traceback.format_list(tb)))
+
+        self.__runtime_overrides[key] = value
+
 _env = {}
 def get_env(*mode):
     global _env

setup.py

@@ -7,7 +7,7 @@
 
 
 setup(name = "clrenv",
-      version = "0.1.7",
+      version = "0.1.8",
       description = "A tool to give easy access to environment yaml file to python.",
       author = "Color Genomics",
       author_email = "[email protected]",