File size: 886 Bytes
affcd23 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import logging
import logging.config
from pathlib import Path
from hw_asr.utils import read_json, ROOT_PATH
def setup_logging(
save_dir, log_config=None, default_level=logging.INFO
):
"""
Setup logging configuration
"""
if log_config is None:
log_config = str(ROOT_PATH / "hw_asr" / "logger" / "logger_config.json")
log_config = Path(log_config)
if log_config.is_file():
config = read_json(log_config)
# modify logging paths based on run config
for _, handler in config["handlers"].items():
if "filename" in handler:
handler["filename"] = str(save_dir / handler["filename"])
logging.config.dictConfig(config)
else:
print(
"Warning: logging configuration file is not found in {}.".format(log_config)
)
logging.basicConfig(level=default_level)
|