| 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}") |