diff --git a/kasa/protocol.py b/kasa/protocol.py index f0828107..54fea080 100755 --- a/kasa/protocol.py +++ b/kasa/protocol.py @@ -28,7 +28,6 @@ class TPLinkSmartHomeProtocol: INITIALIZATION_VECTOR = 171 DEFAULT_PORT = 9999 DEFAULT_TIMEOUT = 5 - BLOCK_SIZE = 4 def __init__(self, host: str) -> None: @@ -111,7 +110,7 @@ class TPLinkSmartHomeProtocol: async def close(self): """Close the connection.""" writer = self.writer - self._reset() + self.reader = self.writer = None if writer: writer.close() with contextlib.suppress(Exception): @@ -119,10 +118,7 @@ class TPLinkSmartHomeProtocol: def _reset(self): """Clear any varibles that should not survive between loops.""" - self.writer = None - self.reader = None - self.query_lock = None - self.loop = None + self.reader = self.writer = self.loop = self.query_lock = None async def _query(self, request: str, retry_count: int, timeout: int) -> Dict: """Try to query a device."""