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

Use device class enum in Surepetcare (#60701)

Browse Source
This commit is contained in:
Daniel Hjelseth Høyer 2021-12-01 10:58:52 +01:00 committed by GitHub
parent ca55216d67
commit 739ce9bc75
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
homeassistant/components/surepetcare/binary_sensor.py
View File

@ -8,8 +8,7 @@ from surepy.entities.pet import Pet as SurepyPet
from surepy.enums import EntityType, Location from surepy.enums import EntityType, Location
from homeassistant.components.binary_sensor import ( from homeassistant.components.binary_sensor import (
DEVICE_CLASS_CONNECTIVITY, BinarySensorDeviceClass,
DEVICE_CLASS_PRESENCE,
BinarySensorEntity, BinarySensorEntity,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
@ -67,7 +66,7 @@ class SurePetcareBinarySensor(SurePetcareEntity, BinarySensorEntity):
class Hub(SurePetcareBinarySensor): class Hub(SurePetcareBinarySensor):
"""Sure Petcare Hub.""" """Sure Petcare Hub."""
_attr_device_class = DEVICE_CLASS_CONNECTIVITY _attr_device_class = BinarySensorDeviceClass.CONNECTIVITY
_attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
@property @property
@ -94,7 +93,7 @@ class Hub(SurePetcareBinarySensor):
class Pet(SurePetcareBinarySensor): class Pet(SurePetcareBinarySensor):
"""Sure Petcare Pet.""" """Sure Petcare Pet."""
_attr_device_class = DEVICE_CLASS_PRESENCE _attr_device_class = BinarySensorDeviceClass.PRESENCE
@callback @callback
def _update_attr(self, surepy_entity: SurepyEntity) -> None: def _update_attr(self, surepy_entity: SurepyEntity) -> None:
@ -117,7 +116,7 @@ class Pet(SurePetcareBinarySensor):
class DeviceConnectivity(SurePetcareBinarySensor): class DeviceConnectivity(SurePetcareBinarySensor):
"""Sure Petcare Device.""" """Sure Petcare Device."""
_attr_device_class = DEVICE_CLASS_CONNECTIVITY _attr_device_class = BinarySensorDeviceClass.CONNECTIVITY
_attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
def __init__( def __init__(

5
homeassistant/components/surepetcare/sensor.py
View File

@ -7,11 +7,10 @@ from surepy.entities import SurepyEntity
from surepy.entities.devices import Felaqua as SurepyFelaqua from surepy.entities.devices import Felaqua as SurepyFelaqua
from surepy.enums import EntityType from surepy.enums import EntityType
from homeassistant.components.sensor import SensorEntity from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ( from homeassistant.const import (
ATTR_VOLTAGE, ATTR_VOLTAGE,
DEVICE_CLASS_BATTERY,
ENTITY_CATEGORY_DIAGNOSTIC, ENTITY_CATEGORY_DIAGNOSTIC,
PERCENTAGE, PERCENTAGE,
VOLUME_MILLILITERS, VOLUME_MILLILITERS,
@ -52,7 +51,7 @@ async def async_setup_entry(
class SureBattery(SurePetcareEntity, SensorEntity): class SureBattery(SurePetcareEntity, SensorEntity):
"""A sensor implementation for Sure Petcare batteries.""" """A sensor implementation for Sure Petcare batteries."""
_attr_device_class = DEVICE_CLASS_BATTERY _attr_device_class = SensorDeviceClass.BATTERY
_attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
_attr_native_unit_of_measurement = PERCENTAGE _attr_native_unit_of_measurement = PERCENTAGE