Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2025-12-20 03:30:12 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Make close behaviour consistent across new protocols and transports (#660)

Browse Source
This commit is contained in:
Steven B
2024-01-20 12:35:05 +00:00
committed by GitHub
parent e94cd118a4
commit 49cfef087c
4 changed files with 46 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
kasa/iotprotocol.py
View File

@@ -45,8 +45,8 @@ class IotProtocol(TPLinkProtocol):
try:
return await self._execute_query(request, retry)
except ConnectionException as sdex:
await self.close()
if retry >= retry_count:
await self.close()
_LOGGER.debug("Giving up on %s after %s retries", self._host, retry)
raise sdex
continue
@@ -57,14 +57,14 @@ class IotProtocol(TPLinkProtocol):
)
raise auex
except RetryableException as ex:
await self.close()
if retry >= retry_count:
await self.close()
_LOGGER.debug("Giving up on %s after %s retries", self._host, retry)
raise ex
continue
except TimeoutException as ex:
await self.close()
if retry >= retry_count:
await self.close()
_LOGGER.debug("Giving up on %s after %s retries", self._host, retry)
raise ex
await asyncio.sleep(self.BACKOFF_SECONDS_AFTER_TIMEOUT)
@@ -85,5 +85,10 @@ class IotProtocol(TPLinkProtocol):
return await self._transport.send(request)
async def close(self) -> None:
"""Close the protocol."""
"""Close the underlying transport.
Some transports may close the connection, and some may
use this as a hint that they need to reconnect, or
reauthenticate.
"""
await self._transport.close()