Add own device type for smartstrip children (#201)

kasa/smartdevice.py

@@ -16,7 +16,7 @@ import inspect
 import logging
 from dataclasses import dataclass
 from datetime import datetime, timedelta
-from enum import Enum
+from enum import Enum, auto
 from typing import Any, Dict, List, Optional
 
 from .exceptions import SmartDeviceException
@@ -28,11 +28,12 @@ _LOGGER = logging.getLogger(__name__)
 class DeviceType(Enum):
     """Device type enum."""
 
-    Plug = 1
+    Plug = auto()
-    Bulb = 2
+    Bulb = auto()
-    Strip = 3
+    Strip = auto()
-    Dimmer = 4
+    StripSocket = auto()
-    LightStrip = 5
+    Dimmer = auto()
+    LightStrip = auto()
     Unknown = -1
 
 
@@ -743,6 +744,11 @@ class SmartDevice:
         """Return True if the device is a strip."""
         return self._device_type == DeviceType.Strip
 
+    @property
+    def is_strip_socket(self) -> bool:
+        """Return True if the device is a strip socket."""
+        return self._device_type == DeviceType.StripSocket
+
     @property
     def is_dimmer(self) -> bool:
         """Return True if the device is a dimmer."""

kasa/smartstrip.py

@@ -211,6 +211,7 @@ class SmartStripPlug(SmartPlug):
         self.child_id = child_id
         self._last_update = parent._last_update
         self._sys_info = parent._sys_info
+        self._device_type = DeviceType.StripSocket
 
     async def update(self):
         """Override the update to no-op and inform the user."""