mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-22 11:13:34 +00:00
f7778aaa53
Also fixes a bug whereby multiple queries for the same module would overwrite each other.
88 lines
2.1 KiB
Python
88 lines
2.1 KiB
Python
"""Base implementation for all rule-based modules."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from dataclasses import dataclass
|
|
from enum import Enum
|
|
|
|
from mashumaro import DataClassDictMixin
|
|
|
|
from ..iotmodule import IotModule, merge
|
|
|
|
|
|
class Action(Enum):
|
|
"""Action to perform."""
|
|
|
|
Disabled = -1
|
|
TurnOff = 0
|
|
TurnOn = 1
|
|
Unknown = 2
|
|
|
|
|
|
class TimeOption(Enum):
|
|
"""Time when the action is executed."""
|
|
|
|
Disabled = -1
|
|
Enabled = 0
|
|
AtSunrise = 1
|
|
AtSunset = 2
|
|
|
|
|
|
@dataclass
|
|
class Rule(DataClassDictMixin):
|
|
"""Representation of a rule."""
|
|
|
|
id: str
|
|
name: str
|
|
enable: int
|
|
wday: list[int]
|
|
repeat: int
|
|
|
|
# start action
|
|
sact: Action | None = None
|
|
stime_opt: TimeOption | None = None
|
|
smin: int | None = None
|
|
|
|
eact: Action | None = None
|
|
etime_opt: TimeOption | None = None
|
|
emin: int | None = None
|
|
|
|
# Only on bulbs
|
|
s_light: dict | None = None
|
|
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class RuleModule(IotModule):
|
|
"""Base class for rule-based modules, such as countdown and antitheft."""
|
|
|
|
def query(self) -> dict:
|
|
"""Prepare the query for rules."""
|
|
q = self.query_for_command("get_rules")
|
|
return merge(q, self.query_for_command("get_next_action"))
|
|
|
|
@property
|
|
def rules(self) -> list[Rule]:
|
|
"""Return the list of rules for the service."""
|
|
try:
|
|
return [
|
|
Rule.from_dict(rule) for rule in self.data["get_rules"]["rule_list"]
|
|
]
|
|
except Exception as ex:
|
|
_LOGGER.error("Unable to read rule list: %s (data: %s)", ex, self.data)
|
|
return []
|
|
|
|
async def set_enabled(self, state: bool) -> dict:
|
|
"""Enable or disable the service."""
|
|
return await self.call("set_overall_enable", {"enable": state})
|
|
|
|
async def delete_rule(self, rule: Rule) -> dict:
|
|
"""Delete the given rule."""
|
|
return await self.call("delete_rule", {"id": rule.id})
|
|
|
|
async def delete_all_rules(self) -> dict:
|
|
"""Delete all rules."""
|
|
return await self.call("delete_all_rules")
|