AI Tools by Category:
Complete Classification Guide

#!/usr/bin/env python3
"""Deploy category-based AI blocking policies using the API."""

import requests
import json

API_BASE = "https://api.aitoolsblocklist.com/v1"
API_KEY  = "your-enterprise-api-key"

POLICY_PROFILES = {
    "engineering": {
        "block": ["ai-chatbots", "ai-code-assistants", "ai-writing",
                  "ai-data-analytics", "ai-developer-tools"],
        "monitor": ["ai-search", "ai-productivity"],
    },
    "marketing": {
        "block": ["ai-chatbots", "ai-code-assistants", "ai-data-analytics"],
        "monitor": ["ai-writing", "ai-marketing", "ai-design"],
    },
    "legal": {
        "block": ["ai-chatbots", "ai-writing", "ai-legal",
                  "ai-code-assistants", "ai-data-analytics"],
        "monitor": ["ai-search"],
    },
}

def fetch_domains_for_categories(categories: list) -> list:
    """Fetch domain list filtered by categories."""
    resp = requests.get(
        f"{API_BASE}/feed/domains",
        params={
            "categories": ",".join(categories),
            "format": "json",
        },
        headers={"Authorization": f"Bearer {API_KEY}"},
        timeout=30,
    )
    resp.raise_for_status()
    return resp.json()["domains"]

def deploy_policy(profile_name: str):
    """Generate blocklist files per department policy."""
    profile = POLICY_PROFILES[profile_name]
    blocked = fetch_domains_for_categories(profile["block"])
    monitored = fetch_domains_for_categories(profile["monitor"])

    with open(f"/etc/blocklists/{profile_name}-block.txt", "w") as f:
        f.write("\n".join(d["domain"] for d in blocked))
    with open(f"/etc/blocklists/{profile_name}-monitor.txt", "w") as f:
        f.write("\n".join(d["domain"] for d in monitored))

    print(f"Policy '{profile_name}': {len(blocked)} blocked, {len(monitored)} monitored")

# Deploy all department policies
for name in POLICY_PROFILES:
    deploy_policy(name)