mysticmomba/AirlockTools
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

Added prompt for number of days

Browse Source
This commit is contained in:
=
2025-09-17 09:39:07 -04:00
parent bfd853fc76
commit bf51d5b916
1 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
AirlockTools.py
+15 -2
View File
@@ -176,6 +176,7 @@ def menu_prepare_to_enforce():
allowlist_parent_id = " " allowlist_parent_id = " "
allowlist_child_name = " " allowlist_child_name = " "
allowlist_child_id = " " allowlist_child_id = " "
history_days = 1
#If the directorys where we're going to store our output dont exist, make them. #If the directorys where we're going to store our output dont exist, make them.
if not os.path.exists("parquet"): os.makedirs("parquet") if not os.path.exists("parquet"): os.makedirs("parquet")
@@ -208,11 +209,23 @@ def menu_prepare_to_enforce():
elif choice == "2": elif choice == "2":
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 <= 150:
break
else:
print("Invalid input. Please enter a number between 1 and 150.")
except ValueError:
print("Invalid input. Please enter a valid integer.")
if not os.path.exists(f"parquet\\execution_history_{first_policy}.parquet"): if not os.path.exists(f"parquet\\execution_history_{first_policy}.parquet"):
utils.policyfunctions.getPolicyInfo(url, first_policy, 150) utils.policyfunctions.getPolicyInfo(url, first_policy, history_days)
if not os.path.exists(f"parquet\\execution_history_{second_policy}.parquet"): if not os.path.exists(f"parquet\\execution_history_{second_policy}.parquet"):
utils.policyfunctions.getPolicyInfo(url, second_policy, 150) utils.policyfunctions.getPolicyInfo(url, second_policy, history_days)
if not os.path.exists(f"parquet\\combined_hashlist_{first_policy}_{second_policy}.parquet"): if not os.path.exists(f"parquet\\combined_hashlist_{first_policy}_{second_policy}.parquet"):
utils.hashfunctions.combineHashes(url, first_policy, second_policy) utils.hashfunctions.combineHashes(url, first_policy, second_policy)