diff --git a/AirlockTools.py b/AirlockTools.py index 4ef9518..49bb55b 100644 --- a/AirlockTools.py +++ b/AirlockTools.py @@ -39,7 +39,7 @@ def apivalidation(): _____ .__ .__ __ ___________ .__ / _ \ |__|______| | ____ ____ | | __ \__ ___/___ ____ | | ______ / /_\ \| \_ __ \ | / _ \_/ ___\| |/ / | | / _ \ / _ \| | / ___/ -/ | \ || | \/ |_( <_> ) \___| < | |( <_> | <_> ) |__\___ \ 4 +/ | \ || | \/ |_( <_> ) \___| < | |( <_> | <_> ) |__\___ \ \____|__ /__||__| |____/\____/ \___ >__|_ \ |____| \____/ \____/|____/____ > \/ \/ \/ \/ """, "green")) @@ -196,11 +196,11 @@ def menu_prepare_to_enforce(): if answer in ("yes", "y"): second_policy_tuple = utils.allowlist.listPolicies(url) second_policy = second_policy_tuple[1][second_policy_tuple[0]] - return False + break elif answer in ("no", "n"): second_policy_tuple = first_policy_tuple second_policy = first_policy - return False + break else: print(ct.colorText("Please answer with 'yes' or 'no'.", "red"))