diff --git a/AirlockTools.py b/AirlockTools.py index d51d7fd..9180503 100644 --- a/AirlockTools.py +++ b/AirlockTools.py @@ -167,7 +167,7 @@ def menu_otp(): def find_unexcluded(): while True: print("\n--- Follow Steps Sequentially, Files will land in directory named 'exclusions'") - print("1. Pull last 24 hours execution for all ATPolicys") + print("1. Pull last 24-72 hours execution for all ATPolicys") print("2. Generate Paths") print("3. Merge") print("4. Pull existing paths") @@ -185,11 +185,24 @@ def find_unexcluded(): # List to hold each policy's DataFrame all_dfs = [] + while True: + try: + history_days = int(input("Enter the how many days in of history do you want to pull - select a number between 1 and 150: ")) + if 1 <= history_days <= 3: + break + else: + print("Invalid input. Please enter a number between 1 and 3.") + except ValueError: + print("Invalid input. Please enter a valid integer.") + + + + # Loop through each policy name for policy_name in atpolicies: try: # Get the policy info DataFrame - df = utils.policyfunctions.getPolicyInfo(url, policy_name, [1, 2, 6, 7], 1, False) + df = utils.policyfunctions.getPolicyInfo(url, policy_name, [1, 2, 6, 7], history_days, False) # Add a column to indicate the policy name df['PolicyName'] = policy_name