Number of Incidents by Severity

import os
from datetime import datetime, timedelta

from validio_sdk.client import Client


async def example(
    source_id: str = os.environ["EXAMPLE_SOURCE_ID"],
):
    client = Client()
    query = """
    query GetIncidents(
      $filter: IncidentGroupsFilter,
      $pagination: PaginationInput
    ) {
      incidentGroups(filter: $filter, pagination: $pagination) {
        elements {
          severityStats {
            highCount
            lowCount
            mediumCount
            totalCount
          }
        }
        pageInfo {
          endCursor
          hasNextPage
        }
      }
    }
    """

    has_next_page = True
    end_cursor = None
    start = datetime.now() - timedelta(days=7)
    end = datetime.now()
    severities = {}

    while has_next_page:
        result = (
            await client.execute(
                query,
                variable_values={
                    "filter": {
                        "source": {
                            "values": [source_id],
                        },
                        "range": {
                            "start": start.isoformat(),
                            "end": end.isoformat(),
                        },
                    },
                    "pagination": {
                        "after": end_cursor,
                        "limit": 1000,
                    },
                },
            )
        )["incidentGroups"]

        for element in result["elements"]:
            if not severities:
                severities = element["severityStats"]
                continue

            for severity, count in element["severityStats"].items():
                severities[severity] += count

        has_next_page = result["pageInfo"]["hasNextPage"]
        end_cursor = result["pageInfo"]["endCursor"]

    for severity, count in severities.items():
        print(f"{severity}: {count}")