File size: 952 Bytes
712579e |
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 31 32 33 34 35 36 37 |
import yaml
import os
from typing import List
from logger.custom_logger import CustomLoggerTracker
# ------------------------------
# Custom Logger Initialization
# ------------------------------
custom_log = CustomLoggerTracker()
logger = custom_log.get_logger("configs")
logger.info("Logger initialized for Configs Query module")
def load_yaml_config(file_dir: str):
with open(file_dir, "r") as f:
config = yaml.safe_load(f)
if config is None:
return {}
return config
def _log(process_log: List[str], message: str, level: str = "info") -> None:
"""Append to process_log AND send to the central logger."""
process_log.append(message)
if level == "error":
logger.error(message)
elif level == "warning":
logger.warning(message)
else:
logger.info(message)
if __name__=="__main__":
configs = load_yaml_config("config.yaml")
logger.info(f"configs: {configs}") |