Pretty HTML and beginnings of Destination Policy

utils/allowlist.py

@@ -17,8 +17,9 @@ import requests
 import json
 import os
 import time
-import utils.colortext as ct
+import utils.pretty as ct
 import ijson
+
 def pullPolicyExechistories(url, choice, policiesnames, outputjson: bool): 
     headers = {
         "X-APIKey": os.getenv('APIKEY')

utils/colortext.py

@@ -1,14 +0,0 @@
-
-def colorText(text: str, color: str) -> str:
-    colors = {
-        "red": "\033[91m",
-        "green": "\033[92m",
-        "yellow": "\033[93m",
-        "blue": "\033[94m",
-        "magenta": "\033[95m",
-        "cyan": "\033[96m",
-        "white": "\033[97m",
-		"reset": "\033[0m"
-    }
-    
-    return f"{colors.get(color, colors['reset'])}{text}{colors['reset']}"

utils/getdeviceevents.py

@@ -16,7 +16,7 @@ import datetime
 import requests
 import json
 import os
-import utils.colortext as ct
+import utils.pretty as ct
 
 def devicehistory(url, outputjson: bool):
 	endpoint = url + '/v1/getexechistory'

utils/hashfunctions.py

@@ -125,3 +125,19 @@ def categorizeHashes(aug_df: pd.DataFrame, threat_tolerance: int, untrusted_publ
     unapproved_df = df[~(mask_needsreview | mask_approved)]
 
     return needsreview_df, approved_df, unapproved_df
+
+def destinationbuilder(df, df2):
+    # Step 1: Explode the 'sha256' list in df2 to create one row per sha256 value
+    df_expanded = df.explode('sha256')
+
+    # Step 2: Create a new dataframe for the result
+    df_hashdestination = df_expanded.copy()
+
+    # Step 3: Populate the 'Destination Allowlist' column based on comparison with df3
+    df_hashdestination['Destination Allowlist'] = df_hashdestination['sha256'].apply(
+        lambda x: 'Parent Policy Baseline' if x in df2['sha256'].values else "Destination Policy Allowlist"
+    )
+
+    # Step 4: Return the new dataframe
+    return df_hashdestination
+

utils/pretty.py

@@ -0,0 +1,138 @@
+
+def colorText(text: str, color: str) -> str:
+    colors = {
+        "red": "\033[91m",
+        "green": "\033[92m",
+        "yellow": "\033[93m",
+        "blue": "\033[94m",
+        "magenta": "\033[95m",
+        "cyan": "\033[96m",
+        "white": "\033[97m",
+		"reset": "\033[0m"
+    }
+    
+    return f"{colors.get(color, colors['reset'])}{text}{colors['reset']}"
+
+def style_dataframe_dark(df, output_html_path=None, overwrite=True):
+    from datetime import datetime
+
+    # Get current date and filename for subtitle
+    today = datetime.now().strftime("%d %B %Y")  # Changed to "Day Month Year"
+    filename = output_html_path.replace('.html', '') if output_html_path else "Report"
+
+    dark_css = """
+    <style>
+    body {
+        background-color: #000000;
+        margin: 0;
+        padding: 0;
+        font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+        color: #f8f8f2;
+    }
+    .header {
+        text-align: center;
+        margin: 20px auto;
+        padding: 10px;
+        border-bottom: 2px solid #ffd700;
+        max-width: 95%;
+    }
+    .header h1 {
+        color: #ffd700;
+        margin: 0;
+        font-size: 32px;
+    }
+    .header p {
+        color: #00bfff;
+        margin: 5px 0 0 0;
+        font-size: 18px;
+    }
+    .table-container {
+        overflow-y: scroll;
+        margin: 0 auto;
+        width: 95%;
+        max-height: calc(80vh - 100px);
+        display: block;
+        border: 1px solid #3a3a4d;
+        margin-bottom: 0;
+    }
+    table {
+        border-collapse: collapse;
+        font-size: 14px;
+        background-color: #1e1e2f;
+        color: #f8f8f2;
+        width: max-content;
+    }
+    th, td {
+        border: 1px solid #3a3a4d;
+        text-align: left;
+        padding: 10px;
+        max-width: 300px;
+        word-wrap: break-word;
+        overflow-wrap: break-word;
+    }
+    /* First column: no wrap */
+    td:nth-child(1), th:nth-child(1) {
+        white-space: nowrap;
+        max-width: none !important;
+        word-wrap: normal !important;
+    }
+    th {
+        background-color: #2e2e40;
+        color: #ffd700;
+        position: sticky;
+        top: 0;
+        z-index: 10;
+    }
+    tr:nth-child(even) {
+        background-color: #262638;
+    }
+    tr:hover {
+        background-color: #33334d;
+        color: #00bfff;
+    }
+    /* Custom scrollbar styling */
+    .table-container::-webkit-scrollbar {
+        width: 12px;
+    }
+    .table-container::-webkit-scrollbar-track {
+        background: #1e1e2f;
+    }
+    .table-container::-webkit-scrollbar-thumb {
+        background-color: #3a3a4d;
+        border-radius: 6px;
+    }
+    </style>
+    """
+
+    header = f"""
+    <div class="header">
+        <h1>Airlock Tools</h1>
+        <p>{filename} - {today}</p>
+    </div>
+    """
+
+    html_table = df.to_html(index=False, escape=False)
+    styled_html = (
+        f"<html>\n"
+        f"<head><title>Airlock Tools Report</title></head>\n"
+        f"<body>\n"
+        f"{dark_css}\n"
+        f"{header}\n"
+        f"<div class='table-container'>\n"
+        f"  {html_table}\n"
+        f"</div>\n"
+        f"</body>\n"
+        f"</html>"
+    )
+    if output_html_path:
+        with open(output_html_path, "w", encoding="utf-8") as f:
+            f.write(styled_html)
+        print(f"✅ Styled table saved to '{output_html_path}'")
+    elif overwrite:
+        import tempfile
+        temp_path = tempfile.mktemp(suffix=".html")
+        with open(temp_path, "w", encoding="utf-8") as f:
+            f.write(styled_html)
+        print(f"✅ Styled table saved to temporary file: {temp_path}")
+    else:
+        return styled_html