From 91c47264b51067169bfaa7fac6462e8427a9d772 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 13 Jan 2022 07:00:14 +0100 Subject: [PATCH] Add setup type hints to rfxtrx (#63963) Co-authored-by: epenet --- homeassistant/components/rfxtrx/binary_sensor.py | 14 ++++++++------ homeassistant/components/rfxtrx/cover.py | 14 ++++++++------ homeassistant/components/rfxtrx/light.py | 14 ++++++++------ homeassistant/components/rfxtrx/sensor.py | 14 ++++++++------ homeassistant/components/rfxtrx/switch.py | 14 ++++++++------ 5 files changed, 40 insertions(+), 30 deletions(-) diff --git a/homeassistant/components/rfxtrx/binary_sensor.py b/homeassistant/components/rfxtrx/binary_sensor.py index dfc8d25ff5e..ef315c7b974 100644 --- a/homeassistant/components/rfxtrx/binary_sensor.py +++ b/homeassistant/components/rfxtrx/binary_sensor.py @@ -10,9 +10,11 @@ from homeassistant.components.binary_sensor import ( BinarySensorEntity, BinarySensorEntityDescription, ) +from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_COMMAND_OFF, CONF_COMMAND_ON, STATE_ON -from homeassistant.core import CALLBACK_TYPE, callback +from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback from homeassistant.helpers import event as evt +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import DeviceTuple, RfxtrxEntity, async_setup_platform_entry, get_pt2262_cmd from .const import ( @@ -83,10 +85,10 @@ def supported(event: rfxtrxmod.RFXtrxEvent): async def async_setup_entry( - hass, - config_entry, - async_add_entities, -): + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up config entry.""" def get_sensor_description(type_string: str): @@ -96,7 +98,7 @@ async def async_setup_entry( def _constructor( event: rfxtrxmod.RFXtrxEvent, - auto: bool, + auto: rfxtrxmod.RFXtrxEvent | None, device_id: DeviceTuple, entity_info: dict, ): diff --git a/homeassistant/components/rfxtrx/cover.py b/homeassistant/components/rfxtrx/cover.py index c8920ccf97b..80b58b6f0c4 100644 --- a/homeassistant/components/rfxtrx/cover.py +++ b/homeassistant/components/rfxtrx/cover.py @@ -14,8 +14,10 @@ from homeassistant.components.cover import ( SUPPORT_STOP_TILT, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry from homeassistant.const import STATE_OPEN -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import ( DEFAULT_SIGNAL_REPETITIONS, @@ -41,15 +43,15 @@ def supported(event: rfxtrxmod.RFXtrxEvent): async def async_setup_entry( - hass, - config_entry, - async_add_entities, -): + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up config entry.""" def _constructor( event: rfxtrxmod.RFXtrxEvent, - auto: bool, + auto: rfxtrxmod.RFXtrxEvent | None, device_id: DeviceTuple, entity_info: dict, ): diff --git a/homeassistant/components/rfxtrx/light.py b/homeassistant/components/rfxtrx/light.py index 981dca37e57..6baada1fb75 100644 --- a/homeassistant/components/rfxtrx/light.py +++ b/homeassistant/components/rfxtrx/light.py @@ -10,8 +10,10 @@ from homeassistant.components.light import ( SUPPORT_BRIGHTNESS, LightEntity, ) +from homeassistant.config_entries import ConfigEntry from homeassistant.const import STATE_ON -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import ( DEFAULT_SIGNAL_REPETITIONS, @@ -35,15 +37,15 @@ def supported(event: rfxtrxmod.RFXtrxEvent): async def async_setup_entry( - hass, - config_entry, - async_add_entities, -): + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up config entry.""" def _constructor( event: rfxtrxmod.RFXtrxEvent, - auto: bool, + auto: rfxtrxmod.RFXtrxEvent | None, device_id: DeviceTuple, entity_info: dict, ): diff --git a/homeassistant/components/rfxtrx/sensor.py b/homeassistant/components/rfxtrx/sensor.py index 623730f1c91..3111f0eabd7 100644 --- a/homeassistant/components/rfxtrx/sensor.py +++ b/homeassistant/components/rfxtrx/sensor.py @@ -13,6 +13,7 @@ from homeassistant.components.sensor import ( SensorEntityDescription, SensorStateClass, ) +from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( DEGREE, ELECTRIC_CURRENT_AMPERE, @@ -28,8 +29,9 @@ from homeassistant.const import ( TEMP_CELSIUS, UV_INDEX, ) -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.entity import EntityCategory +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import DeviceTuple, RfxtrxEntity, async_setup_platform_entry, get_rfx_object from .const import ATTR_EVENT @@ -208,10 +210,10 @@ SENSOR_TYPES_DICT = {desc.key: desc for desc in SENSOR_TYPES} async def async_setup_entry( - hass, - config_entry, - async_add_entities, -): + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up config entry.""" def _supported(event): @@ -219,7 +221,7 @@ async def async_setup_entry( def _constructor( event: RFXtrxEvent, - auto: bool, + auto: RFXtrxEvent | None, device_id: DeviceTuple, entity_info: dict, ): diff --git a/homeassistant/components/rfxtrx/switch.py b/homeassistant/components/rfxtrx/switch.py index b62164a0cf3..988beaa8fb6 100644 --- a/homeassistant/components/rfxtrx/switch.py +++ b/homeassistant/components/rfxtrx/switch.py @@ -6,8 +6,10 @@ import logging import RFXtrx as rfxtrxmod from homeassistant.components.switch import SwitchEntity +from homeassistant.config_entries import ConfigEntry from homeassistant.const import STATE_ON -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import ( DEFAULT_SIGNAL_REPETITIONS, @@ -34,15 +36,15 @@ def supported(event): async def async_setup_entry( - hass, - config_entry, - async_add_entities, -): + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up config entry.""" def _constructor( event: rfxtrxmod.RFXtrxEvent, - auto: bool, + auto: rfxtrxmod.RFXtrxEvent | None, device_id: DeviceTuple, entity_info: dict, ):