Close socket properly

vrachieru · Jun 3, 2019 · 22eac6a · 22eac6a
1 parent c626999
commit 22eac6a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tplink_smartplug/api.py b/tplink_smartplug/api.py
@@ -244,12 +244,14 @@ def command(self, cmd):
         if isinstance(cmd, dict):
             cmd = json.dumps(cmd)
 
+        sock = None
         try:
             sock = socket.create_connection((self.host, self.port), self.timeout)
             sock.send(self.encrypt(cmd))
             data = sock.recv(4096)
         finally:
-            sock.close()
+            if sock:
+                sock.close()
 
         response = self.decrypt(data[4:])
         response = json.loads(response)