AirlockTools_Client.py

@@ -32,6 +32,7 @@ from services.API import AirlockAPIWrapper
 from services.security import getAPI
 from utils.setup import get_base_directory, setup
 from utils.utils import irtang
+
 urllib3.disable_warnings(
     urllib3.exceptions.InsecureRequestWarning
 )

flows/otp.py

@@ -17,15 +17,15 @@
 
 import logging
 import os
+from datetime import datetime
+
 import pandas as pd
 
 from services.agenthandler import selectAgents
 from services.API import AirlockAPIWrapper
-from utils.selector import Selector
 from utils.configmanager import load_env
+from utils.selector import Selector
 from utils.utils import colorText, get_sanitized_input
-from datetime import datetime
-from services.agenthandler import selectAgents
 
 logger = logging.getLogger(__name__)
 
@@ -159,7 +159,6 @@ def revoke(api: AirlockAPIWrapper):
         columns=['otpid', 'hostname', 'status','purpose','granted'],
         header="OTP Sessions"
     )
-    combined_df = pd.DataFrame()
 
     for row in selected_rows:
         otpid = row['otpid']

utils/menus.py

@@ -35,13 +35,13 @@ from utils.configmanager import load_env
 from utils.selector import Selector
 from utils.utils import (
     areYouSure,
+    clear_screen,
     colorText,
     displayIntro,
     get_sanitized_input,
     locked,
     open_directory,
     printEnforceChecklist,
-    clear_screen
 )
 
 logger = logging.getLogger(__name__)

utils/selector.py

@@ -1,6 +1,8 @@
 import logging
 from typing import Any, Callable, List, Optional, Union
+
 import pandas as pd
+
 from utils.utils import colorText, get_sanitized_input
 
 logger = logging.getLogger(__name__)