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

Fixed typo, pulled what appeared to be unused modules

Browse Source
This commit is contained in:
Driven-Element
2025-08-24 14:20:12 -04:00
parent bdf3d4b9ec
commit 6577d5f0e9
1 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
AirlockTools.py
+8 -10
View File
@@ -20,11 +20,9 @@ import utils.allowlist
import utils.hashfunctions
import utils.pathfunctions
import utils.allowfunctions
import utils.colortext as ct
import urllib3
import pandas as pd
import sqlite3
import pathlib
import utils.colortext as ct
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
dotenv.load_dotenv()
@@ -32,7 +30,7 @@ dotenv.load_dotenv()
url = "https://172.17.22.240:3129"
badpublisherlist = ["Brave Software, Inc.", "Zoom Video Communications, Inc."]
path_exclusion_constant = 3
treat_tolerance_constant = 4
threat_tolerance_constant = 4
@@ -56,16 +54,16 @@ def apivalidation():
def menu_main():
while True:
print(ct.colorText("\n-----------------------------------", "yellow"))
print(ct.colorText("------------ Main Menu ------------", "yellow"))
print(ct.colorText("-----------------------------------", "yellow"))
print(ct.colorText("\n-----------------------------------", "white"))
print(ct.colorText("------------ Main Menu ------------", "white"))
print(ct.colorText("-----------------------------------", "white"))
print(ct.colorText("1. Get All Events for Single Device", "yellow"))
print(ct.colorText("2. Placeholder for Local Approval", "yellow"))
print(ct.colorText("3. Placeholder for Another Tool", "yellow"))
print(ct.colorText("4. Prepare Policy For Enforcement", "yellow"))
print(ct.colorText("11. Exit", "yellow"))
choice = input(ct.colorText("Enter Menu Item: ", "white"))
choice = input(ct.colorText("\nEnter Menu Item: ", "white"))
if choice == '1':
utils.getdeviceevents.devicehistory(url,False)
elif choice == "2":
@@ -189,7 +187,7 @@ def menu_prepare_to_enforce():
print(ct.colorText("11. Exit", "cyan"))
choice = input(ct.colorText("Enter your choice: ", "yellow"))
choice = input(ct.colorText("\nEnter your choice: ", "white"))
if choice == "1":
first_policy_tuple = utils.allowlist.listPolicies(url)
first_policy = first_policy_tuple[1][first_policy_tuple[0]]
@@ -256,7 +254,7 @@ def menu_prepare_to_enforce():
elif choice == "6":
if os.path.exists(f"dataframe_csv\\df_path_ineligible_{first_policy}_{second_policy}.csv") == True and os.path.exists(f"dataframe_csv\\df_path_eligible_{first_policy}_{second_policy}.csv") == True:
categorized = utils.hashfunctions.categorizeHashes(pd.read_csv(f"dataframe_csv\\df_augmented_combo_{first_policy}_{second_policy}.csv"), treat_tolerance_constant, badpublisherlist)
categorized = utils.hashfunctions.categorizeHashes(pd.read_csv(f"dataframe_csv\\df_augmented_combo_{first_policy}_{second_policy}.csv"), threat_tolerance_constant, badpublisherlist)
categorized[0].to_html(f"dataframe_html\\df_hashes_needing_approval_{first_policy}_{second_policy}.html", index=False)
categorized[0].to_csv(f"dataframe_csv\\df_hashes_needing_approval_{first_policy}_{second_policy}.csv", index=False)
categorized[1].to_html(f"dataframe_html\\df_automatically_approved_hashes_{first_policy}_{second_policy}.html", index=False)