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

Migrate esphome button platform to use _on_static_info_update (#95007)

Browse Source
This commit is contained in:
J. Nick Koston 2023-06-21 22:24:26 +02:00 committed by GitHub
parent 8d2daaa694
commit 235f50a341
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/components/esphome/button.py
View File

@ -1,7 +1,7 @@
"""Support for ESPHome buttons."""
from __future__ import annotations
from aioesphomeapi import ButtonInfo, EntityState
from aioesphomeapi import ButtonInfo, EntityInfo, EntityState
from homeassistant.components.button import ButtonDeviceClass, ButtonEntity
from homeassistant.config_entries import ConfigEntry
@ -30,10 +30,13 @@ async def async_setup_entry(
class EsphomeButton(EsphomeEntity[ButtonInfo, EntityState], ButtonEntity):
"""A button implementation for ESPHome."""
@property
def device_class(self) -> ButtonDeviceClass | None:
"""Return the class of this entity."""
return try_parse_enum(ButtonDeviceClass, self._static_info.device_class)
@callback
def _on_static_info_update(self, static_info: EntityInfo) -> None:
"""Set attrs from static info."""
super()._on_static_info_update(static_info)
self._attr_device_class = try_parse_enum(
ButtonDeviceClass, self._static_info.device_class
)
@callback
def _on_device_update(self) -> None:
@ -44,4 +47,4 @@ class EsphomeButton(EsphomeEntity[ButtonInfo, EntityState], ButtonEntity):
async def async_press(self) -> None:
"""Press the button."""
await self._client.button_command(self._static_info.key)
await self._client.button_command(self._key)