#553 feat(zigbee): Support IEEE and NWK that are already the correct …

…type in ZigbeeMixin
TWilkin · Dec 7, 2024 · 57d8912 · 57d8912
1 parent e650f2c
commit 57d8912
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/controllers/zigbee/zigbee_controller/zigbee/device.py b/controllers/zigbee/zigbee_controller/zigbee/device.py
@@ -13,13 +13,13 @@ class ZigbeeMixin(InitialisableMixin):
     def __init__(
         self,
         controller: ZigbeeController,
-        ieee: str,
-        nwk: str,
+        ieee: str | EUI64,
+        nwk: str | NWK,
         **_
     ):
         self.__controller = controller
-        self.__ieee = EUI64.convert(ieee)
-        self.__nwk = NWK.convert(nwk[2:])
+        self.__ieee = ieee if isinstance(ieee, EUI64) else EUI64.convert(ieee)
+        self.__nwk = nwk if isinstance(nwk, NWK) else NWK.convert(nwk[2:])
 
     @property
     def ieee(self):