diff --git a/common/src/aaredaqlib/logger_config.py b/common/src/aaredaqlib/logger_config.py new file mode 100644 index 00000000..f56e376a --- /dev/null +++ b/common/src/aaredaqlib/logger_config.py @@ -0,0 +1,17 @@ +import logging +import logging.config +import yaml +import os + +def setup_logger(name="aaredaq"): + # switch to production mode using: $ APP_ENV=prod python main.py + env = os.getenv("APP_ENV", "dev") # default: dev + config_file = ( + "/sls/MX/applications/aaredaq/common/src/aaredaqlib/logging_prod.yaml" if env == "prod" else "/sls/MX/applications/aaredaq/common/src/aaredaqlib/logging_dev.yaml" + ) + + with open(config_file, "r") as f: + config = yaml.safe_load(f.read()) + logging.config.dictConfig(config) + + return logging.getLogger(name)