Post Black Linting

flows/localApproval.py

@@ -41,7 +41,9 @@ def getLocalApprovals(api: AirlockAPIWrapper):
     result = api.otp_find_awaiting()
     local_approval = pd.DataFrame(result["response"]["otpusage"])
     if os.path.exists(f"{base_dir}\\cache\\newest_local_approval.parquet"):
-        previous_run = pd.read_parquet(f"{base_dir}\\cache\\newest_local_approval.parquet")
+        previous_run = pd.read_parquet(
+            f"{base_dir}\\cache\\newest_local_approval.parquet"
+        )
         previous_run.to_parquet(
             f"{base_dir}\\cache\\last_local_approval.parquet", index=False
         )
@@ -66,10 +68,10 @@ def getLocalApprovals(api: AirlockAPIWrapper):
 
 def scheduleAddingLAHashes(api: AirlockAPIWrapper):
 
-    policy_relationship_map = get_protected_json("POLICY_MAP_ENF_AUD","{}")
+    policy_relationship_map = get_protected_json("POLICY_MAP_ENF_AUD", "{}")
     bad_publisher_list = load_env_json("BAD_PUBLISHER", "[]")
     pups = load_env_json("PUPS", "[]")
-    threat_tolerance_constant = load_env("VT_THREAT_TOLERANCE", cast_type = int)
+    threat_tolerance_constant = load_env("VT_THREAT_TOLERANCE", cast_type=int)
 
     try:
         register_function("add_hash", returnFromLocalApproval)
@@ -93,7 +95,9 @@ def scheduleAddingLAHashes(api: AirlockAPIWrapper):
             duration_minutes = int(batch_df["duration"].iloc[0])
             start_time = datetime.datetime.now()
             run_time = start_time + datetime.timedelta(minutes=duration_minutes)
-            early_time = start_time + datetime.timedelta(minutes=np.floor(duration_minutes * 0.95))
+            early_time = start_time + datetime.timedelta(
+                minutes=np.floor(duration_minutes * 0.95)
+            )
 
             early_timestamp = early_time.timestamp()
             run_timestamp = run_time.timestamp()
@@ -148,7 +152,13 @@ def scheduleAddingLAHashes(api: AirlockAPIWrapper):
             logger.warning(f"Failed to process batch {batchid}: {e}")
 
 
-def returnFromLocalApproval(api, device_df, policy_relationship_map, bad_publisher_list, pups, threat_tolerance_constant
+def returnFromLocalApproval(
+    api,
+    device_df,
+    policy_relationship_map,
+    bad_publisher_list,
+    pups,
+    threat_tolerance_constant,
 ):
     """
       # Get unique policy names from device list
@@ -166,11 +176,14 @@ def returnFromLocalApproval(api, device_df, policy_relationship_map, bad_publish
     #TODO finish logic for adding hashes
     """
     working_dir = load_env("WORKING_DIR")
-    policy_relationship_map = get_protected_json("POLICY_MAP_ENF_AUD","{}")
+    policy_relationship_map = get_protected_json("POLICY_MAP_ENF_AUD", "{}")
     bad_publisher_list = load_env_json("BAD_PUBLISHER", "[]")
     pups = load_env_json("PUPS", "[]")
     threat_tolerance_constant = load_env("VT_THREAT_TOLERANCE")
-    print(f"{working_dir}, {policy_relationship_map}, {bad_publisher_list}, {pups}, {threat_tolerance_constant}")
+    print(
+        f"{working_dir}, {policy_relationship_map}, {bad_publisher_list}, {pups}, {threat_tolerance_constant}"
+    )
+
 
 def moveToLocalApproval(api: AirlockAPIWrapper):
     possible_durations = [15, 60, 360, 1440, 10080]
@@ -213,10 +226,9 @@ def moveToLocalApproval(api: AirlockAPIWrapper):
 
 
 def addLocalApproval(api: AirlockAPIWrapper, batchid, duration_selected, agentid):
-    
+
     purpose = f"🎫 Local Approval 🎫 - {duration_selected} mins - batch:{batchid} Client:{agentid}"
     api.otp_generate(agentid, duration_selected, purpose)
- 
 
 
 def monitorAuditStatus(api: AirlockAPIWrapper):
@@ -224,11 +236,13 @@ def monitorAuditStatus(api: AirlockAPIWrapper):
     last_agents = []
     if not last_agents:
         last_agents = current_agents
-    policy_relationship_map = get_protected_json("POLICY_MAP_ENF_AUD","{}")
+    policy_relationship_map = get_protected_json("POLICY_MAP_ENF_AUD", "{}")
 
     # Reverse map for audit → enforcement
     reverse_policy_map = {v: k for k, v in policy_relationship_map.items()}
-    known_transitions = set(policy_relationship_map.items()) | set(reverse_policy_map.items())
+    known_transitions = set(policy_relationship_map.items()) | set(
+        reverse_policy_map.items()
+    )
 
     # Index last_agents by hostname for quick lookup
     last_agent_map = {agent.hostname: agent for agent in last_agents}
@@ -261,8 +275,8 @@ def monitorAuditStatus(api: AirlockAPIWrapper):
 
 
 def getNewLocalApprovals(api: AirlockAPIWrapper):
-    
-    working_dir = load_env("WORKING_DIR")                   
+
+    working_dir = load_env("WORKING_DIR")
     current_la = getLocalApprovals(api)
 
     # Load old approval list
@@ -273,15 +287,21 @@ def getNewLocalApprovals(api: AirlockAPIWrapper):
         old_la = pd.DataFrame(columns=current_la.columns)
 
     # Create composite keys
-    current_la["key"] = current_la["clientid"].astype(str) + "_" + current_la["granted"].astype(str)
+    current_la["key"] = (
+        current_la["clientid"].astype(str) + "_" + current_la["granted"].astype(str)
+    )
     old_la["key"] = old_la["clientid"].astype(str) + "_" + old_la["granted"].astype(str)
 
     # Find new entries
     new_entries = current_la[~current_la["key"].isin(old_la["key"])]
 
     # Convert 'granted' to datetime and filter by last 10 minutes
-    new_entries["granted"] = pd.to_datetime(new_entries["granted"], utc=True, errors="coerce")
-    ten_minutes_ago = datetime.datetime.now(datetime.timezone.utc) - datetime.timedelta(minutes=10)
+    new_entries["granted"] = pd.to_datetime(
+        new_entries["granted"], utc=True, errors="coerce"
+    )
+    ten_minutes_ago = datetime.datetime.now(datetime.timezone.utc) - datetime.timedelta(
+        minutes=10
+    )
     recent_entries = new_entries[new_entries["granted"] > ten_minutes_ago]
 
     # Save current approvals for next run