Cleanup

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