import os
import datetime
from collections import defaultdict
import customer_db
def generate_report():
customers = customer_db._db.get_all_customers()
events = customer_db._db.get_recent_events(limit=10000)
total_events = len(events)
blocked_events = sum(1 for e in events if e.get('outcome') == 'block')
block_rate = (blocked_events / total_events * 100) if total_events else 0
total_mrr = sum(c.get('mrr', 0) for c in customers)
latencies = [e.get('latency_ms', 0) for e in events if e.get('latency_ms') is not None]
avg_latency = sum(latencies) / len(latencies) if latencies else 0
bots = defaultdict(lambda: {'total': 0, 'blocked': 0})
ips = defaultdict(int)
for e in events:
bot_class = e.get('bot_class') or 'unknown'
bots[bot_class]['total'] += 1
if e.get('outcome') == 'block':
bots[bot_class]['blocked'] += 1
ip_addr = e.get('ip_address') or e.get('ip') or 'unknown'
ips[ip_addr] += 1
top_ips = sorted(ips.items(), key=lambda x: x[1], reverse=True)[:10]
report_dir = "/home/snuffleupagus/teamwork_projects/axiom_zero_audit/reports"
os.makedirs(report_dir, exist_ok=True)
today = datetime.datetime.now().strftime("%Y-%m-%d")
report_path = os.path.join(report_dir, f"axiom_daily_{today}.html")
html = f"""
Axiom Zero - Executive Report
AXIOM ZERO — INTERNAL EXECUTIVE REPORT — CONFIDENTIAL
Block Rate
{block_rate:.1f}%
Total Events
{total_events}
Monthly Revenue
${total_mrr:,}
Avg Latency
{avg_latency:.1f} ms
Customer Status
| Company |
Tier |
MRR |
Credential Status |
"""
for c in customers:
status = c.get("credential_status", "PENDING")
status_class = "status-valid" if status == "VALID" else "status-invalid"
html += f"""
| {c.get('company_name', 'Unknown')} |
{c.get('tier', 'Unknown')} |
${c.get('mrr', 0):,} |
{status} |
"""
html += """
Detection Breakdown
| Bot Class |
Total Events |
Blocked |
Block Rate |
"""
for bot, data in bots.items():
rate = (data['blocked'] / data['total'] * 100) if data['total'] else 0
html += f"""
| {bot} |
{data['total']} |
{data['blocked']} |
{rate:.1f}% |
"""
html += """
Top 10 Threat IPs
| IP Address |
Event Count |
"""
for ip, count in top_ips:
html += f"""
| {ip} |
{count} |
"""
html += f"""
CONFIDENTIAL - GENERATED {datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
"""
with open(report_path, "w") as f:
f.write(html)
print(f"Report generated successfully: {report_path}")
if __name__ == "__main__":
generate_report()