From 33bc38b57f4a87d83f051e9636fffb491a43ac10 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Sun, 26 Sep 2021 12:38:33 -0500
Subject: [PATCH] Fix lock being unexpectedly reset on close (#218)

* Implement a backoff for legacy devices

* do not clear self.query_lock at close()

* revert backoff
---
 kasa/protocol.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

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."""