small refactor

monitor_utils/config.py

@@ -2,21 +2,25 @@ import logging
 import json
 import pathlib
 from functools import cache
+from typing import Enum
 from logging.handlers import RotatingFileHandler
 
 from monitor_utils.constants import (
     PROXIES_PATH,
     PROJECT_ROOT,
-    DEVELOPMENT_MODE,
     PRODUCTION_BOT_TOKEN,
     DEVELOPMENT_BOT_TOKEN,
     LOGGING_LEVEL,
-    DevelopmentMode,
 )
 
 RESERVED = frozenset(logging.LogRecord("", 0, "", 0, "", (), None).__dict__)
 
 
+class DevelopmentMode(str, Enum):
+    PRODUCTION = "PRODUCTION"
+    DEVELOPMENT = "DEVELOPMENT"
+
+
 class JSONFormatter(logging.Formatter):
     def format(self, record: logging.LogRecord) -> str:
         payload = {
@@ -63,11 +67,8 @@ def configure_logging(monitor_name: str):
     logging.getLogger("pymongo").setLevel(logging.WARNING)
 
 
-def is_production(override_production=False) -> bool:
-    if override_production:
-        return False
-
-    return DEVELOPMENT_MODE == DevelopmentMode.PRODUCTION
+def is_production(development_mode: DevelopmentMode) -> bool:
+    return development_mode == DevelopmentMode.PRODUCTION
 
 
 def get_bot_token(is_production: bool) -> str: