diff --git a/composer/agent.py b/composer/agent.py index f1116f4..0b04f9b 100644 --- a/composer/agent.py +++ b/composer/agent.py @@ -14,7 +14,16 @@ formatter = logging.Formatter("[%(levelname)s] - %(message)s") handler.setFormatter(formatter) logger = logging.getLogger("composer-agent-logger") logger.addHandler(handler) -logger.setLevel(logging.DEBUG) #TODO: add something to change log levles +log_levels = { + "DEBUG": logging.DEBUG, + "INFO": logging.INFO, + "WARN": logging.WARN, + "ERROR": logging.ERROR +} +if getenv("LOG_LEVEL") in log_levels: + logger.setLevel(log_levels[getenv("LOG_LEVEL")]) +else: + logger.setLevel(logging.INFO) environ['GIT_TERMINAL_PROMPT'] = '0'