diff --git a/mapping_provider/backends/brian.py b/mapping_provider/backends/brian.py index 3f5febfa5f82cd2a390e386d3b06ad3301330437..d3f179ba8115e4aa81b56cc316fd604b2cd1681c 100644 --- a/mapping_provider/backends/brian.py +++ b/mapping_provider/backends/brian.py @@ -131,27 +131,28 @@ def _load_scid_rates_rows( def load_scid_rates(influx_params: config.InfluxConnectionParams) -> list[dict[str, Any]]: rates = [] - for r in _load_scid_rates_rows(influx_params): - def _bitrate_or_none(field_name: str) -> float | None: - _rate = r.get(field_name, None) - assert isinstance(_rate, float | None) # mypy noise - if _rate: - return _rate * 8 - return _rate # could be 0 or None + def _bitrate_or_none(row: dict[str, Any], field_name: str) -> float | None: + _rate = row.get(field_name, None) + assert isinstance(_rate, float | None) # mypy noise + if _rate: + return _rate * 8 + return _rate # could be 0 or None + + for r in _load_scid_rates_rows(influx_params): values = { 'latest': { - 'ingress': _bitrate_or_none('latest_ingress'), - 'egress': _bitrate_or_none('latest_egress') + 'ingress': _bitrate_or_none(r, 'latest_ingress'), + 'egress': _bitrate_or_none(r, 'latest_egress') }, 'mean': { - 'ingress': _bitrate_or_none('mean_ingress'), - 'egress': _bitrate_or_none('mean_egress') + 'ingress': _bitrate_or_none(r, 'mean_ingress'), + 'egress': _bitrate_or_none(r, 'mean_egress') }, 'max': { - 'ingress': _bitrate_or_none('max_ingress'), - 'egress': _bitrate_or_none('max_egress') + 'ingress': _bitrate_or_none(r, 'max_ingress'), + 'egress': _bitrate_or_none(r, 'max_egress') }, } rates.append({