From d05127cb711e8f0db243b8fb3f1da2e84b5364f3 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Sun, 24 Oct 2021 20:56:15 -1000 Subject: [PATCH] Add entity category to august (#58359) --- homeassistant/components/august/binary_sensor.py | 2 ++ homeassistant/components/august/sensor.py | 9 ++++++++- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/august/binary_sensor.py b/homeassistant/components/august/binary_sensor.py index 9a38cd1e301..cf34952309b 100644 --- a/homeassistant/components/august/binary_sensor.py +++ b/homeassistant/components/august/binary_sensor.py @@ -26,6 +26,7 @@ from homeassistant.components.binary_sensor import ( BinarySensorEntity, BinarySensorEntityDescription, ) +from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC from homeassistant.core import callback from homeassistant.helpers.event import async_call_later @@ -129,6 +130,7 @@ SENSOR_TYPES_DOORBELL: tuple[AugustBinarySensorEntityDescription, ...] = ( key="doorbell_online", name="Online", device_class=DEVICE_CLASS_CONNECTIVITY, + entity_category=ENTITY_CATEGORY_DIAGNOSTIC, value_fn=_retrieve_online_state, is_time_based=False, ), diff --git a/homeassistant/components/august/sensor.py b/homeassistant/components/august/sensor.py index b6fa767edb7..744177cbef3 100644 --- a/homeassistant/components/august/sensor.py +++ b/homeassistant/components/august/sensor.py @@ -16,7 +16,12 @@ from homeassistant.components.sensor import ( SensorEntity, SensorEntityDescription, ) -from homeassistant.const import ATTR_ENTITY_PICTURE, PERCENTAGE, STATE_UNAVAILABLE +from homeassistant.const import ( + ATTR_ENTITY_PICTURE, + ENTITY_CATEGORY_DIAGNOSTIC, + PERCENTAGE, + STATE_UNAVAILABLE, +) from homeassistant.core import callback from homeassistant.helpers.entity_registry import async_get_registry from homeassistant.helpers.restore_state import RestoreEntity @@ -68,12 +73,14 @@ class AugustSensorEntityDescription( SENSOR_TYPE_DEVICE_BATTERY = AugustSensorEntityDescription[LockDetail]( key="device_battery", name="Battery", + entity_category=ENTITY_CATEGORY_DIAGNOSTIC, value_fn=_retrieve_device_battery_state, ) SENSOR_TYPE_KEYPAD_BATTERY = AugustSensorEntityDescription[KeypadDetail]( key="linked_keypad_battery", name="Battery", + entity_category=ENTITY_CATEGORY_DIAGNOSTIC, value_fn=_retrieve_linked_keypad_battery_state, )