Post Black Linting

services/policyhandler.py

@@ -34,7 +34,6 @@ from utils.utils import areYouSure, colorText, get_sanitized_input
 logger = logging.getLogger(__name__)
 
 
-
 def pullPolicyExechistories(
     api: AirlockAPIWrapper,
     policy: Policy,
@@ -72,7 +71,7 @@ def pullPolicyExechistories(
         ) as pbar:
             while True:
                 histories = api.history_logging(
-                    type=type, checkpoint=checkpoint, policy= [policy.name]
+                    type=type, checkpoint=checkpoint, policy=[policy.name]
                 )
 
                 # Ensure histories is a list of dictionaries
@@ -98,13 +97,17 @@ def pullPolicyExechistories(
 
                     # Update checkpoint on last item
                     if index == len(histories) - 1:
-                        checkpoint = history_item["checkpoint"]  # pyright: ignore[reportArgumentType]
+                        checkpoint = history_item[
+                            "checkpoint"
+                        ]  # pyright: ignore[reportArgumentType]
                         filebar.desc = f"Checkpoint Progress: {checkpoint}"
                         break
 
                     try:
                         history_date = datetime.datetime.strptime(
-                            history_item["datetime"].replace(" +0000 UTC", ""),  # pyright: ignore[reportArgumentType]
+                            history_item["datetime"].replace(
+                                " +0000 UTC", ""
+                            ),  # pyright: ignore[reportArgumentType]
                             "%Y-%m-%dT%H:%M:%SZ",
                         ).date()
                     except ValueError:
@@ -178,7 +181,7 @@ def pullPolicyExechistories(
 
 def getPolicyInfo(api: AirlockAPIWrapper, policy, type, days):
     import airlock_libs
-    
+
     executionhist_policy = pd.DataFrame()
     exehist = airlock_libs.pull_policy_exec_histories(api, policy.name, str(type), days)
     if exehist is not None:
@@ -205,7 +208,7 @@ def getPolicyInfo(api: AirlockAPIWrapper, policy, type, days):
             executionhist_policy = executionhist_policy.sort_values(
                 by=["sha256", "filename"]
             )
-        logger.debug( f"Staging of Execution history for policy: {policy} is complete")    
+        logger.debug(f"Staging of Execution history for policy: {policy} is complete")
         print(
             colorText(
                 f"Staging of Execution history for policy: {policy} is complete",
@@ -237,10 +240,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)
+        updateAuditPoliciesFromEnforcementPolices(api)