Build a client assessment summary

import json
import os
import sys
from collections import Counter
from datetime import datetime, timezone
from urllib.error import HTTPError, URLError
from urllib.parse import urlencode
from urllib.request import Request, urlopen
from uuid import UUID

BASE = "https://api.alignr.io/api/v1"
TOKEN = os.environ["ALIGNR_API_KEY"]


def get(path, params=None):
    query = "?" + urlencode(params) if params else ""
    request = Request(
        BASE + path + query,
        headers={"Authorization": "Bearer " + TOKEN},
    )
    with urlopen(request, timeout=30) as response:
        return json.load(response)


def build_summary():
    clients = []
    page = 1
    while True:
        batch = get("/organizations", {"page": page, "pageSize": 100})
        for client in batch["items"]:
            client_id = str(UUID(client["id"]))
            grid = get("/organizations/" + client_id + "/compliance")
            if grid["organizationId"] != client_id:
                raise ValueError("Response client does not match the request")
            clients.append({
                "organizationId": client_id,
                "organizationName": grid["organizationName"],
                "statusCounts": dict(Counter(row["status"] for row in grid["items"])),
                "hasControlRows": bool(grid["items"]),
                "assessment": grid,
            })
        if page >= batch["totalPages"]:
            break
        page += 1
    return {
        "retrievedAt": datetime.now(timezone.utc).isoformat(),
        "clients": clients,
    }


try:
    result = build_summary()
except HTTPError as error:
    sys.exit("Request failed: HTTP " + str(error.code) + ". No summary produced.")
except (URLError, TimeoutError):
    sys.exit("Network request failed. No summary produced.")
except (KeyError, ValueError, TypeError):
    sys.exit("Unexpected response shape. Check the live API reference.")
else:
    print(json.dumps(result, indent=2))

python3 assessment_summary.py