mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-23 19:53:34 +00:00
65874c0365
Moves `FeatureType` into `Feature` to make it easier to use the API. This also enforces that no invalid types are accepted (i.e., `Category.Config` cannot be a `Sensor`) If `--verbose` is used with the cli tool, some extra information is displayed for features when in the state command.
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
"""Implementation of battery module."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from ...feature import Feature
|
|
from ..smartmodule import SmartModule
|
|
|
|
if TYPE_CHECKING:
|
|
from ..smartdevice import SmartDevice
|
|
|
|
|
|
class BatterySensor(SmartModule):
|
|
"""Implementation of battery module."""
|
|
|
|
REQUIRED_COMPONENT = "battery_detect"
|
|
QUERY_GETTER_NAME = "get_battery_detect_info"
|
|
|
|
def __init__(self, device: SmartDevice, module: str):
|
|
super().__init__(device, module)
|
|
self._add_feature(
|
|
Feature(
|
|
device,
|
|
"Battery level",
|
|
container=self,
|
|
attribute_getter="battery",
|
|
icon="mdi:battery",
|
|
)
|
|
)
|
|
self._add_feature(
|
|
Feature(
|
|
device,
|
|
"Battery low",
|
|
container=self,
|
|
attribute_getter="battery_low",
|
|
icon="mdi:alert",
|
|
type=Feature.Type.BinarySensor,
|
|
)
|
|
)
|
|
|
|
@property
|
|
def battery(self):
|
|
"""Return battery level."""
|
|
return self._device.sys_info["battery_percentage"]
|
|
|
|
@property
|
|
def battery_low(self):
|
|
"""Return True if battery is low."""
|
|
return self._device.sys_info["at_low_battery"]
|