From 9f26850a19bd7076a41a759df328f9ff0d1316cf Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Tue, 30 Nov 2021 01:44:21 +0100 Subject: [PATCH] Add device class support for button entity (#60560) --- homeassistant/components/button/__init__.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/homeassistant/components/button/__init__.py b/homeassistant/components/button/__init__.py index fe2d1cb6435..621effd5d16 100644 --- a/homeassistant/components/button/__init__.py +++ b/homeassistant/components/button/__init__.py @@ -6,6 +6,8 @@ from datetime import datetime, timedelta import logging from typing import final +import voluptuous as vol + from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.config_validation import ( # noqa: F401 @@ -28,6 +30,13 @@ MIN_TIME_BETWEEN_SCANS = timedelta(seconds=10) _LOGGER = logging.getLogger(__name__) +DEVICE_CLASS_RESTART = "restart" +DEVICE_CLASS_UPDATE = "update" + +DEVICE_CLASSES = [DEVICE_CLASS_RESTART, DEVICE_CLASS_UPDATE] + +DEVICE_CLASSES_SCHEMA = vol.All(vol.Lower, vol.In(DEVICE_CLASSES)) + async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: """Set up Button entities."""