Upstream/aioesphomeapi
1
0
Fork 0
mirror of https://github.com/esphome/aioesphomeapi.git synced 2024-12-27 17:37:39 +01:00
Code Issues Projects Releases Wiki Activity

Only try remove callbacks when not None (#268)

Browse Source
This commit is contained in:
Jesse Hills 2022-10-03 12:32:13 +13:00 committed by GitHub
parent fa41b09232
commit a6f18ae9df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
aioesphomeapi/client.py
View File

@ -421,8 +421,8 @@ class APIClient:
) )
def unsub() -> None: def unsub() -> None:
assert self._connection is not None if self._connection is not None:
self._connection.remove_message_callback(on_msg) self._connection.remove_message_callback(on_msg)
return unsub return unsub
@ -442,8 +442,8 @@ class APIClient:
) )
def unsub() -> None: def unsub() -> None:
assert self._connection is not None if self._connection is not None:
self._connection.remove_message_callback(on_msg) self._connection.remove_message_callback(on_msg)
return unsub return unsub
@ -474,8 +474,8 @@ class APIClient:
) )
def unsub() -> None: def unsub() -> None:
assert self._connection is not None if self._connection is not None:
self._connection.remove_message_callback(on_msg) self._connection.remove_message_callback(on_msg)
try: try:
async with async_timeout.timeout(timeout): async with async_timeout.timeout(timeout):
@ -633,7 +633,9 @@ class APIClient:
) )
async def stop_notify() -> None: async def stop_notify() -> None:
assert self._connection is not None if self._connection is None:
return
self._connection.remove_message_callback(on_msg) self._connection.remove_message_callback(on_msg)
self._check_authenticated() self._check_authenticated()