jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-16 17:57:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use shorthand attrs for tasmota (#100390)

Browse Source
This commit is contained in:
Jan Bouwhuis 2023-09-14 18:17:23 +02:00 committed by GitHub
parent 8b7061b634
commit 6701a449bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 23 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
homeassistant/components/tasmota/mixins.py
View File

@ -38,6 +38,9 @@ class TasmotaEntity(Entity):
"""Initialize."""
self._tasmota_entity = tasmota_entity
self._unique_id = tasmota_entity.unique_id
self._attr_device_info = DeviceInfo(
connections={(CONNECTION_NETWORK_MAC, tasmota_entity.mac)}
)
async def async_added_to_hass(self) -> None:
"""Subscribe to MQTT events."""
@ -61,13 +64,6 @@ class TasmotaEntity(Entity):
"""(Re)Subscribe to topics."""
await self._tasmota_entity.subscribe_topics()
@property
def device_info(self) -> DeviceInfo:
"""Return a device description for device registry."""
return DeviceInfo(
connections={(CONNECTION_NETWORK_MAC, self._tasmota_entity.mac)}
)
@property
def name(self) -> str | None:
"""Return the name of the binary sensor."""

69
homeassistant/components/tasmota/sensor.py
View File

@ -274,6 +274,26 @@ class TasmotaSensor(TasmotaAvailability, TasmotaDiscoveryUpdate, SensorEntity):
**kwds,
)
class_or_icon = SENSOR_DEVICE_CLASS_ICON_MAP.get(
self._tasmota_entity.quantity, {}
)
self._attr_device_class = class_or_icon.get(DEVICE_CLASS)
self._attr_state_class = class_or_icon.get(STATE_CLASS)
if self._tasmota_entity.quantity in status_sensor.SENSORS:
self._attr_entity_category = EntityCategory.DIAGNOSTIC
# Hide fast changing status sensors
if self._tasmota_entity.quantity in (
hc.SENSOR_STATUS_IP,
hc.SENSOR_STATUS_RSSI,
hc.SENSOR_STATUS_SIGNAL,
hc.SENSOR_STATUS_VERSION,
):
self._attr_entity_registry_enabled_default = False
self._attr_icon = class_or_icon.get(ICON)
self._attr_native_unit_of_measurement = SENSOR_UNIT_MAP.get(
self._tasmota_entity.unit, self._tasmota_entity.unit
)
async def async_added_to_hass(self) -> None:
"""Subscribe to MQTT events."""
self._tasmota_entity.set_on_state_callback(self.sensor_state_updated)
@ -288,58 +308,9 @@ class TasmotaSensor(TasmotaAvailability, TasmotaDiscoveryUpdate, SensorEntity):
self._state = state
self.async_write_ha_state()
@property
def device_class(self) -> SensorDeviceClass | None:
"""Return the device class of the sensor."""
class_or_icon = SENSOR_DEVICE_CLASS_ICON_MAP.get(
self._tasmota_entity.quantity, {}
)
return class_or_icon.get(DEVICE_CLASS)
@property
def state_class(self) -> str | None:
"""Return the state class of the sensor."""
class_or_icon = SENSOR_DEVICE_CLASS_ICON_MAP.get(
self._tasmota_entity.quantity, {}
)
return class_or_icon.get(STATE_CLASS)
@property
def entity_category(self) -> EntityCategory | None:
"""Return the category of the entity, if any."""
if self._tasmota_entity.quantity in status_sensor.SENSORS:
return EntityCategory.DIAGNOSTIC
return None
@property
def entity_registry_enabled_default(self) -> bool:
"""Return if the entity should be enabled when first added to the entity registry."""
# Hide fast changing status sensors
if self._tasmota_entity.quantity in (
hc.SENSOR_STATUS_IP,
hc.SENSOR_STATUS_RSSI,
hc.SENSOR_STATUS_SIGNAL,
hc.SENSOR_STATUS_VERSION,
):
return False
return True
@property
def icon(self) -> str | None:
"""Return the icon."""
class_or_icon = SENSOR_DEVICE_CLASS_ICON_MAP.get(
self._tasmota_entity.quantity, {}
)
return class_or_icon.get(ICON)
@property
def native_value(self) -> datetime | str | None:
"""Return the state of the entity."""
if self._state_timestamp and self.device_class == SensorDeviceClass.TIMESTAMP:
return self._state_timestamp
return self._state
@property
def native_unit_of_measurement(self) -> str | None:
"""Return the unit this state is expressed in."""
return SENSOR_UNIT_MAP.get(self._tasmota_entity.unit, self._tasmota_entity.unit)