Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2024-12-22 11:13:34 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Convert to mashumaro, use kwargs for features

Browse Source
This commit is contained in:
Teemu Rytilahti 2024-11-30 16:54:47 +01:00
parent c509b601a0
commit 5928683056
1 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
kasa/smart/modules/vacuumrecords.py
View File

@ -1,11 +1,12 @@
"""Implementation of vacuum records for experimental vacuum support.""" """Implementation of vacuum cleaning records."""
from __future__ import annotations from __future__ import annotations
import logging import logging
from dataclasses import dataclass
from datetime import datetime from datetime import datetime
from pydantic.v1 import BaseModel from mashumaro import DataClassDictMixin
from ...feature import Feature from ...feature import Feature
from ..smartmodule import SmartModule from ..smartmodule import SmartModule
@ -13,7 +14,8 @@ from ..smartmodule import SmartModule
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
class LatestRecord(BaseModel): @dataclass
class LatestRecord(DataClassDictMixin):
"""Stats from last clean. """Stats from last clean.
TODO: this is just a list-formatted Record, with only some fields being available. TODO: this is just a list-formatted Record, with only some fields being available.
@ -25,7 +27,8 @@ class LatestRecord(BaseModel):
error: int # most likely error: int # most likely
class Record(BaseModel): @dataclass
class Record(DataClassDictMixin):
"""Historical cleanup result. """Historical cleanup result.
Example: Example:
@ -55,7 +58,8 @@ class Record(BaseModel):
record_index: int record_index: int
class Records(BaseModel): @dataclass
class Records(DataClassDictMixin):
"""Response payload for getCleanRecords. """Response payload for getCleanRecords.
Example: Example:
@ -85,21 +89,21 @@ class Records(BaseModel):
class VacuumRecords(SmartModule): class VacuumRecords(SmartModule):
"""Implementation of vacuum records for experimental vacuum support.""" """Implementation of vacuum cleaning records."""
REQUIRED_COMPONENT = "consumables" REQUIRED_COMPONENT = "consumables"
QUERY_GETTER_NAME = "getCleanRecords" QUERY_GETTER_NAME = "getCleanRecords"
def _initialize_features(self): def _initialize_features(self) -> None:
"""Initialize features.""" """Initialize features."""
self._add_feature( self._add_feature(
Feature( Feature(
self._device, self._device,
"total_clean_area", id="total_clean_area",
"Total area cleaned", name="Total area cleaned",
container=self, container=self,
attribute_getter="total_clean_area", attribute_getter="total_clean_area",
unit="sqm", unit_getter=lambda: "sqm",
category=Feature.Category.Info, category=Feature.Category.Info,
type=Feature.Type.Sensor, type=Feature.Type.Sensor,
) )
@ -107,8 +111,8 @@ class VacuumRecords(SmartModule):
self._add_feature( self._add_feature(
Feature( Feature(
self._device, self._device,
"total_clean_time", id="total_clean_time",
"Total time cleaned", name="Total time cleaned",
container=self, container=self,
attribute_getter="total_clean_time", attribute_getter="total_clean_time",
category=Feature.Category.Info, category=Feature.Category.Info,
@ -118,8 +122,8 @@ class VacuumRecords(SmartModule):
self._add_feature( self._add_feature(
Feature( Feature(
self._device, self._device,
"total_clean_count", id="total_clean_count",
"Total clean count", name="Total clean count",
container=self, container=self,
attribute_getter="total_clean_count", attribute_getter="total_clean_count",
category=Feature.Category.Debug, category=Feature.Category.Debug,