Add base class to Smarty (#129112)

This commit is contained in:
Joost Lekkerkerker 2024-10-25 12:46:46 +02:00 committed by GitHub
parent 97eb768748
commit 61e2283146
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 15 additions and 6 deletions

View File

@ -15,9 +15,9 @@ from homeassistant.components.binary_sensor import (
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import SmartyConfigEntry, SmartyCoordinator from .coordinator import SmartyConfigEntry, SmartyCoordinator
from .entity import SmartyEntity
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -64,7 +64,7 @@ async def async_setup_entry(
) )
class SmartyBinarySensor(CoordinatorEntity[SmartyCoordinator], BinarySensorEntity): class SmartyBinarySensor(SmartyEntity, BinarySensorEntity):
"""Representation of a Smarty Binary Sensor.""" """Representation of a Smarty Binary Sensor."""
entity_description: SmartyBinarySensorEntityDescription entity_description: SmartyBinarySensorEntityDescription

View File

@ -0,0 +1,9 @@
"""Smarty Entity class."""
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .coordinator import SmartyCoordinator
class SmartyEntity(CoordinatorEntity[SmartyCoordinator]):
"""Representation of a Smarty Entity."""

View File

@ -10,7 +10,6 @@ from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from homeassistant.util.percentage import ( from homeassistant.util.percentage import (
percentage_to_ranged_value, percentage_to_ranged_value,
ranged_value_to_percentage, ranged_value_to_percentage,
@ -19,6 +18,7 @@ from homeassistant.util.scaling import int_states_in_range
from . import SmartyConfigEntry from . import SmartyConfigEntry
from .coordinator import SmartyCoordinator from .coordinator import SmartyCoordinator
from .entity import SmartyEntity
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -38,7 +38,7 @@ async def async_setup_entry(
async_add_entities([SmartyFan(coordinator)]) async_add_entities([SmartyFan(coordinator)])
class SmartyFan(CoordinatorEntity[SmartyCoordinator], FanEntity): class SmartyFan(SmartyEntity, FanEntity):
"""Representation of a Smarty Fan.""" """Representation of a Smarty Fan."""
_attr_icon = "mdi:air-conditioner" _attr_icon = "mdi:air-conditioner"

View File

@ -17,10 +17,10 @@ from homeassistant.components.sensor import (
from homeassistant.const import UnitOfTemperature from homeassistant.const import UnitOfTemperature
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
from .coordinator import SmartyConfigEntry, SmartyCoordinator from .coordinator import SmartyConfigEntry, SmartyCoordinator
from .entity import SmartyEntity
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -94,7 +94,7 @@ async def async_setup_entry(
) )
class SmartySensor(CoordinatorEntity[SmartyCoordinator], SensorEntity): class SmartySensor(SmartyEntity, SensorEntity):
"""Representation of a Smarty Sensor.""" """Representation of a Smarty Sensor."""
entity_description: SmartySensorDescription entity_description: SmartySensorDescription