home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-05 05:35:11 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add missing baseclass for rituals perfume genie entities (#55166)

Browse Source
This commit is contained in:
Milan Meulemans
2021-08-24 21:08:48 +02:00
committed by GitHub
parent 18f80c32d7
commit 9555a34691
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
homeassistant/components/rituals_perfume_genie/sensor.py
View File

@@ -3,6 +3,7 @@ from __future__ import annotations
from pyrituals import Diffuser from pyrituals import Diffuser
from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ( from homeassistant.const import (
DEVICE_CLASS_BATTERY, DEVICE_CLASS_BATTERY,
@@ -42,7 +43,7 @@ async def async_setup_entry(
async_add_entities(entities) async_add_entities(entities)
class DiffuserPerfumeSensor(DiffuserEntity): class DiffuserPerfumeSensor(DiffuserEntity, SensorEntity):
"""Representation of a diffuser perfume sensor.""" """Representation of a diffuser perfume sensor."""
def __init__( def __init__(
@@ -64,7 +65,7 @@ class DiffuserPerfumeSensor(DiffuserEntity):
return self._diffuser.perfume return self._diffuser.perfume
class DiffuserFillSensor(DiffuserEntity): class DiffuserFillSensor(DiffuserEntity, SensorEntity):
"""Representation of a diffuser fill sensor.""" """Representation of a diffuser fill sensor."""
def __init__( def __init__(
@@ -86,7 +87,7 @@ class DiffuserFillSensor(DiffuserEntity):
return self._diffuser.fill return self._diffuser.fill
class DiffuserBatterySensor(DiffuserEntity): class DiffuserBatterySensor(DiffuserEntity, SensorEntity):
"""Representation of a diffuser battery sensor.""" """Representation of a diffuser battery sensor."""
_attr_device_class = DEVICE_CLASS_BATTERY _attr_device_class = DEVICE_CLASS_BATTERY
@@ -104,7 +105,7 @@ class DiffuserBatterySensor(DiffuserEntity):
return self._diffuser.battery_percentage return self._diffuser.battery_percentage
class DiffuserWifiSensor(DiffuserEntity): class DiffuserWifiSensor(DiffuserEntity, SensorEntity):
"""Representation of a diffuser wifi sensor.""" """Representation of a diffuser wifi sensor."""
_attr_device_class = DEVICE_CLASS_SIGNAL_STRENGTH _attr_device_class = DEVICE_CLASS_SIGNAL_STRENGTH

2
homeassistant/components/rituals_perfume_genie/switch.py
View File

@@ -31,7 +31,7 @@ async def async_setup_entry(
async_add_entities(entities) async_add_entities(entities)
class DiffuserSwitch(SwitchEntity, DiffuserEntity): class DiffuserSwitch(DiffuserEntity, SwitchEntity):
"""Representation of a diffuser switch.""" """Representation of a diffuser switch."""
_attr_icon = "mdi:fan" _attr_icon = "mdi:fan"