From ff19bfa6ace97fcdd294fc101549f03080a1446d Mon Sep 17 00:00:00 2001 From: Erik Reid <erik.reid@geant.org> Date: Mon, 26 May 2025 10:20:59 +0200 Subject: [PATCH] ruffed --- mapping_provider/__init__.py | 2 +- mapping_provider/backends/brian.py | 16 +++++++++------- mapping_provider/backends/services.py | 2 +- test/test_utilization.py | 6 ++++-- 4 files changed, 15 insertions(+), 11 deletions(-) diff --git a/mapping_provider/__init__.py b/mapping_provider/__init__.py index 2a9675f..22fda74 100644 --- a/mapping_provider/__init__.py +++ b/mapping_provider/__init__.py @@ -12,7 +12,7 @@ from fastapi import FastAPI from mapping_provider import config, environment from mapping_provider.api import common, map -from mapping_provider.backends import cache, correlator, inventory, brian +from mapping_provider.backends import brian, cache, correlator, inventory logger = logging.getLogger(__name__) diff --git a/mapping_provider/backends/brian.py b/mapping_provider/backends/brian.py index d3f179b..b111a85 100644 --- a/mapping_provider/backends/brian.py +++ b/mapping_provider/backends/brian.py @@ -1,10 +1,11 @@ import contextlib -from influxdb import InfluxDBClient -from threading import Event +import logging import time -from typing import Generator, Any +from collections.abc import Generator +from threading import Event +from typing import Any -import logging +from influxdb import InfluxDBClient from mapping_provider import config from mapping_provider.backends import cache @@ -59,7 +60,8 @@ CACHED_SCID_RATES_SCHEMA = { } @contextlib.contextmanager -def influx_client(influx_params: config.InfluxConnectionParams) -> Generator[InfluxDBClient]: +def influx_client( + influx_params: config.InfluxConnectionParams) -> Generator[InfluxDBClient]: """ context manager for creating an influx db connection @@ -89,7 +91,7 @@ def influx_client(influx_params: config.InfluxConnectionParams) -> Generator[Inf def _load_scid_rates_rows( influx_params: config.InfluxConnectionParams, - window: str = '24h') -> Generator[dict[str, str|float|None], None, None]: + window: str = '24h') -> Generator[dict[str, str | float | None]]: """ Return the count of all fields, grouped by hostname & interface_name. @@ -124,7 +126,7 @@ def _load_scid_rates_rows( for row in values: rate = {'scid': tags['scid']} - for field_name, field_value in zip(columns, row): + for field_name, field_value in zip(columns, row, strict=False): rate[field_name] = field_value yield rate diff --git a/mapping_provider/backends/services.py b/mapping_provider/backends/services.py index a51eb41..d186196 100644 --- a/mapping_provider/backends/services.py +++ b/mapping_provider/backends/services.py @@ -3,7 +3,7 @@ from collections.abc import Generator from pydantic import BaseModel -from . import cache, correlator, inventory, brian +from . import brian, cache, correlator, inventory logger = logging.getLogger(__name__) diff --git a/test/test_utilization.py b/test/test_utilization.py index 666e656..e0fe27a 100644 --- a/test/test_utilization.py +++ b/test/test_utilization.py @@ -1,12 +1,14 @@ import tempfile -from unittest.mock import patch, MagicMock +from unittest.mock import MagicMock, patch import jsonschema -from .common import load_test_data from mapping_provider.backends import brian, cache from mapping_provider.config import InfluxConnectionParams +from .common import load_test_data + + def test_utilization(): with tempfile.TemporaryDirectory() as tmpdir: -- GitLab