From 65f21891db75ce63c4d540da838899b772945427 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Fri, 14 Jan 2022 10:10:40 +0100 Subject: [PATCH] Move list to dict in repetier discovery (#64001) * Move list to dict in repetier discovery * Add setup type hints to repetier --- homeassistant/components/repetier/__init__.py | 4 +++- homeassistant/components/repetier/sensor.py | 16 +++++++++++++--- 2 files changed, 16 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/repetier/__init__.py b/homeassistant/components/repetier/__init__.py index d927a6d854dd..c657d5d05021 100644 --- a/homeassistant/components/repetier/__init__.py +++ b/homeassistant/components/repetier/__init__.py @@ -315,4 +315,6 @@ class PrinterAPI: if not sensor_info: return - load_platform(self._hass, "sensor", DOMAIN, sensor_info, self.config) + load_platform( + self._hass, "sensor", DOMAIN, {"sensors": sensor_info}, self.config + ) diff --git a/homeassistant/components/repetier/sensor.py b/homeassistant/components/repetier/sensor.py index 8ab05ee058ce..f3341cf5e612 100644 --- a/homeassistant/components/repetier/sensor.py +++ b/homeassistant/components/repetier/sensor.py @@ -1,18 +1,27 @@ """Support for monitoring Repetier Server Sensors.""" +from __future__ import annotations + from datetime import datetime import logging import time from homeassistant.components.sensor import SensorDeviceClass, SensorEntity -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.dispatcher import async_dispatcher_connect +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import REPETIER_API, SENSOR_TYPES, UPDATE_SIGNAL, RepetierSensorEntityDescription _LOGGER = logging.getLogger(__name__) -def setup_platform(hass, config, add_entities, discovery_info=None): +def setup_platform( + hass: HomeAssistant, + config: ConfigType, + add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Set up the available Repetier Server sensors.""" if discovery_info is None: return @@ -27,8 +36,9 @@ def setup_platform(hass, config, add_entities, discovery_info=None): "job_start": RepetierJobStartSensor, } + sensors_info: list[dict] = discovery_info["sensors"] entities = [] - for info in discovery_info: + for info in sensors_info: printer_name = info["printer_name"] api = hass.data[REPETIER_API][printer_name] printer_id = info["printer_id"]