Clean API only on close

zigpy · Oct 27, 2024 · 5780b39 · 5780b39
1 parent 68dd3a0
commit 5780b39
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/zigpy_zigate/uart.py b/zigpy_zigate/uart.py
@@ -20,13 +20,16 @@ def __init__(self, api):
         super().__init__()
         self._api = api
 
-    def connection_lost(self, exc) -> None:
+    def connection_lost(self, exc: Exception | None) -> None:
         """Port was closed expectedly or unexpectedly."""
         super().connection_lost(exc)
 
         if self._api is not None:
             self._api.connection_lost(exc)
-            self._api = None
+
+    def close(self):
+        super().close()
+        self._api = None
 
     def send(self, cmd, data=b""):
         """Send data, taking care of escaping and framing"""