def _get_log_level()

in services/jenkins-autoscaling/lambda_mxnet_ci/autoscaling/handler.py [0:0]

• 17 lines of code
• 6 McCabe index (conditional complexity)

def _get_log_level(env_var_key, default_level) -> int:
    """
    Read the log level from an environment variable or set it to a default if it does not exist
    :param env_var_key:
    :param default_level:
    :return:
    """
    try:
        logging_level = os.environ[env_var_key]
    except KeyError:
        logging.warning('Unable to find %s env var. Defaulting to %s.', env_var_key, default_level)
        return default_level
    else:
        if logging_level == 'DEBUG':
            return logging.DEBUG
        elif logging_level == 'INFO':
            return logging.INFO
        elif logging_level == 'WARNING':
            return logging.WARNING
        elif logging_level == 'ERROR':
            return logging.ERROR
        else:
            raise KeyError('Unable to match logging level {} for {}'.format(logging_level, env_var_key))