Bugfix, plus some QoL upgrades from the Async branch

2025-10-20 14:25:03 -04:00
parent aa8380ac0a
commit f1f33948e3
7 changed files with 299 additions and 218 deletions
@@ -44,7 +44,11 @@ def generate(api: AirlockAPIWrapper):
        print(colorText("Please select a duration in minutes:  ", "white"))
        print(colorText("15 mins, 60 mins, 360 mins(6 Hours), 1440 mins (24 Hours), 10080 mins (7 Days):", "white"))
        duration_selected = Selector.select_int(possible_durations)
-        if duration_selected:
+
+        if isinstance(duration_selected, list):
+            duration_selected = duration_selected[0] if duration_selected else None
+
+        if duration_selected is not None:
            for agent in agents:
                otp_code = api.otp_generate(agent.agentid, duration_selected, purpose)
                logger.info(f"Generated OTP for {agent.hostname}: {otp_code}")