Generate a Threat Flow Report

import time

from flareio import FlareApiClient


api_client = FlareApiClient.from_env()

# Create an intel request
intel_request_resp = api_client.post(
    "/firework/v4/threat_flow/intel/requests",
    json={
        "question": "xss vulnerability in banks",
        "tone": "analytical",
    },
)
intel_request_resp.raise_for_status()
request_id: str = intel_request_resp.json()["request_id"]
print(f"Created intel request: {request_id=}")

# Poll the intel request until it's completed
report_id: int | None = None
timeout = time.monotonic() + 60 * 15
while time.monotonic() < timeout:
    intel_resp = api_client.get(
        f"/firework/v4/threat_flow/intel/requests/{request_id}",
    )
    intel_resp.raise_for_status()
    intel_data = intel_resp.json()
    status = intel_data["status"]

    if status == "completed":
        report_id = intel_data["result"]["report_id"]
        break
    elif status == "error":
        raise Exception(f"Intel request failed: {intel_data=}")

    print(f"Waiting for intel request to complete ({status=})...")
    time.sleep(5)
else:
    raise Exception("Failed to complete intel request within 15 minutes")

print(f"Report generated: {report_id=}")
print("This report can be retrieved for up to 30 days.")