From d70737ea2b6c174eabd3f92095a6139566485b5e Mon Sep 17 00:00:00 2001
From: Fabio Manganiello <fabio@manganiello.tech>
Date: Fri, 9 Aug 2024 11:03:04 +0200
Subject: [PATCH] [zigbee.mqtt] Added more logging lines on `device_set`
 action.

---
 platypush/plugins/zigbee/mqtt/__init__.py | 35 +++++++++++++++++++----
 1 file changed, 29 insertions(+), 6 deletions(-)

diff --git a/platypush/plugins/zigbee/mqtt/__init__.py b/platypush/plugins/zigbee/mqtt/__init__.py
index 73aa88429b..f70de46062 100644
--- a/platypush/plugins/zigbee/mqtt/__init__.py
+++ b/platypush/plugins/zigbee/mqtt/__init__.py
@@ -1154,6 +1154,14 @@ class ZigbeeMqttPlugin(
             if self._is_write_only(stored_property):
                 reply_topic = None
 
+        if property and reply_topic:
+            self.logger.debug(
+                'Waiting for updated value of %s for device %s, reply topic: %s',
+                property,
+                device,
+                reply_topic,
+            )
+
         properties = self._run_request(
             topic=self._topic(device + '/set'),
             reply_topic=reply_topic,
@@ -1162,9 +1170,22 @@ class ZigbeeMqttPlugin(
         )
 
         if property and reply_topic:
-            assert (
-                property in properties
-            ), f'Could not retrieve the new state for {property}'
+            if property not in (properties or {}):
+                self.logger.warning(
+                    (
+                        'Could not retrieve value of property %s on device %s, '
+                        'reply topic: %s, retrieved properties: %s',
+                    ),
+                    property,
+                    device,
+                    reply_topic,
+                    properties,
+                )
+
+                raise AssertionError(
+                    f'Cound not retrieve the new state for {property} on device {device}'
+                )
+
             return {property: properties[property]}
 
         return properties
@@ -1670,9 +1691,11 @@ class ZigbeeMqttPlugin(
             prop,
             prop_info.get(
                 'value_toggle',
-                'OFF'
-                if device_state.get(prop) == prop_info.get('value_on', 'ON')
-                else 'ON',
+                (
+                    'OFF'
+                    if device_state.get(prop) == prop_info.get('value_on', 'ON')
+                    else 'ON'
+                ),
             ),
         )