mysticmomba/AirlockTools
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

Workflow upgrade for Move Audit/Enforcement

Browse Source
This commit is contained in:
Zarithas
2025-10-27 12:56:40 -04:00
parent f7ae588502
commit 07021fe072
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
services/agenthandler.py
+2 -1
View File
@@ -283,4 +283,5 @@ def moveAgentToRelatedPolicy(
logger.error(f"Unknown mode '{mode}'. Use 'audit' or 'enforcement'.") logger.error(f"Unknown mode '{mode}'. Use 'audit' or 'enforcement'.")
return return
api.agent_move(agent.agentid, target_policy) result = api.agent_move(agent.agentid, target_policy)
return result
utils/menus.py
+5 -3
View File
@@ -72,10 +72,10 @@ def menu_main(api: AirlockAPIWrapper):
elif choice == "2": elif choice == "2":
menu_otp(api) menu_otp(api)
elif choice == "3": elif choice == "3":
choices = ["audit", "enforcement", "Cancel"] choices = ["audit", "enforcement", "Exit"]
print(colorText("Move devices to which state?:", "yellow")) print(colorText("Move devices to which state?:", "yellow"))
direction = Selector.select_string(choices, False, False) direction = Selector.select_string(choices, False, False)
if direction == "Cancel": if direction == "Exit":
pass pass
else: else:
devices = selectAgents(api) devices = selectAgents(api)
@@ -85,7 +85,9 @@ def menu_main(api: AirlockAPIWrapper):
confirm = Selector.confirm() confirm = Selector.confirm()
if direction and devices and confirm: if direction and devices and confirm:
for device in devices: for device in devices:
moveAgentToRelatedPolicy(api,device, str(direction)) result = moveAgentToRelatedPolicy(api,device, str(direction))
logger.info(f"{device.hostname}: result: {result}")
get_sanitized_input("Press enter to continue")
elif choice == "4": elif choice == "4":
findAgents(api,False) findAgents(api,False)
elif choice == "5": elif choice == "5":