chickliu/ha_xiaomi_home
1
0
Fork 0
mirror of https://github.com/XiaoMi/ha_xiaomi_home.git synced 2026-01-17 23:50:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix mqtt broker check logic

Browse Source
This commit is contained in:
sworld 2024-12-30 18:01:32 +08:00
parent 48348201a2
commit e169df4980
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
custom_components/xiaomi_home/config_flow.py
View File

@ -358,13 +358,12 @@ class XiaomiMihomeConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
# pylint: disable=import-outside-toplevel # pylint: disable=import-outside-toplevel
try: try:
from paho.mqtt import client from paho.mqtt import client
from paho.mqtt.enums import MQTTErrorCode
mqtt_client = client.Client( mqtt_client = client.Client(
client_id=f'ha.{self._uid}', client_id=f'ha.{self._uid}',
protocol=client.MQTTv5) protocol=client.MQTTv5) # type: ignore
if mqtt_client.connect( if mqtt_client.connect(
host=f'{self._cloud_server}-ha.mqtt.io.mi.com', host=f'{self._cloud_server}-ha.mqtt.io.mi.com',
port=8883) != MQTTErrorCode.MQTT_ERR_SUCCESS: port=8883) != 0:
raise RuntimeError('mqtt connect error') raise RuntimeError('mqtt connect error')
mqtt_client.disconnect() mqtt_client.disconnect()
except Exception as err: # pylint: disable=broad-exception-caught except Exception as err: # pylint: disable=broad-exception-caught
@ -1811,13 +1810,12 @@ class OptionsFlowHandler(config_entries.OptionsFlow):
# pylint: disable=import-outside-toplevel # pylint: disable=import-outside-toplevel
try: try:
from paho.mqtt import client from paho.mqtt import client
from paho.mqtt.enums import MQTTErrorCode
mqtt_client = client.Client( mqtt_client = client.Client(
client_id=f'ha.{self._uid}', client_id=f'ha.{self._uid}',
protocol=client.MQTTv5) protocol=client.MQTTv5) # type: ignore
if mqtt_client.connect( if mqtt_client.connect(
host=f'{self._cloud_server}-ha.mqtt.io.mi.com', host=f'{self._cloud_server}-ha.mqtt.io.mi.com',
port=8883) != MQTTErrorCode.MQTT_ERR_SUCCESS: port=8883) != 0:
raise RuntimeError('mqtt connect error') raise RuntimeError('mqtt connect error')
mqtt_client.disconnect() mqtt_client.disconnect()
except Exception as err: # pylint: disable=broad-exception-caught except Exception as err: # pylint: disable=broad-exception-caught