refactor: fix typing issues

dbuezas · Jan 16, 2024 · de4d4df · de4d4df
1 parent 12506a8
commit de4d4df
Show file tree

Hide file tree

Showing 3 changed files with 79 additions and 2 deletions.
diff --git a/eq3btsmart/bleakconnection.py b/eq3btsmart/bleakconnection.py
@@ -128,7 +128,7 @@ async def async_get_connection(self) -> BleakClient:
 
         self._on_connection_event()
 
-        if self._conn.is_connected:
+        if self._conn is not None and self._conn.is_connected:
             _LOGGER.debug("[%s] Connected", self._name)
         else:
             raise BackendException("Can't connect")

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,6 +15,7 @@ bleak = "^0.21.1"
 construct = "^2.10.68"
 construct-typing = "^0.6.2"
 homeassistant = "^2023.1.1"
+bleak-retry-connector = "^3.4.0"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.1.13"