Major UI Improvment with blessing

services/policyhandler.py

@@ -21,15 +21,15 @@ import logging
 import os
 import sys
 
+from bson import ObjectId
 import pandas as pd
 import tqdm
-from bson import ObjectId
 
 from models.policy import Policy
 from services.API import AirlockAPIWrapper
 from utils.configmanager import get_protected_json
 from utils.setup import get_base_directory
-from utils.utils import colorText
+from utils.utils import areYouSure, colorText, get_sanitized_input
 
 logger = logging.getLogger(__name__)
 
@@ -235,3 +235,10 @@ def updateAuditPoliciesFromEnforcementPolices(api: AirlockAPIWrapper):
     for enforcement_policy, audit_policy in policy_relationship_map.items():
         api.policy_clone(enforcement_policy, audit_policy)
         api.policy_set_auditmode(audit_policy, "1")
+ 
+
+def confirmUpdateAfromE(api: AirlockAPIWrapper):
+    areYouSure()
+    confirmation = get_sanitized_input("Type 'I AGREE' to continue: ")
+    if confirmation.strip() == "I AGREE":
+        updateAuditPoliciesFromEnforcementPolices(api)