update base class

This commit is contained in:
Fanani M. Ihsan 2016-05-26 12:45:28 +07:00
parent d3a0e90ad7
commit 616641d073

View File

@ -8,8 +8,9 @@ from zk.user import User
class ZK(object): class ZK(object):
__data_recv = None __data_recv = None
is_connect = False
def __init__(self, ip, port=4370, timeout=5): def __init__(self, ip, port=4370, timeout=60):
self.__address = (ip, port) self.__address = (ip, port)
self.__sock = socket(AF_INET, SOCK_DGRAM) self.__sock = socket(AF_INET, SOCK_DGRAM)
self.__sock.settimeout(timeout) self.__sock.settimeout(timeout)
@ -116,6 +117,7 @@ class ZK(object):
cmd_response = self.__send_command(command) cmd_response = self.__send_command(command)
cmd_response['data'] = '' cmd_response['data'] = ''
if cmd_response.get('status'): if cmd_response.get('status'):
self.is_connect = True
cmd_response['message'] = 'connected' cmd_response['message'] = 'connected'
return cmd_response return cmd_response
else: else:
@ -289,6 +291,24 @@ class ZK(object):
cmd_response['data'] = '' cmd_response['data'] = ''
return cmd_response return cmd_response
def cancel_capture(self):
command = const.CMD_CANCELCAPTURE
cmd_response = self.__send_command(command=command)
print cmd_response
def verify_user(self):
command = const.CMD_STARTVERIFY
# uid = chr(uid % 256) + chr(uid >> 8)
cmd_response = self.__send_command(command=command)
print cmd_response
def enroll_user(self, uid):
command = const.CMD_STARTENROLL
uid = chr(uid % 256) + chr(uid >> 8)
command_string = pack('2s', uid)
cmd_response = self.__send_command(command=command, command_string=command_string)
print cmd_response
def clear_user(self): def clear_user(self):
''' '''
Not implemented yet Not implemented yet