From e169df49808d7e00cd9942bb6c08efcc86f213ab Mon Sep 17 00:00:00 2001 From: sworld Date: Mon, 30 Dec 2024 18:01:32 +0800 Subject: [PATCH] fix: fix mqtt broker check logic --- custom_components/xiaomi_home/config_flow.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/custom_components/xiaomi_home/config_flow.py b/custom_components/xiaomi_home/config_flow.py index 2372b2c..1dac2a9 100644 --- a/custom_components/xiaomi_home/config_flow.py +++ b/custom_components/xiaomi_home/config_flow.py @@ -358,13 +358,12 @@ class XiaomiMihomeConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): # pylint: disable=import-outside-toplevel try: from paho.mqtt import client - from paho.mqtt.enums import MQTTErrorCode mqtt_client = client.Client( client_id=f'ha.{self._uid}', - protocol=client.MQTTv5) + protocol=client.MQTTv5) # type: ignore if mqtt_client.connect( host=f'{self._cloud_server}-ha.mqtt.io.mi.com', - port=8883) != MQTTErrorCode.MQTT_ERR_SUCCESS: + port=8883) != 0: raise RuntimeError('mqtt connect error') mqtt_client.disconnect() except Exception as err: # pylint: disable=broad-exception-caught @@ -1811,13 +1810,12 @@ class OptionsFlowHandler(config_entries.OptionsFlow): # pylint: disable=import-outside-toplevel try: from paho.mqtt import client - from paho.mqtt.enums import MQTTErrorCode mqtt_client = client.Client( client_id=f'ha.{self._uid}', - protocol=client.MQTTv5) + protocol=client.MQTTv5) # type: ignore if mqtt_client.connect( host=f'{self._cloud_server}-ha.mqtt.io.mi.com', - port=8883) != MQTTErrorCode.MQTT_ERR_SUCCESS: + port=8883) != 0: raise RuntimeError('mqtt connect error') mqtt_client.disconnect() except Exception as err: # pylint: disable=broad-exception-caught