diff --git a/AirlockTools_Client.py b/AirlockTools_Client.py index c80d29a..db572f7 100644 --- a/AirlockTools_Client.py +++ b/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 ) diff --git a/flows/otp.py b/flows/otp.py index d0f5e0d..d299799 100644 --- a/flows/otp.py +++ b/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'] diff --git a/utils/menus.py b/utils/menus.py index f6c24b9..65f25bd 100644 --- a/utils/menus.py +++ b/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__) diff --git a/utils/selector.py b/utils/selector.py index b500efe..4bc6b76 100644 --- a/utils/selector.py +++ b/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__)