evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add button device classes to Shelly (#60625)

Browse source
This commit is contained in:
Franck Nijhof 2021-11-30 15:44:47 +01:00 • committed by GitHub
parent 2366fbe846
commit 6be1b0c704
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
homeassistant/components/shelly/button.py
View file

@ -5,7 +5,11 @@ from collections.abc import Callable
from dataclasses import dataclass from dataclasses import dataclass
from typing import Final, cast from typing import Final, cast
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription from homeassistant.components.button import (
ButtonDeviceClass,
ButtonEntity,
ButtonEntityDescription,
)
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ENTITY_CATEGORY_CONFIG from homeassistant.const import ENTITY_CATEGORY_CONFIG
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
@ -35,14 +39,14 @@ BUTTONS: Final = [
ShellyButtonDescription( ShellyButtonDescription(
key="ota_update", key="ota_update",
name="OTA Update", name="OTA Update",
icon="mdi:package-up", device_class=ButtonDeviceClass.UPDATE,
entity_category=ENTITY_CATEGORY_CONFIG, entity_category=ENTITY_CATEGORY_CONFIG,
press_action=lambda wrapper: wrapper.async_trigger_ota_update(), press_action=lambda wrapper: wrapper.async_trigger_ota_update(),
), ),
ShellyButtonDescription( ShellyButtonDescription(
key="ota_update_beta", key="ota_update_beta",
name="OTA Update Beta", name="OTA Update Beta",
icon="mdi:flask-outline", device_class=ButtonDeviceClass.UPDATE,
entity_registry_enabled_default=False, entity_registry_enabled_default=False,
entity_category=ENTITY_CATEGORY_CONFIG, entity_category=ENTITY_CATEGORY_CONFIG,
press_action=lambda wrapper: wrapper.async_trigger_ota_update(beta=True), press_action=lambda wrapper: wrapper.async_trigger_ota_update(beta=True),
@ -50,7 +54,7 @@ BUTTONS: Final = [
ShellyButtonDescription( ShellyButtonDescription(
key="reboot", key="reboot",
name="Reboot", name="Reboot",
icon="mdi:restart", device_class=ButtonDeviceClass.RESTART,
entity_category=ENTITY_CATEGORY_CONFIG, entity_category=ENTITY_CATEGORY_CONFIG,
press_action=lambda wrapper: wrapper.device.trigger_reboot(), press_action=lambda wrapper: wrapper.device.trigger_reboot(),
), ),