diff --git a/.gitignore b/.gitignore index 7862b06..d19b860 100644 --- a/.gitignore +++ b/.gitignore @@ -10,4 +10,5 @@ jobs.json securitytest.py *.toml system_config.json -Devel_unused/ \ No newline at end of file +Devel_unused/ +AirlockTools_client*/ \ No newline at end of file diff --git a/flows/localApproval.py b/flows/localApproval.py index 952f03f..789b917 100644 --- a/flows/localApproval.py +++ b/flows/localApproval.py @@ -37,7 +37,6 @@ dotenv.load_dotenv() def getLocalApprovals(api: AirlockAPIWrapper): - working_dir = load_env("WORKING_DIR") base_dir = get_base_directory result = api.otp_find_awaiting() local_approval = pd.DataFrame(result["response"]["otpusage"]) diff --git a/flows/otp.py b/flows/otp.py index e9348eb..4f5d52b 100644 --- a/flows/otp.py +++ b/flows/otp.py @@ -18,13 +18,13 @@ import logging from services.agenthandler import selectAgents +from services.API import AirlockAPIWrapper from utils.selector import Selector from utils.utils import colorText, get_sanitized_input logger = logging.getLogger(__name__) -from services.API import AirlockAPIWrapper def generate(api: AirlockAPIWrapper): diff --git a/flows/quietAgent.py b/flows/quietAgent.py index 5e02204..6e8a642 100644 --- a/flows/quietAgent.py +++ b/flows/quietAgent.py @@ -16,6 +16,7 @@ import datetime import logging +import dotenv import pandas as pd from flows.prepPolicy import selectPolicies @@ -25,7 +26,7 @@ from utils.selector import Selector from utils.utils import colorText, load_env logger = logging.getLogger(__name__) -import dotenv + dotenv.load_dotenv() diff --git a/utils/selector.py b/utils/selector.py index 6c69df6..87f1bfb 100644 --- a/utils/selector.py +++ b/utils/selector.py @@ -15,18 +15,13 @@ import logging -from typing import Any, List, Optional, Union +from typing import Any, Callable, List, Optional, Union from utils.utils import get_sanitized_input logger = logging.getLogger(__name__) -import logging -from typing import Callable - -logger = logging.getLogger(__name__) - class Selector: @staticmethod def _display_choices(