diff --git a/homeassistant/components/xiaomi_miio/air_quality.py b/homeassistant/components/xiaomi_miio/air_quality.py index 4190f49e30c..c96a29a423c 100644 --- a/homeassistant/components/xiaomi_miio/air_quality.py +++ b/homeassistant/components/xiaomi_miio/air_quality.py @@ -3,7 +3,12 @@ from collections.abc import Callable import logging -from miio import AirQualityMonitor, AirQualityMonitorCGDN1, DeviceException +from miio import ( + AirQualityMonitor, + AirQualityMonitorCGDN1, + Device as MiioDevice, + DeviceException, +) from homeassistant.components.air_quality import AirQualityEntity from homeassistant.const import CONF_DEVICE, CONF_HOST, CONF_MODEL, CONF_TOKEN @@ -40,12 +45,17 @@ PROP_TO_ATTR = { class AirMonitorB1(XiaomiMiioEntity, AirQualityEntity): """Air Quality class for Xiaomi cgllc.airmonitor.b1 device.""" - def __init__(self, name, device, entry, unique_id): + def __init__( + self, + name: str, + device: MiioDevice, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the entity.""" super().__init__(name, device, entry, unique_id) self._icon = "mdi:cloud" - self._available = None self._air_quality_index = None self._carbon_dioxide = None self._carbon_dioxide_equivalent = None @@ -170,12 +180,17 @@ class AirMonitorV1(AirMonitorB1): class AirMonitorCGDN1(XiaomiMiioEntity, AirQualityEntity): """Air Quality class for cgllc.airm.cgdn1 device.""" - def __init__(self, name, device, entry, unique_id): + def __init__( + self, + name: str, + device: MiioDevice, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the entity.""" super().__init__(name, device, entry, unique_id) self._icon = "mdi:cloud" - self._available = None self._carbon_dioxide = None self._particulate_matter_2_5 = None self._particulate_matter_10 = None diff --git a/homeassistant/components/xiaomi_miio/entity.py b/homeassistant/components/xiaomi_miio/entity.py index f8cdc69a12e..bb4e68f9f71 100644 --- a/homeassistant/components/xiaomi_miio/entity.py +++ b/homeassistant/components/xiaomi_miio/entity.py @@ -27,7 +27,13 @@ _LOGGER = logging.getLogger(__name__) class XiaomiMiioEntity(Entity): """Representation of a base Xiaomi Miio Entity.""" - def __init__(self, name, device, entry, unique_id): + def __init__( + self, + name: str, + device: MiioDevice, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the Xiaomi Miio Device.""" self._device = device self._model = entry.data[CONF_MODEL] @@ -35,7 +41,7 @@ class XiaomiMiioEntity(Entity): self._device_id = entry.unique_id self._unique_id = unique_id self._name = name - self._available = None + self._available = False @property def unique_id(self): @@ -50,6 +56,8 @@ class XiaomiMiioEntity(Entity): @property def device_info(self) -> DeviceInfo: """Return the device info.""" + if TYPE_CHECKING: + assert self._device_id is not None device_info = DeviceInfo( identifiers={(DOMAIN, self._device_id)}, manufacturer="Xiaomi", diff --git a/homeassistant/components/xiaomi_miio/light.py b/homeassistant/components/xiaomi_miio/light.py index 03341ea9541..f452c704db2 100644 --- a/homeassistant/components/xiaomi_miio/light.py +++ b/homeassistant/components/xiaomi_miio/light.py @@ -259,15 +259,21 @@ class XiaomiPhilipsAbstractLight(XiaomiMiioEntity, LightEntity): _attr_color_mode = ColorMode.BRIGHTNESS _attr_supported_color_modes = {ColorMode.BRIGHTNESS} + _device: Ceil | PhilipsBulb | PhilipsEyecare | PhilipsMoonlight - def __init__(self, name, device, entry, unique_id): + def __init__( + self, + name: str, + device: Ceil | PhilipsBulb | PhilipsEyecare | PhilipsMoonlight, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the light device.""" super().__init__(name, device, entry, unique_id) self._brightness = None - self._available = False self._state = None - self._state_attrs = {} + self._state_attrs: dict[str, Any] = {} @property def available(self) -> bool: @@ -348,7 +354,15 @@ class XiaomiPhilipsAbstractLight(XiaomiMiioEntity, LightEntity): class XiaomiPhilipsGenericLight(XiaomiPhilipsAbstractLight): """Representation of a Generic Xiaomi Philips Light.""" - def __init__(self, name, device, entry, unique_id): + _device: Ceil | PhilipsBulb | PhilipsEyecare | PhilipsMoonlight + + def __init__( + self, + name: str, + device: Ceil | PhilipsBulb | PhilipsEyecare | PhilipsMoonlight, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the light device.""" super().__init__(name, device, entry, unique_id) @@ -390,7 +404,7 @@ class XiaomiPhilipsGenericLight(XiaomiPhilipsAbstractLight): """Set delayed turn off.""" await self._try_command( "Setting the turn off delay failed.", - self._device.delay_off, + self._device.delay_off, # type: ignore[union-attr] time_period.total_seconds(), ) @@ -421,12 +435,19 @@ class XiaomiPhilipsBulb(XiaomiPhilipsGenericLight): _attr_color_mode = ColorMode.COLOR_TEMP _attr_supported_color_modes = {ColorMode.COLOR_TEMP} + _device: Ceil | PhilipsBulb | PhilipsMoonlight - def __init__(self, name, device, entry, unique_id): + def __init__( + self, + name: str, + device: Ceil | PhilipsBulb | PhilipsMoonlight, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the light device.""" super().__init__(name, device, entry, unique_id) - self._color_temp = None + self._color_temp: int | None = None @property def _current_mireds(self): @@ -575,7 +596,15 @@ class XiaomiPhilipsBulb(XiaomiPhilipsGenericLight): class XiaomiPhilipsCeilingLamp(XiaomiPhilipsBulb): """Representation of a Xiaomi Philips Ceiling Lamp.""" - def __init__(self, name, device, entry, unique_id): + _device: Ceil + + def __init__( + self, + name: str, + device: Ceil, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the light device.""" super().__init__(name, device, entry, unique_id) @@ -635,7 +664,15 @@ class XiaomiPhilipsCeilingLamp(XiaomiPhilipsBulb): class XiaomiPhilipsEyecareLamp(XiaomiPhilipsGenericLight): """Representation of a Xiaomi Philips Eyecare Lamp 2.""" - def __init__(self, name, device, entry, unique_id): + _device: PhilipsEyecare + + def __init__( + self, + name: str, + device: PhilipsEyecare, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the light device.""" super().__init__(name, device, entry, unique_id) @@ -748,7 +785,15 @@ class XiaomiPhilipsEyecareLamp(XiaomiPhilipsGenericLight): class XiaomiPhilipsEyecareLampAmbientLight(XiaomiPhilipsAbstractLight): """Representation of a Xiaomi Philips Eyecare Lamp Ambient Light.""" - def __init__(self, name, device, entry, unique_id): + _device: PhilipsEyecare + + def __init__( + self, + name: str, + device: PhilipsEyecare, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the light device.""" name = f"{name} Ambient Light" if unique_id is not None: @@ -807,12 +852,19 @@ class XiaomiPhilipsMoonlightLamp(XiaomiPhilipsBulb): """Representation of a Xiaomi Philips Zhirui Bedside Lamp.""" _attr_supported_color_modes = {ColorMode.COLOR_TEMP, ColorMode.HS} + _device: PhilipsMoonlight - def __init__(self, name, device, entry, unique_id): + def __init__( + self, + name: str, + device: PhilipsMoonlight, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the light device.""" super().__init__(name, device, entry, unique_id) - self._hs_color = None + self._hs_color: tuple[float, float] | None = None self._state_attrs.pop(ATTR_DELAYED_TURN_OFF) self._state_attrs.update( { diff --git a/homeassistant/components/xiaomi_miio/sensor.py b/homeassistant/components/xiaomi_miio/sensor.py index c0631d66e56..9088dbb3a06 100644 --- a/homeassistant/components/xiaomi_miio/sensor.py +++ b/homeassistant/components/xiaomi_miio/sensor.py @@ -925,11 +925,19 @@ class XiaomiGenericSensor( class XiaomiAirQualityMonitor(XiaomiMiioEntity, SensorEntity): """Representation of a Xiaomi Air Quality Monitor.""" - def __init__(self, name, device, entry, unique_id, description): + _device: AirQualityMonitor + + def __init__( + self, + name: str, + device: AirQualityMonitor, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + description: XiaomiMiioSensorDescription, + ) -> None: """Initialize the entity.""" super().__init__(name, device, entry, unique_id) - self._available = None self._state = None self._state_attrs = { ATTR_POWER: None, diff --git a/homeassistant/components/xiaomi_miio/switch.py b/homeassistant/components/xiaomi_miio/switch.py index 6711c45922b..2bd9e406a14 100644 --- a/homeassistant/components/xiaomi_miio/switch.py +++ b/homeassistant/components/xiaomi_miio/switch.py @@ -803,13 +803,20 @@ class XiaomiGatewaySwitch(XiaomiGatewayDevice, SwitchEntity): class XiaomiPlugGenericSwitch(XiaomiMiioEntity, SwitchEntity): """Representation of a Xiaomi Plug Generic.""" - def __init__(self, name, device, entry, unique_id): + _device: AirConditioningCompanionV3 | ChuangmiPlug | PowerStrip + + def __init__( + self, + name: str, + device: AirConditioningCompanionV3 | ChuangmiPlug | PowerStrip, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the plug switch.""" super().__init__(name, device, entry, unique_id) self._icon = "mdi:power-socket" - self._available = False - self._state = None + self._state: bool | None = None self._state_attrs = {ATTR_TEMPERATURE: None, ATTR_MODEL: self._model} self._device_features = FEATURE_FLAGS_GENERIC self._skip_update = False @@ -918,7 +925,7 @@ class XiaomiPlugGenericSwitch(XiaomiMiioEntity, SwitchEntity): await self._try_command( "Setting the power price of the power strip failed", - self._device.set_power_price, + self._device.set_power_price, # type: ignore[union-attr] price, ) @@ -926,9 +933,17 @@ class XiaomiPlugGenericSwitch(XiaomiMiioEntity, SwitchEntity): class XiaomiPowerStripSwitch(XiaomiPlugGenericSwitch): """Representation of a Xiaomi Power Strip.""" - def __init__(self, name, plug, model, unique_id): + _device: PowerStrip + + def __init__( + self, + name: str, + plug: PowerStrip, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the plug switch.""" - super().__init__(name, plug, model, unique_id) + super().__init__(name, plug, entry, unique_id) if self._model == MODEL_POWER_STRIP_V2: self._device_features = FEATURE_FLAGS_POWER_STRIP_V2 @@ -995,7 +1010,16 @@ class XiaomiPowerStripSwitch(XiaomiPlugGenericSwitch): class ChuangMiPlugSwitch(XiaomiPlugGenericSwitch): """Representation of a Chuang Mi Plug V1 and V3.""" - def __init__(self, name, plug, entry, unique_id, channel_usb): + _device: ChuangmiPlug + + def __init__( + self, + name: str, + plug: ChuangmiPlug, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + channel_usb: bool, + ) -> None: """Initialize the plug switch.""" name = f"{name} USB" if channel_usb else name @@ -1015,11 +1039,13 @@ class ChuangMiPlugSwitch(XiaomiPlugGenericSwitch): """Turn a channel on.""" if self._channel_usb: result = await self._try_command( - "Turning the plug on failed", self._device.usb_on + "Turning the plug on failed", + self._device.usb_on, ) else: result = await self._try_command( - "Turning the plug on failed", self._device.on + "Turning the plug on failed", + self._device.on, ) if result: @@ -1030,7 +1056,8 @@ class ChuangMiPlugSwitch(XiaomiPlugGenericSwitch): """Turn a channel off.""" if self._channel_usb: result = await self._try_command( - "Turning the plug off failed", self._device.usb_off + "Turning the plug off failed", + self._device.usb_off, ) else: result = await self._try_command( @@ -1075,16 +1102,25 @@ class ChuangMiPlugSwitch(XiaomiPlugGenericSwitch): class XiaomiAirConditioningCompanionSwitch(XiaomiPlugGenericSwitch): """Representation of a Xiaomi AirConditioning Companion.""" - def __init__(self, name, plug, model, unique_id): + _device: AirConditioningCompanionV3 + + def __init__( + self, + name: str, + plug: AirConditioningCompanionV3, + entry: XiaomiMiioConfigEntry, + unique_id: str | None, + ) -> None: """Initialize the acpartner switch.""" - super().__init__(name, plug, model, unique_id) + super().__init__(name, plug, entry, unique_id) self._state_attrs.update({ATTR_TEMPERATURE: None, ATTR_LOAD_POWER: None}) async def async_turn_on(self, **kwargs: Any) -> None: """Turn the socket on.""" result = await self._try_command( - "Turning the socket on failed", self._device.socket_on + "Turning the socket on failed", + self._device.socket_on, ) if result: @@ -1094,7 +1130,8 @@ class XiaomiAirConditioningCompanionSwitch(XiaomiPlugGenericSwitch): async def async_turn_off(self, **kwargs: Any) -> None: """Turn the socket off.""" result = await self._try_command( - "Turning the socket off failed", self._device.socket_off + "Turning the socket off failed", + self._device.socket_off, ) if result: