|
15 | 15 | from zigpy.zdo.types import ZDOCmd, ZDOHeader, CLUSTERS as ZDO_CLUSTERS
|
16 | 16 |
|
17 | 17 | from zigpy.zcl import clusters
|
18 |
| -from zigpy.types import ExtendedPanId, deserialize as list_deserialize |
| 18 | +from zigpy.types import ( |
| 19 | + ExtendedPanId, |
| 20 | + deserialize as list_deserialize, |
| 21 | + Struct as ZigpyStruct, |
| 22 | +) |
19 | 23 | from zigpy.exceptions import DeliveryError
|
20 | 24 |
|
21 | 25 | import zigpy_znp.config as conf
|
@@ -156,7 +160,23 @@ def _receive_zdo_message(
|
156 | 160 | field_names, field_types = ZDO_CLUSTERS[cluster]
|
157 | 161 | assert set(zdo_kwargs) == set(field_names)
|
158 | 162 |
|
159 |
| - zdo_args = [t(zdo_kwargs[n]) for n, t in zip(field_names, field_types)] |
| 163 | + # Type cast all of the field args and kwargs |
| 164 | + zdo_args = [] |
| 165 | + |
| 166 | + for name, field_type in zip(field_names, field_types): |
| 167 | + zdo_arg = zdo_kwargs[name] |
| 168 | + |
| 169 | + if issubclass(field_type, ZigpyStruct) and hasattr(ZigpyStruct, "_fields"): |
| 170 | + # Old-style zigpy structs do not have "copy constructors" |
| 171 | + new_obj = field_type() |
| 172 | + |
| 173 | + for field_name, _ in new_obj._fields: |
| 174 | + setattr(new_obj, field_name, getattr(zdo_arg, field_name)) |
| 175 | + else: |
| 176 | + new_obj = field_type(zdo_arg) |
| 177 | + |
| 178 | + zdo_args.append(zdo_arg) |
| 179 | + |
160 | 180 | message = t.serialize_list([t.uint8_t(tsn)] + zdo_args)
|
161 | 181 |
|
162 | 182 | LOGGER.debug("Pretending we received a ZDO message: %s", message)
|
|
0 commit comments