Cleaning up menu, added Move to Audit/Enforcement

flows/prepPolicy.py

@@ -24,13 +24,12 @@ import pandas as pd
 from models.execution import ExecutionHistoryRecord, Hash
 from models.policy import Allowlist, Policy
 from services.API import AirlockAPIWrapper
+from utils.configmanager import get_protected_value, load_env, load_env_json
 from utils.selector import Selector
 from utils.utils import (
     colorText,
     formatHTML,
     import_to_dataframe,
-    load_env,
-    load_env_json,
     regulator,
 )
 
@@ -250,8 +249,8 @@ def buildPreflights():
         formatHTML(df, f"{working_dir}\\Preflight\\HTML\\{name}.html")
 
 def splitFilepathsGrouped(df, col="filename"):
-    path_exclusion_constant = load_env("PATH_EXCLUSION_CONST", cast_type= int)
-    min_files_for_path = load_env("MIN_FILES_FOR_PATH", cast_type= int)
+    path_exclusion_constant = get_protected_value("PATH_EXCLUSION_CONST", cast_type= int)
+    min_files_for_path = get_protected_value("MIN_FILES_FOR_PATH", cast_type= int)
 
     def clean_split(path):
         if not isinstance(path, (str, bytes, os.PathLike)):
@@ -317,7 +316,7 @@ def calculatePath(approved_hashes, split):
         dfs_by_policy = [approved_hashes]
 
     badpathparts = load_env_json("BAD_PATH_PARTS", "[]")
-    min_files_for_path = load_env("MIN_FILES_FOR_PATH", cast_type = int)
+    min_files_for_path = get_protected_value("MIN_FILES_FOR_PATH", cast_type = int)
 
     processed_dfs = []