Move and rename crownstone base entity to separate module (#126034)

This commit is contained in:
epenet 2024-09-16 14:26:20 +02:00 committed by GitHub
parent 8370a55263
commit e85ab067bd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 6 deletions

View file

@ -10,7 +10,7 @@ from homeassistant.helpers.entity import Entity
from .const import CROWNSTONE_INCLUDE_TYPES, DOMAIN from .const import CROWNSTONE_INCLUDE_TYPES, DOMAIN
class CrownstoneBaseEntity(Entity): class CrownstoneEntity(Entity):
"""Base entity class for Crownstone devices.""" """Base entity class for Crownstone devices."""
_attr_should_poll = False _attr_should_poll = False

View file

@ -24,7 +24,7 @@ from .const import (
SIG_CROWNSTONE_STATE_UPDATE, SIG_CROWNSTONE_STATE_UPDATE,
SIG_UART_STATE_CHANGE, SIG_UART_STATE_CHANGE,
) )
from .devices import CrownstoneBaseEntity from .entity import CrownstoneEntity
from .helpers import map_from_to from .helpers import map_from_to
if TYPE_CHECKING: if TYPE_CHECKING:
@ -39,7 +39,7 @@ async def async_setup_entry(
"""Set up crownstones from a config entry.""" """Set up crownstones from a config entry."""
manager: CrownstoneEntryManager = hass.data[DOMAIN][config_entry.entry_id] manager: CrownstoneEntryManager = hass.data[DOMAIN][config_entry.entry_id]
entities: list[CrownstoneEntity] = [] entities: list[CrownstoneLightEntity] = []
# Add Crownstone entities that support switching/dimming # Add Crownstone entities that support switching/dimming
for sphere in manager.cloud.cloud_data: for sphere in manager.cloud.cloud_data:
@ -47,10 +47,10 @@ async def async_setup_entry(
if crownstone.type in CROWNSTONE_INCLUDE_TYPES: if crownstone.type in CROWNSTONE_INCLUDE_TYPES:
# Crownstone can communicate with Crownstone USB # Crownstone can communicate with Crownstone USB
if manager.uart and sphere.cloud_id == manager.usb_sphere_id: if manager.uart and sphere.cloud_id == manager.usb_sphere_id:
entities.append(CrownstoneEntity(crownstone, manager.uart)) entities.append(CrownstoneLightEntity(crownstone, manager.uart))
# Crownstone can't communicate with Crownstone USB # Crownstone can't communicate with Crownstone USB
else: else:
entities.append(CrownstoneEntity(crownstone)) entities.append(CrownstoneLightEntity(crownstone))
async_add_entities(entities) async_add_entities(entities)
@ -65,7 +65,7 @@ def hass_to_crownstone_state(value: int) -> int:
return map_from_to(value, 0, 255, 0, 100) return map_from_to(value, 0, 255, 0, 100)
class CrownstoneEntity(CrownstoneBaseEntity, LightEntity): class CrownstoneLightEntity(CrownstoneEntity, LightEntity):
"""Representation of a crownstone. """Representation of a crownstone.
Light platform is used to support dimming. Light platform is used to support dimming.