Add vacuumconsumables

This commit is contained in:
Teemu Rytilahti 2024-11-30 17:11:46 +01:00
parent 74b59d7f98
commit dc04e83b4c
2 changed files with 134 additions and 0 deletions

View File

@ -30,6 +30,7 @@ from .temperaturesensor import TemperatureSensor
from .thermostat import Thermostat
from .time import Time
from .triggerlogs import TriggerLogs
from .vacuumconsumables import VacuumConsumables
from .waterleaksensor import WaterleakSensor
__all__ = [
@ -63,5 +64,6 @@ __all__ = [
"TriggerLogs",
"FrostProtection",
"Thermostat",
"VacuumConsumables",
"SmartLightEffect",
]

View File

@ -0,0 +1,132 @@
"""Implementation of vacuum consumables."""
from __future__ import annotations
import logging
from ...feature import Feature
from ..smartmodule import SmartModule
_LOGGER = logging.getLogger(__name__)
class VacuumConsumables(SmartModule):
"""Implementation of vacuum consumables."""
REQUIRED_COMPONENT = "consumables"
QUERY_GETTER_NAME = "getConsumablesInfo"
def _initialize_features(self) -> None:
"""Initialize features."""
self._add_feature(
Feature(
self._device,
id="consumable_charge_contact",
name="Charge contact time",
container=self,
attribute_getter="charge_contact",
category=Feature.Category.Info,
type=Feature.Type.Sensor,
)
)
self._add_feature(
Feature(
self._device,
id="consumable_main_brush_lid",
name="Main brush lid time",
container=self,
attribute_getter="main_brush_lid",
category=Feature.Category.Info,
type=Feature.Type.Sensor,
)
)
self._add_feature(
Feature(
self._device,
id="consumable_edge_brush",
name="Edge brush time",
container=self,
attribute_getter="edge_brush",
category=Feature.Category.Info,
type=Feature.Type.Sensor,
)
)
self._add_feature(
Feature(
self._device,
id="consumable_roll_brush",
name="Roll brush time",
container=self,
attribute_getter="roll_brush",
category=Feature.Category.Info,
type=Feature.Type.Sensor,
)
)
self._add_feature(
Feature(
self._device,
id="consumable_filter",
name="Filter time",
container=self,
attribute_getter="filter",
category=Feature.Category.Info,
type=Feature.Type.Sensor,
)
)
self._add_feature(
Feature(
self._device,
id="consumable_rag",
name="Rag time",
container=self,
attribute_getter="rag",
category=Feature.Category.Info,
type=Feature.Type.Sensor,
)
)
self._add_feature(
Feature(
self._device,
id="consumable_sensor",
name="Sensor time",
container=self,
attribute_getter="sensor",
category=Feature.Category.Info,
type=Feature.Type.Sensor,
)
)
@property
def charge_contact(self) -> int:
"""Time disconnected from charger?"""
return self.data["charge_contact_time"]
@property
def main_brush_lid(self) -> int:
"""Main brush time? Or something else?"""
return self.data["main_brush_lid_time"]
@property
def edge_brush(self) -> int:
"""Edge brush time."""
return self.data["edge_brush_time"]
@property
def roll_brush(self) -> int:
"""Roll brush time."""
return self.data["roll_brush_time"]
@property
def filter(self) -> int:
"""Filter time."""
return self.data["filter_time"]
@property
def rag(self) -> int:
"""Rag time."""
return self.data["rag_time"]
@property
def sensor(self) -> int:
"""Sensor time.."""
return self.data["sensor_time"]