python-kasa/kasa/smart/modules/battery.py

"""Implementation of battery module."""

from typing import TYPE_CHECKING

from ...feature import Feature, FeatureType
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=FeatureType.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"]