Policy Tree Tab added

2025-11-03 15:59:08 -05:00
parent 5b9f1ab4fc
commit 70477f3885
2 changed files with 78 additions and 24 deletions
@@ -236,6 +236,22 @@ class AirlockAPIWrapper:
        """Set audit mode for a policy group. 1=Audit, 0=Enforcement"""
        payload = {"groupid": groupid, "auditmode": auditmode}
        return self._post("/v1/group/settings/auditmode", payload)
+    
+    def policy_set_script_custom(self, 
+                                groupid: str, 
+                                script_custom: int, 
+                                scripts_audit: List[str],
+                                scripts_disabled: List[str],
+                                scripts_respect: List[str],
+                                ) -> dict:
+        """Set audit mode for a policy group. 1=Audit, 0=Enforcement"""
+        payload = {"groupid": groupid,
+                   "script_custom": script_custom,
+                   "scripts_audit": scripts_audit,
+                   "scripts_disabled": scripts_disabled,
+                   "scripts_respect": scripts_respect
+                   }
+        return self._post("/v1/group/settings/script_custom", payload)

    # Execution History
    def history_logging(self, type: List[str], checkpoint: str, policy: List[str]) -> str: