Add WallSwitch device type and autogenerate supported devices docs (#758)

kasa/device_factory.py

@@ -5,9 +5,18 @@ from typing import Any, Dict, Optional, Tuple, Type
 
 from .aestransport import AesTransport
 from .device import Device
+from .device_type import DeviceType
 from .deviceconfig import DeviceConfig
 from .exceptions import KasaException, UnsupportedDeviceError
-from .iot import IotBulb, IotDevice, IotDimmer, IotLightStrip, IotPlug, IotStrip
+from .iot import (
+    IotBulb,
+    IotDevice,
+    IotDimmer,
+    IotLightStrip,
+    IotPlug,
+    IotStrip,
+    IotWallSwitch,
+)
 from .iotprotocol import IotProtocol
 from .klaptransport import KlapTransport, KlapTransportV2
 from .protocol import (
@@ -105,7 +114,7 @@ async def _connect(config: DeviceConfig, protocol: BaseProtocol) -> "Device":
         )
 
 
-def get_device_class_from_sys_info(info: Dict[str, Any]) -> Type[IotDevice]:
+def _get_device_type_from_sys_info(info: Dict[str, Any]) -> DeviceType:
     """Find SmartDevice subclass for device described by passed data."""
     if "system" not in info or "get_sysinfo" not in info["system"]:
         raise KasaException("No 'system' or 'get_sysinfo' in response")
@@ -116,22 +125,36 @@ def get_device_class_from_sys_info(info: Dict[str, Any]) -> Type[IotDevice]:
         raise KasaException("Unable to find the device type field!")
 
     if "dev_name" in sysinfo and "Dimmer" in sysinfo["dev_name"]:
-        return IotDimmer
+        return DeviceType.Dimmer
 
     if "smartplug" in type_.lower():
         if "children" in sysinfo:
-            return IotStrip
-
-        return IotPlug
+            return DeviceType.Strip
+        if (dev_name := sysinfo.get("dev_name")) and "light" in dev_name.lower():
+            return DeviceType.WallSwitch
+        return DeviceType.Plug
 
     if "smartbulb" in type_.lower():
         if "length" in sysinfo:  # strips have length
-            return IotLightStrip
+            return DeviceType.LightStrip
 
-        return IotBulb
+        return DeviceType.Bulb
     raise UnsupportedDeviceError("Unknown device type: %s" % type_)
 
 
+def get_device_class_from_sys_info(sysinfo: Dict[str, Any]) -> Type[IotDevice]:
+    """Find SmartDevice subclass for device described by passed data."""
+    TYPE_TO_CLASS = {
+        DeviceType.Bulb: IotBulb,
+        DeviceType.Plug: IotPlug,
+        DeviceType.Dimmer: IotDimmer,
+        DeviceType.Strip: IotStrip,
+        DeviceType.WallSwitch: IotWallSwitch,
+        DeviceType.LightStrip: IotLightStrip,
+    }
+    return TYPE_TO_CLASS[_get_device_type_from_sys_info(sysinfo)]
+
+
 def get_device_class_from_family(device_type: str) -> Optional[Type[Device]]:
     """Return the device class from the type name."""
     supported_device_types: Dict[str, Type[Device]] = {