Add category diagnostic to Surepetcare battery sensor (#57852)

This commit is contained in:
Daniel Hjelseth Høyer 2021-10-17 19:26:29 +02:00 committed by GitHub
parent dd2d708cb9
commit 9c3aa8156d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -12,6 +12,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ( from homeassistant.const import (
ATTR_VOLTAGE, ATTR_VOLTAGE,
DEVICE_CLASS_BATTERY, DEVICE_CLASS_BATTERY,
ENTITY_CATEGORY_DIAGNOSTIC,
PERCENTAGE, PERCENTAGE,
VOLUME_MILLILITERS, VOLUME_MILLILITERS,
) )
@ -49,9 +50,10 @@ async def async_setup_entry(
class SureBattery(SurePetcareEntity, SensorEntity): class SureBattery(SurePetcareEntity, SensorEntity):
"""A sensor implementation for Sure Petcare Entities.""" """A sensor implementation for Sure Petcare batteries."""
_attr_device_class = DEVICE_CLASS_BATTERY _attr_device_class = DEVICE_CLASS_BATTERY
_attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
_attr_native_unit_of_measurement = PERCENTAGE _attr_native_unit_of_measurement = PERCENTAGE
def __init__( def __init__(