Post Black Linting

utils/configmanager.py

@@ -29,14 +29,15 @@ PROTECTED_KEYS = [
     "PATH_EXCLUSION_CONST",
     "MIN_FILES_FOR_PATH",
     "VT_THREAT_TOLERANCE",
-    "POLICY_MAP_ENF_AUD"
+    "POLICY_MAP_ENF_AUD",
 ]
 
 _protected_config = {}
 
+
 def get_system_config_path() -> Path:
     # Check inside bundled EXE directory first
-    bundled_dir = Path(getattr(sys, '_MEIPASS', ''))
+    bundled_dir = Path(getattr(sys, "_MEIPASS", ""))
     bundled_path = bundled_dir / "system_config.json"
     if bundled_path.exists():
         return bundled_path
@@ -44,6 +45,7 @@ def get_system_config_path() -> Path:
     # Fallback to external location
     return Path(__file__).parent.parent / "system_config.json"
 
+
 def load_protected_config() -> dict:
     global _protected_config
     try:
@@ -56,15 +58,16 @@ def load_protected_config() -> dict:
             "PATH_EXCLUSION_CONST": 4,
             "MIN_FILES_FOR_PATH": 4,
             "VT_THREAT_TOLERANCE": 4,
-            "POLICY_MAP_ENF_AUD": {
-                "enforced_id": "audit_id"
-            }
+            "POLICY_MAP_ENF_AUD": {"enforced_id": "audit_id"},
         }
 
     _protected_config = {key: system_config[key] for key in PROTECTED_KEYS}
     return _protected_config
 
-def get_protected_value(key: str, cast_type: Callable[[str], T] = str, default: Optional[T] = None) -> Optional[T]:
+
+def get_protected_value(
+    key: str, cast_type: Callable[[str], T] = str, default: Optional[T] = None
+) -> Optional[T]:
     value = _protected_config.get(key)
     if value is None:
         logging.warning(f"Protected config key '{key}' not found.")
@@ -74,9 +77,12 @@ def get_protected_value(key: str, cast_type: Callable[[str], T] = str, default:
             value = value.strip("'\"")
         return cast_type(value)
     except (ValueError, TypeError):
-        logging.warning(f"Invalid value for protected key '{key}': {value}. Expected type {cast_type.__name__}.")
+        logging.warning(
+            f"Invalid value for protected key '{key}': {value}. Expected type {cast_type.__name__}."
+        )
         return default
 
+
 def get_protected_json(key: str, default: str = "{}") -> dict:
     raw = _protected_config.get(key, default)
     if isinstance(raw, dict):
@@ -85,13 +91,11 @@ def get_protected_json(key: str, default: str = "{}") -> dict:
         return json.loads(raw)
     except json.JSONDecodeError:
         try:
-            escaped = raw.encode('unicode_escape').decode('utf-8')
+            escaped = raw.encode("unicode_escape").decode("utf-8")
             return json.loads(escaped)
         except Exception as e:
             logging.error(f"Failed to parse protected JSON key '{key}': {e}")
             return json.loads(default)
-        
-
 
 
 def load_env_json(key: str, default: str):
@@ -100,13 +104,16 @@ def load_env_json(key: str, default: str):
         return json.loads(raw)
     except json.JSONDecodeError:
         try:
-            escaped = raw.encode('unicode_escape').decode('utf-8')
+            escaped = raw.encode("unicode_escape").decode("utf-8")
             return json.loads(escaped)
         except Exception as e:
             logging.error(f"Failed to parse {key}: {e}")
             return json.loads(default)
 
-def load_env(key: str, cast_type: Callable[[str], T] = str, default: Optional[T] = None) -> Optional[T]:
+
+def load_env(
+    key: str, cast_type: Callable[[str], T] = str, default: Optional[T] = None
+) -> Optional[T]:
     """
     Safely retrieves an environment variable and casts it to the desired type.
 
@@ -126,5 +133,7 @@ def load_env(key: str, cast_type: Callable[[str], T] = str, default: Optional[T]
         value = value.strip("'\"")  # Strip surrounding quotes
         return cast_type(value)
     except (ValueError, TypeError):
-        logger.warning(f"Invalid value for env var '{key}': {value}. Expected type {cast_type.__name__}.")
-        return default
+        logger.warning(
+            f"Invalid value for env var '{key}': {value}. Expected type {cast_type.__name__}."
+        )
+        return default