Commit 0c9e170d authored by Grant, Josh's avatar Grant, Josh
Browse files

add pythonic True/False to boolify; set ENVAR if not set.

parent 7907196a
Loading
Loading
Loading
Loading
+8 −5
Original line number Diff line number Diff line
@@ -9,9 +9,9 @@ def boolify(var):

    :param var: the variable to check to see if it can be converted to bool
    """
    if var in [0, '0', 'FALSE', 'False', 'false']:
    if var in [0, '0', 'FALSE', 'False', 'false', False]:
        return False
    if var in [1, '1', 'TRUE', 'True', 'true']:
    if var in [1, '1', 'TRUE', 'True', 'true', True]:
        return True
    raise TypeError('unable to evaluate expected boolean')

@@ -34,9 +34,12 @@ def check_environment(env_var, default=None):
        return os.environ[env_var]
    # assume if in python environment, it is already a bool or int
    if env_var in globals():
        os.environ[env_var] = str(globals()[env_var])
        return globals()[env_var]
    if env_var in locals():
        os.environ[env_var] = str(locals()[env_var])
        return locals()[env_var]
    os.environ[env_var] = str(default)
    return default