"""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)