Add generic interface (descriptors) for obtaining device features

kasa/iot/modules/cloud.py

@@ -4,6 +4,7 @@ try:
 except ImportError:
     from pydantic import BaseModel
 
+from ...descriptors import Descriptor, DescriptorType
 from .module import IotModule
 
 
@@ -25,6 +26,23 @@ class CloudInfo(BaseModel):
 class Cloud(IotModule):
     """Module implementing support for cloud services."""
 
+    def __init__(self, device, module):
+        super().__init__(device, module)
+        self.add_descriptor(
+            Descriptor(
+                device=self,
+                name="Cloud Connection",
+                icon="mdi:cloud",
+                attribute_getter="is_connected",
+                type=DescriptorType.BinarySensor,
+            )
+        )
+
+    @property
+    def is_connected(self) -> bool:
+        """Return true if device is connected to the cloud."""
+        return self.info.binded
+
     def query(self):
         """Request cloud connectivity info."""
         return self.query_for_command("get_info")

kasa/iot/modules/module.py

@@ -2,8 +2,9 @@
 import collections
 import logging
 from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Dict
 
+from ...descriptors import Descriptor
 from ...exceptions import SmartDeviceException
 
 if TYPE_CHECKING:
@@ -34,6 +35,14 @@ class IotModule(ABC):
     def __init__(self, device: "IotDevice", module: str):
         self._device = device
         self._module = module
+        self._module_descriptors: Dict[str, Descriptor] = {}
+
+    def add_descriptor(self, desc):
+        """Add module descriptor."""
+        module_desc_name = f"{self._module}_{desc.name}"
+        if module_desc_name in self._module_descriptors:
+            raise Exception("Duplicate name detected %s" % module_desc_name)
+        self._module_descriptors[module_desc_name] = desc
 
     @abstractmethod
     def query(self):