Post Black Linting

services/API.py

@@ -23,7 +23,6 @@ import requests
 logger = logging.getLogger(__name__)
 
 
-
 class AirlockAPIWrapper:
     """
     A wrapper class for interacting with the Airlock API.
@@ -141,25 +140,25 @@ class AirlockAPIWrapper:
         payload = {"status": "0"}
         result = self._post("/v1/otp/usage", payload)
         return pd.DataFrame(result["response"]["otpusage"])
-    
+
     def otp_find_enforced(self) -> pd.DataFrame:
         """Find OTPs that are awaiting activation."""
         payload = {"status": "2"}
         result = self._post("/v1/otp/usage", payload)
         return pd.DataFrame(result["response"]["otpusage"])
-    
+
     def otp_find_revoked(self) -> pd.DataFrame:
         """Find OTPs that are awaiting activation."""
         payload = {"status": "3"}
         result = self._post("/v1/otp/usage", payload)
         return pd.DataFrame(result["response"]["otpusage"])
-    
+
     def otp_find_by_agent(self, agentid) -> pd.DataFrame:
         """Find OTP by agent."""
         payload = {"agentid": agentid}
         result = self._post("/v1/otp/usage", payload)
         return pd.DataFrame(result["response"]["otpusage"])
-   
+
     def otp_generate(self, agentid: str, duration: int, purpose: str) -> str:
         """Generate a new OTP for an agent."""
         payload = {
@@ -175,7 +174,7 @@ class AirlockAPIWrapper:
         payload = {"otpid": otpid}
         result = self._post("/v1/otp/activities", payload)
         return pd.DataFrame(result["response"]["otpactivities"])
-        
+
     def otp_revoke(self, otpid: str) -> dict:
         """
         Revoke an active OTP.
@@ -186,18 +185,17 @@ class AirlockAPIWrapper:
         """
         payload = {"otpid": otpid}
         return self._post("/v1/otp/revoke", payload)
-    
-    def otp_validate(self, otpcode: str) -> dict:
-            """
-            Validate an OTP code.
-            Parameters:
-            - otpcode (str): The OTP code to validate.
-            Returns:
-            - dict: JSON response indicating validity.
-            """
-            payload = {"otpcode": otpcode}
-            return self._post("/v1/otp/validate", payload)
 
+    def otp_validate(self, otpcode: str) -> dict:
+        """
+        Validate an OTP code.
+        Parameters:
+        - otpcode (str): The OTP code to validate.
+        Returns:
+        - dict: JSON response indicating validity.
+        """
+        payload = {"otpcode": otpcode}
+        return self._post("/v1/otp/validate", payload)
 
     # Policy Management
     def policy_add_path_exclusions(self, groupid: str, paths: List[str]) -> dict:
@@ -225,7 +223,7 @@ class AirlockAPIWrapper:
         payload = {"groupid": groupid}
         result = self._post("/v1/group/agents", payload)
         return pd.DataFrame(result["response"]["agents"])
-        
+
     def policy_list_allowlists(self, groupid: str) -> pd.DataFrame:
         """List allowlists assigned to a specific policy group."""
         payload = {"groupid": groupid}
@@ -236,31 +234,37 @@ 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:
+
+    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
-                   }
+        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:
+    def history_logging(
+        self, type: List[str], checkpoint: str, policy: List[str]
+    ) -> str:
         """Retrieve execution history logs."""
         payload = {"type": type, "checkpoint": checkpoint, "policy": policy}
         result = self._post("/v1/logging/exechistories", payload)
         return result["response"]["exechistories"]
 
-    def history_execution(self, today: str, date_selected: str, agent_name: str) -> List[Dict]:
+    def history_execution(
+        self, today: str, date_selected: str, agent_name: str
+    ) -> List[Dict]:
         """
         Retrieve execution history logs.