Export Metrics to CSV

import csv
import os
from datetime import datetime, timedelta
from pathlib import Path

from validio_sdk.client import Client


async def example(
    validator_id: str = os.environ["EXAMPLE_VALIDATOR_ID"],
    segment_id: str = os.environ["EXAMPLE_SEGMENT_ID"],
):
    client = Client()
    query = """
    query ValidatorSegmentMetrics($input: ValidatorSegmentMetricsInput!) {
        validatorSegmentMetrics(input: $input) {
            ... on ValidatorMetricWithDifferenceThresholdHistory {
                values {
                    startTime
                    endTime
                    isIncident
                    diffUpperBound: upperBound
                    diffLowerBound: lowerBound
                    value
                }
            }
            ... on ValidatorMetricWithDynamicThresholdHistory {
                values {
                    startTime
                    endTime
                    isIncident
                    decisionBoundsType
                    upperBound
                    lowerBound
                    value
                }
            }
            ... on ValidatorMetricWithFixedThresholdHistory {
                values {
                    startTime
                    endTime
                    isIncident
                    operator
                    bound
                    value
                }
            }
        }
    }
    """

    with Path("metrics.csv").open("w", encoding="utf-8") as csv_file:
        writer = csv.writer(csv_file)
        result_keys = []

        # Fetch metrics for the last 7 days, one day at a time to account for
        # response limits.
        for days_ago in range(7, 0, -1):
            start = datetime.now() - timedelta(days=days_ago)
            end = datetime.now() - timedelta(days=days_ago - 1)

            result = await client.execute(
                query,
                variable_values={
                    "input": {
                        "validatorId": validator_id,
                        "segmentId": segment_id,
                        "timeRange": {
                            "start": start.isoformat(),
                            "end": end.isoformat(),
                        },
                    },
                },
            )

            print(f"Fetching metrics between {start.isoformat()} and {end.isoformat()}")

            for data in result["validatorSegmentMetrics"]["values"]:
                # If we haven't seen any result yet, we don't know which fields
                # the result will have. Grab them and write the header for the
                # CSV file.
                if not result_keys:
                    result_keys = data.keys()
                    writer.writerow(result_keys)

                csv_row = data.values()
                writer.writerow(csv_row)

{
  "validatorSegmentMetrics": {
    "values": [
      {
        "startTime": "2024-01-01T12:00:00Z",
        "endTime": "2024-01-02T12:00:00Z",
        "isIncident": true,
        "decisionBoundsType": "UPPER_AND_LOWER",
        "upperBound": 80.0,
        "lowerBound": 60.5,
        "value": 99.9
      }
    ]
  }
}