mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-23 11:43:34 +00:00
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
"""Implementation of the motion detection (PIR) module found in some dimmers."""
|
|
from enum import Enum
|
|
from typing import Optional
|
|
|
|
from kasa.smartdevice import SmartDeviceException
|
|
|
|
from .module import Module
|
|
|
|
|
|
class Range(Enum):
|
|
"""Range for motion detection."""
|
|
|
|
Far = 0
|
|
Mid = 1
|
|
Near = 2
|
|
Custom = 3
|
|
|
|
|
|
# TODO: use the config reply in tests
|
|
# {"enable":0,"version":"1.0","trigger_index":2,"cold_time":60000,
|
|
# "min_adc":0,"max_adc":4095,"array":[80,50,20,0],"err_code":0}}}
|
|
|
|
|
|
class Motion(Module):
|
|
"""Implements the motion detection (PIR) module."""
|
|
|
|
def query(self):
|
|
"""Request PIR configuration."""
|
|
return self.query_for_command("get_config")
|
|
|
|
@property
|
|
def range(self) -> Range:
|
|
"""Return motion detection range."""
|
|
return Range(self.data["trigger_index"])
|
|
|
|
@property
|
|
def enabled(self) -> bool:
|
|
"""Return True if module is enabled."""
|
|
return bool(self.data["enable"])
|
|
|
|
async def set_enabled(self, state: bool):
|
|
"""Enable/disable PIR."""
|
|
return await self.call("set_enable", {"enable": int(state)})
|
|
|
|
async def set_range(
|
|
self, *, range: Optional[Range] = None, custom_range: Optional[int] = None
|
|
):
|
|
"""Set the range for the sensor.
|
|
|
|
:param range: for using standard ranges
|
|
:param custom_range: range in decimeters, overrides the range parameter
|
|
"""
|
|
if custom_range is not None:
|
|
payload = {"index": Range.Custom.value, "value": custom_range}
|
|
elif range is not None:
|
|
payload = {"index": range.value}
|
|
else:
|
|
raise SmartDeviceException(
|
|
"Either range or custom_range need to be defined"
|
|
)
|
|
|
|
return await self.call("set_trigger_sens", payload)
|