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")

services/security.py

@@ -135,15 +135,15 @@ def getAPI(USERNAME, SERVICE_NAME):
         raise ValueError("Failed to retrieve API key after 3 incorrect attempts.")
     else:
         logging.warning(f"No API key found for user '{USERNAME}' in service '{SERVICE_NAME}'.")
-        api_key = getpass(f"🗝️ No API key found. Please enter your API key for '{SERVICE_NAME}': ").strip()
+        api_key = getpass(f"No API key found. Please enter your API key for '{SERVICE_NAME}': ").strip()
         print("Please exit and relaunch program after saving your credential to avoid errors")
 
         while True:
-            password = getpass("🔓 Create a password to encrypt your API key: ")
-            confirm_password = getpass("🔒 Confirm your password: ")
+            password = getpass("Create a password to encrypt your API key: ")
+            confirm_password = getpass("Confirm your password: ")
 
             if password != confirm_password:
-                logging.warning("❌ Passwords do not match. Try again.")
+                logging.warning("Passwords do not match. Try again.")
                 continue
 
             if check_password_complexity(password):
@@ -155,7 +155,7 @@ def getAPI(USERNAME, SERVICE_NAME):
                     logging.error(f"Failed to store API key: {e}")
                     break
             else:
-                logging.warning("❌ Password does not meet complexity requirements. Try again.")
+                logging.warning("Password does not meet complexity requirements. Try again.")
 
 
 class APIKeyManager: