mirror of
https://github.com/python-kasa/python-kasa.git
synced 2025-01-08 22:07:06 +00:00
905a14895d
Ensures that all modules try to access their data in `_post_update_hook` in a safe manner and disable themselves if there's an error. Also adds parameters to get_preset_rules and get_on_off_gradually_info to fix issues with recent firmware updates. [#1033](https://github.com/python-kasa/python-kasa/issues/1033)
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
"""Implementation of cloud 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 Cloud(SmartModule):
|
|
"""Implementation of cloud module."""
|
|
|
|
QUERY_GETTER_NAME = "get_connect_cloud_state"
|
|
REQUIRED_COMPONENT = "cloud_connect"
|
|
|
|
def _post_update_hook(self):
|
|
"""Perform actions after a device update.
|
|
|
|
Overrides the default behaviour to disable a module if the query returns
|
|
an error because the logic here is to treat that as not connected.
|
|
"""
|
|
|
|
def __init__(self, device: SmartDevice, module: str):
|
|
super().__init__(device, module)
|
|
|
|
self._add_feature(
|
|
Feature(
|
|
device,
|
|
id="cloud_connection",
|
|
name="Cloud connection",
|
|
container=self,
|
|
attribute_getter="is_connected",
|
|
icon="mdi:cloud",
|
|
type=Feature.Type.BinarySensor,
|
|
category=Feature.Category.Info,
|
|
)
|
|
)
|
|
|
|
@property
|
|
def is_connected(self):
|
|
"""Return True if device is connected to the cloud."""
|
|
if self._has_data_error():
|
|
return False
|
|
return self.data["status"] == 0
|