mirror of
https://github.com/home-assistant/core.git
synced 2025-04-28 11:17:53 +00:00
d754ea7e22
* Add new Rabbit Air integration * Remove py.typed file It is not needed and was just accidentally added to the commit. * Enable strict type checking for rabbitair component Keeping the code fully type hinted is a good idea. * Add missing type annotations * Remove translation file * Prevent data to be added to hass.data if refresh fails * Reload the config entry when the options change * Add missing type parameters for generics * Avoid using assert in production code * Move zeroconf to optional dependencies * Remove unnecessary logging Co-authored-by: Franck Nijhof <frenck@frenck.nl> * Remove unused keys from the manifest Co-authored-by: Franck Nijhof <frenck@frenck.nl> * Replace property with attr Co-authored-by: Franck Nijhof <frenck@frenck.nl> * Allow to return None for power The type of the is_on property now allows this. Co-authored-by: Franck Nijhof <frenck@frenck.nl> * Remove unnecessary method call Co-authored-by: Franck Nijhof <frenck@frenck.nl> * Update the python library The new version properly re-exports names from the package root. * Remove options flow Scan interval should not be part of integration configuration. This was the only option, so the options flow can be fully removed. * Replace properties with attrs * Remove multiline ternary operator * Use NamedTuple for hass.data * Remove unused logger variable * Move async_setup_entry up in the file * Adjust debouncer settings to use request_refresh * Prevent status updates during the cooldown period * Move device polling code to the update coordinator * Fix the problem with the switch jumping back and forth The UI seems to have a timeout of 2 seconds somewhere, which is just a little bit less than what we normally need to get an updated state. So the power switch would jump to its previous state and then immediately return to the new state. * Update the python library The new version fixes errors when multiple requests are executed simultaneously. * Fix incorrect check for pending call in debouncer This caused the polling to stop. * Fix tests * Update .coveragerc to exclude new file. * Remove test for Options Flow. * Update the existing entry when device access details change * Add Zeroconf discovery step * Fix tests The ZeroconfServiceInfo constructor now requires one more argument. * Fix typing for CoordinatorEntity * Fix signature of async_turn_on * Fix depreciation warnings * Fix manifest formatting * Fix warning about debouncer typing relates to 5ae5ae5392729b4c94a8004bd02e147d60227341 * Wait for config entry platform forwards * Apply some of the suggested changes * Do not put the MAC address in the title. Use a fixed title instead. * Do not format the MAC to use as a unique ID. * Do not catch exceptions in _async_update_data(). * Remove unused _entry field in the base entity class. * Use the standard attribute self._attr_is_on to keep the power state. * Store the MAC in the config entry data * Change the order of except clauses OSError is an ancestor class of TimeoutError, so TimeoutError should be handled first * Fix depreciation warnings * Fix tests The ZeroconfServiceInfo constructor arguments have changed. * Fix DeviceInfo import * Rename the method to make it clearer what it does * Apply suggestions from code review * Fix tests * Change speed/mode logic to use is_on from the base class * A zero value is more appropriate than None since None means "unknown", but we actually know that the speed is zero when the power is off. --------- Co-authored-by: Franck Nijhof <frenck@frenck.nl> Co-authored-by: Erik Montnemery <erik@montnemery.com>
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
"""A base class for Rabbit Air entities."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
from rabbitair import Model
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_MAC
|
|
from homeassistant.helpers.device_registry import DeviceInfo
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|
|
|
from .const import DOMAIN
|
|
from .coordinator import RabbitAirDataUpdateCoordinator
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
MODELS = {
|
|
Model.A3: "A3",
|
|
Model.BioGS: "BioGS 2.0",
|
|
Model.MinusA2: "MinusA2",
|
|
None: None,
|
|
}
|
|
|
|
|
|
class RabbitAirBaseEntity(CoordinatorEntity[RabbitAirDataUpdateCoordinator]):
|
|
"""Base class for Rabbit Air entity."""
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: RabbitAirDataUpdateCoordinator,
|
|
entry: ConfigEntry,
|
|
) -> None:
|
|
"""Initialize the entity."""
|
|
super().__init__(coordinator)
|
|
self._attr_name = entry.title
|
|
self._attr_unique_id = entry.unique_id
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, entry.data[CONF_MAC])},
|
|
manufacturer="Rabbit Air",
|
|
model=MODELS.get(coordinator.data.model),
|
|
name=entry.title,
|
|
sw_version=coordinator.data.wifi_firmware,
|
|
hw_version=coordinator.data.main_firmware,
|
|
)
|
|
|
|
def _is_model(self, model: Model | list[Model]) -> bool:
|
|
"""Check the model of the device."""
|
|
if isinstance(model, list):
|
|
return self.coordinator.data.model in model
|
|
return self.coordinator.data.model is model
|
|
|
|
async def _set_state(self, **kwargs: Any) -> None:
|
|
"""Change the state of the device."""
|
|
_LOGGER.debug("Set state %s", kwargs)
|
|
await self.coordinator.device.set_state(**kwargs)
|
|
# Force polling of the device, because changing one parameter often
|
|
# causes other parameters to change as well. By getting updated status
|
|
# we provide a better user experience, especially if the default
|
|
# polling interval is set too long.
|
|
await self.coordinator.async_request_refresh()
|