diff --git a/test.py b/test.py index 72508a6..c38d345 100644 --- a/test.py +++ b/test.py @@ -8,7 +8,15 @@ zk = zk.ZK('192.168.1.201') status, message = zk.connect() if status: print 'Firmware Version: : {}'.format(zk.get_firmware_version()) - zk.restart() + users = zk.get_users() + if users: + for uid in users: + if users[uid][2] == 14: + level = 'Admin' + else: + level = 'User' + print "[UID %d]: ID: %s, Name: %s, Level: %s, Password: %s" % ( uid, users[uid][0], users[uid][1], level, users[uid][3] ) + print zk.disconnect() else: print message diff --git a/zk/base.py b/zk/base.py index 68d3e3e..2554ef8 100644 --- a/zk/base.py +++ b/zk/base.py @@ -133,6 +133,68 @@ class ZK(object): except Exception, e: return (False, e) + def __get_size_user(self): + """Checks a returned packet to see if it returned CMD_PREPARE_DATA, + indicating that data packets are to be sent + + Returns the amount of bytes that are going to be sent""" + response = self.__response + if response == const.CMD_PREPARE_DATA: + size = unpack('I', self.__data_recv[8:12])[0] + return size + else: + return 0 + + def get_users(self): + """Start a connection with the time clock""" + command = const.CMD_USERTEMP_RRQ + command_string = '\x05' + chksum = 0 + + buf = self.__create_header(command=command, chksum=chksum, session_id=self.__sesion_id, reply_id=self.__reply_id, command_string=command_string) + self.__sending_packet(buf) + try: + bytes = self.__get_size_user() + userdata = [] + while bytes > 0: + data_recv, addr = self.__sock.recvfrom(1032) + userdata.append(data_recv) + bytes -= 1024 + + self.__sock.recvfrom(8) + + users = {} + if len(userdata) > 0: + # The first 4 bytes don't seem to be related to the user + for x in xrange(len(userdata)): + if x > 0: + userdata[x] = userdata[x][8:] + + userdata = ''.join(userdata) + userdata = userdata[11:] + + while len(userdata) > 72: + uid, role, password, name, userid = unpack( '2s2s8s28sx31s', userdata.ljust(72)[:72]) + + uid = int( uid.encode("hex"), 16) + # Clean up some messy characters from the user name + password = password.split('\x00', 1)[0] + password = unicode(password.strip('\x00|\x01\x10x'), errors='ignore') + + userid = unicode(userid.strip('\x00|\x01\x10x'), errors='ignore') + + name = name.split('\x00', 1)[0] + + if not name: + name = uid + + users[uid] = (userid, name, int( role.encode("hex"), 16 ), password) + + userdata = userdata[72:] + + return users + except Exception, e: + return (False, e) def power_off(self): '''