Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use device class enum in Surepetcare (#60701)

pull/60709/head
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 homeassistant.components.binary_sensor import (
DEVICE_CLASS_CONNECTIVITY,
DEVICE_CLASS_PRESENCE,
BinarySensorDeviceClass,
BinarySensorEntity,
)
from homeassistant.config_entries import ConfigEntry
@ -67,7 +66,7 @@ class SurePetcareBinarySensor(SurePetcareEntity, BinarySensorEntity):
class Hub(SurePetcareBinarySensor):
"""Sure Petcare Hub."""
_attr_device_class = DEVICE_CLASS_CONNECTIVITY
_attr_device_class = BinarySensorDeviceClass.CONNECTIVITY
_attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
@property
@ -94,7 +93,7 @@ class Hub(SurePetcareBinarySensor):
class Pet(SurePetcareBinarySensor):
"""Sure Petcare Pet."""
_attr_device_class = DEVICE_CLASS_PRESENCE
_attr_device_class = BinarySensorDeviceClass.PRESENCE
@callback
def _update_attr(self, surepy_entity: SurepyEntity) -> None:
@ -117,7 +116,7 @@ class Pet(SurePetcareBinarySensor):
class DeviceConnectivity(SurePetcareBinarySensor):
"""Sure Petcare Device."""
_attr_device_class = DEVICE_CLASS_CONNECTIVITY
_attr_device_class = BinarySensorDeviceClass.CONNECTIVITY
_attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
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.enums import EntityType
from homeassistant.components.sensor import SensorEntity
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
ATTR_VOLTAGE,
DEVICE_CLASS_BATTERY,
ENTITY_CATEGORY_DIAGNOSTIC,
PERCENTAGE,
VOLUME_MILLILITERS,
@ -52,7 +51,7 @@ async def async_setup_entry(
class SureBattery(SurePetcareEntity, SensorEntity):
"""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_native_unit_of_measurement = PERCENTAGE