Added Move Agent, UI Tweaks

services/agenthandler.py

@@ -24,6 +24,7 @@ from typing import List
 
 import pandas as pd
 
+from flows.prepPolicy import selectPolicies
 from models.agent import Agent
 from models.policy import Policy
 from services.API import AirlockAPIWrapper
@@ -241,6 +242,7 @@ def selectAgents(api: 'AirlockAPIWrapper') -> List['Agent']:
     enrich_agents(matched_agents, policies)
     return matched_agents
 
+
 def moveAgentToRelatedPolicy(
     api: AirlockAPIWrapper,
     agent: Agent,
@@ -284,4 +286,38 @@ def moveAgentToRelatedPolicy(
         return
 
     result = api.agent_move(agent.agentid, target_policy)
-    return result
+    return result
+
+
+def toggleEnforcement(api: AirlockAPIWrapper):
+    choices = ["Audit", "Enforcement", "Exit"]
+    print(colorText("Move devices to which state?:", "yellow"))
+    direction = Selector.select_string(choices, False, False)
+    if direction == "Exit":
+        pass
+    else:
+        devices = selectAgents(api)
+        for device in devices:
+            print(device.hostname)
+        confirm = Selector.confirm("Would you like to continue with these devices? Y/N: ")
+        if direction and devices and confirm:
+            for device in devices:
+                result = moveAgentToRelatedPolicy(api,device, str(direction).lower())
+                logger.info(f"{device.hostname}: result: {result}")
+    get_sanitized_input("Press enter to continue")
+
+def moveAgents(api: AirlockAPIWrapper):
+    devices = selectAgents(api)
+    for device in devices:
+        print(device.hostname)
+    confirm_devices = Selector.confirm("Would you like to continue with these devices? Y/N: ")
+    if devices and confirm_devices:
+        policies = selectPolicies(api, False)
+        confirm_move = Selector.confirm(f"Would you like to move these devices to {policies[0].name}?")
+        if confirm_move:
+            for device in devices:
+                result = api.agent_move(device.agentid, policies[0].groupid)
+                logger.info(f"{device.hostname}: result: {result}")
+        else:
+            logger.info("Exiting without change")
+    get_sanitized_input("Press enter to continue")