Fix ezsp flasher

custom_components/xiaomi_gateway3/core/ezsp.py

@@ -3,7 +3,6 @@
 import logging
 import socket
 import time
-from typing import Optional
 
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_create_clientsession
@@ -19,7 +18,7 @@
 async def update_zigbee_firmware(hass: HomeAssistant, host: str, custom: bool):
     tar_fw = "6.7.10.0" if custom else "6.6.2.0"
 
-    _LOGGER.debug(f"{host} [FWUP] Target zigbee firmware v{tar_fw}")
+    _LOGGER.info(f"{host} [FWUP] Target zigbee firmware v{tar_fw}")
 
     session = Session(host)
 
@@ -51,6 +50,7 @@ async def update_zigbee_firmware(hass: HomeAssistant, host: str, custom: bool):
 
         # some users have broken firmware, so unknown firmware also OK
         cur_fw = await read_firmware(host)
+        _LOGGER.info(f"{host} [FWUP] Firmware before update: {cur_fw}")
         if cur_fw and cur_fw.startswith(tar_fw):
             _LOGGER.debug(f"{host} [FWUP] No need to update")
             return True
@@ -83,6 +83,7 @@ async def update_zigbee_firmware(hass: HomeAssistant, host: str, custom: bool):
         await asyncio.sleep(2)
 
         cur_fw = await read_firmware(host)
+        _LOGGER.info(f"{host} [FWUP] Firmware after update: {cur_fw}")
         return cur_fw and cur_fw.startswith(tar_fw)
 
     except Exception as e:
@@ -96,23 +97,19 @@ async def update_zigbee_firmware(hass: HomeAssistant, host: str, custom: bool):
         await sh.close()
 
 
-async def read_firmware(host: str) -> Optional[str]:
-    ezsp = None
+async def read_firmware(host: str) -> str | None:
+    version = None
     try:
         from bellows.ezsp import EZSP
 
         ezsp = EZSP(
             {"path": f"socket://{host}:8889", "baudrate": 0, "flow_control": None}
         )
         await ezsp.connect(use_thread=False)
-        await ezsp.startup_reset()
         _, _, version = await ezsp.get_board_info()
+        await ezsp.disconnect()
     except Exception as e:
-        _LOGGER.debug(f"{host} [FWUP] Read firmware error: {e}")
-        return None
-    finally:
-        if ezsp:
-            ezsp.close()
+        _LOGGER.warning(f"{host} [FWUP] Read firmware error: {e}")
 
     _LOGGER.debug(f"{host} [FWUP] Current zigbee firmware v{version}")