Skip to content

Commit 1bcc080

Browse files
committed
move log level parsing to its own function
1 parent cdc365e commit 1bcc080

File tree

1 file changed

+17
-14
lines changed

1 file changed

+17
-14
lines changed

config.py

Lines changed: 17 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,21 @@ def _get_required_env_var(key: str) -> str:
4242
raise ValueError(f"{key} environment variable is required.")
4343
return value
4444

45+
@staticmethod
46+
def _parse_log_level(log_level_name: str) -> int:
47+
log_levels = {
48+
"debug": logging.DEBUG,
49+
"info": logging.INFO,
50+
"warning": logging.WARNING,
51+
"error": logging.ERROR,
52+
"critical": logging.CRITICAL,
53+
}
54+
if log_level_name not in log_levels:
55+
raise ValueError(
56+
f"Invalid LOG_LEVEL: '{log_level_name}'. Must be one of: {', '.join(log_levels.keys())}"
57+
)
58+
return log_levels[log_level_name]
59+
4560
@staticmethod
4661
def _init_db_provider(db_type: str) -> DBProvider:
4762
"""
@@ -99,22 +114,10 @@ def load() -> "Config":
99114
get = Config._get_required_env_var
100115

101116
# Initialize logger
102-
log_level_name = get("LOG_LEVEL").lower()
103-
log_levels = {
104-
"debug": 10,
105-
"info": 20,
106-
"warning": 30,
107-
"error": 40,
108-
"critical": 50,
109-
}
110-
if log_level_name not in log_levels:
111-
raise ValueError(
112-
f"Invalid LOG_LEVEL: '{log_level_name}'. Must be one of: {', '.join(log_levels)}"
113-
)
114-
log_level = log_levels[log_level_name]
117+
log_level = Config._parse_log_level(get("LOG_LEVEL").lower())
115118
logging.basicConfig(level=log_level)
116119
logger = logging.getLogger(__name__)
117-
logger.debug("Logging initialized at level: %s", log_level_name.upper())
120+
logger.debug("Logging initialized at level: %s", log_level.upper())
118121

119122
# Initialize db
120123
db_type = get("DB_TYPE")

0 commit comments

Comments
 (0)