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)