From 850bfe578f669100beacba80fd5fbd6b92cd47e8 Mon Sep 17 00:00:00 2001
From: Arturo Hernandez <ahernandez@agetic.gob.bo>
Date: Mon, 2 Jul 2018 16:25:51 -0400
Subject: [PATCH] fixed self connect detections

---
 zk/base.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/zk/base.py b/zk/base.py
index 29baa14..4e7d876 100644
--- a/zk/base.py
+++ b/zk/base.py
@@ -131,6 +131,10 @@ class ZK(object):
         self.__data_recv = None
         self.__data = None
 
+    def __nonzero__(self):
+        """ for boolean test"""
+        return self.is_connect
+
     def __create_socket(self):
         """ based on self.tcp"""
         if self.tcp:
@@ -342,9 +346,9 @@ class ZK(object):
         '''
         diconnect from the connected device
         '''
+        self.is_connect = False
         cmd_response = self.__send_command(const.CMD_EXIT)
         if cmd_response.get('status'):
-            self.is_connect = False
             if self.__sock:
                 self.__sock.close() #leave to GC
             return True