Fix InvalidStateError: invalid state when enable statistics sensors.

AlexxIT · Sep 18, 2024 · a0ed833 · a0ed833
1 parent 82ad2d9
commit a0ed833
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/custom_components/xiaomi_gateway3/core/gate/miot.py b/custom_components/xiaomi_gateway3/core/gate/miot.py
@@ -90,12 +90,17 @@ async def mqtt_publish_multiple(
         self, device: XDevice, payload: dict, gw2, delay: float = 1.0
     ):
         fut = asyncio.get_event_loop().create_future()
-        device.add_listener(fut.set_result)
+
+        def try_set_result(r):
+            if not fut.done():
+                fut.set_result(r)
+
+        device.add_listener(try_set_result)
         await self.mqtt.publish("miio/command", payload)
         try:
             async with asyncio.timeout(delay):
                 await fut
         except TimeoutError:
             await gw2.mqtt.publish("miio/command", payload)
         finally:
-            device.remove_listener(fut.set_result)
+            device.remove_listener(try_set_result)