Merge pull request #65 from samson0v/master

Fixed getting device from message
thingsboard · Jun 17, 2024 · 9b68ae9 · 9b68ae9
2 parents f403cc7 + 731dcd8
commit 9b68ae9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tb_device_mqtt.py b/tb_device_mqtt.py
@@ -895,7 +895,7 @@ def _count_datapoints_in_message(data, device=None):
                 for data_object in data[device]:
                     datapoints += TBDeviceMqttClient._count_datapoints_in_message(data_object)  # noqa
             else:
-                datapoints += TBDeviceMqttClient._count_datapoints_in_message(data[device])  # noqa
+                datapoints += TBDeviceMqttClient._count_datapoints_in_message(data.get(device, data.get('device')))
         else:
             if isinstance(data, dict):
                 datapoints += TBDeviceMqttClient._get_data_points_from_message(data)