Major UI Improvment with blessing

flows/otp.py

@@ -15,9 +15,9 @@
 
 
 
+from datetime import datetime
 import logging
 import os
-from datetime import datetime
 
 import pandas as pd
 
@@ -30,9 +30,7 @@ from utils.utils import colorText, get_sanitized_input
 logger = logging.getLogger(__name__)
 
 
-
-
-def generate(api: AirlockAPIWrapper):
+def otp_generate(api: AirlockAPIWrapper):
     otp_dict = {}
     agents = selectAgents(api)
     print(colorText("Would you like to continue with these devices?","white"))
@@ -59,7 +57,9 @@ def generate(api: AirlockAPIWrapper):
                 logger.info(f"Generated OTP for {agent.hostname}: {otp_code}")
                 otp_dict[agent.hostname] = otp_code
 
-    return otp_dict
+    print(colorText("Requested Codes:", "green"))
+    for key, value in otp_dict.items():
+        print(colorText(f"{key} | {value}","green"))
 
 def otp_activities_by_agent(api: AirlockAPIWrapper):
     activeagents = api.otp_find_active()
@@ -128,7 +128,7 @@ def otp_activities_by_agent(api: AirlockAPIWrapper):
 
 
 
-def revoke(api: AirlockAPIWrapper):
+def otp_revoke(api: AirlockAPIWrapper):
 
     activeagents = api.otp_find_active()
     awaitingagents = api.otp_find_awaiting()