Ensure we hold strong references to tasks

see python/cpython#88831
esphome · Feb 13, 2023 · 380f0b0 · 380f0b0
1 parent 35276b6
commit 380f0b0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/aioesphomeapi/connection.py b/aioesphomeapi/connection.py
@@ -118,6 +118,7 @@ def __init__(
         self._ping_stop_event = asyncio.Event()
 
         self._connect_task: Optional[asyncio.Task[None]] = None
+        self._keep_alive_task: Optional[asyncio.Task[None]] = None
         self._fatal_exception: Optional[Exception] = None
         self._expected_disconnect = False
 

diff --git a/aioesphomeapi/reconnect_logic.py b/aioesphomeapi/reconnect_logic.py
@@ -59,6 +59,7 @@ def __init__(
         self._wait_task_lock = asyncio.Lock()
         # Event for tracking when logic should stop
         self._stop_event = asyncio.Event()
+        self._stop_task: Optional[asyncio.Task[None]] = None
 
     @property
     def _is_stopped(self) -> bool: