262 lines
11 KiB
Python
Executable File
262 lines
11 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
# # -*- coding: utf-8 -*-
|
|
import sys
|
|
import traceback
|
|
import argparse
|
|
import time
|
|
import datetime
|
|
import codecs
|
|
from builtins import input
|
|
|
|
sys.path.append("zk")
|
|
|
|
from zk import ZK, const
|
|
from zk.user import User
|
|
from zk.finger import Finger
|
|
from zk.attendance import Attendance
|
|
from zk.exception import ZKErrorResponse, ZKNetworkError
|
|
|
|
class BasicException(Exception):
|
|
pass
|
|
|
|
conn = None
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='ZK Basic Reading Tests')
|
|
parser.add_argument('-a', '--address',
|
|
help='ZK device Address [192.168.1.201]', default='192.168.1.201')
|
|
parser.add_argument('-p', '--port', type=int,
|
|
help='ZK device port [4370]', default=4370)
|
|
parser.add_argument('-T', '--timeout', type=int,
|
|
help='Default [10] seconds (0: disable timeout)', default=1)
|
|
parser.add_argument('-P', '--password', type=int,
|
|
help='Device code/password', default=0)
|
|
parser.add_argument('-b', '--basic', action="store_true",
|
|
help='get Basic Information only. (no bulk read, ie: users)')
|
|
parser.add_argument('-f', '--force-udp', action="store_true",
|
|
help='Force UDP communication')
|
|
parser.add_argument('-v', '--verbose', action="store_true",
|
|
help='Print debug information')
|
|
parser.add_argument('-t', '--templates', action="store_true",
|
|
help='Get templates / fingers (compare bulk and single read)')
|
|
parser.add_argument('-tr', '--templates-raw', action="store_true",
|
|
help='Get raw templates (dump templates)')
|
|
parser.add_argument('-r', '--records', action="store_true",
|
|
help='Get attendance records')
|
|
parser.add_argument('-u', '--updatetime', action="store_true",
|
|
help='Update Date/Time')
|
|
parser.add_argument('-l', '--live-capture', action="store_true",
|
|
help='Live Event Capture')
|
|
parser.add_argument('-o', '--open-door', action="store_true",
|
|
help='Open door')
|
|
parser.add_argument('-D', '--deleteuser', type=int,
|
|
help='Delete a User (uid)', default=0)
|
|
parser.add_argument('-A', '--adduser', type=int,
|
|
help='Add a User (uid) (and enroll)', default=0)
|
|
parser.add_argument('-E', '--enrolluser', type=int,
|
|
help='Enroll a User (uid)', default=0)
|
|
parser.add_argument('-F', '--finger', type=int,
|
|
help='Finger for enroll (fid=0)', default=0)
|
|
|
|
args = parser.parse_args()
|
|
|
|
zk = ZK(args.address, port=args.port, timeout=args.timeout, password=args.password, force_udp=args.force_udp, verbose=args.verbose)
|
|
try:
|
|
print('Connecting to device ...')
|
|
conn = zk.connect()
|
|
print('SDK build=1 : %s' % conn.set_sdk_build_1()) # why?
|
|
print ('Disabling device ...')
|
|
conn.disable_device()
|
|
fmt = conn.get_extend_fmt()
|
|
print ('ExtendFmt : {}'.format(fmt))
|
|
fmt = conn.get_user_extend_fmt()
|
|
print ('UsrExtFmt : {}'.format(fmt))
|
|
print ('Face FunOn : {}'.format(conn.get_face_fun_on()))
|
|
print ('Face Version : {}'.format(conn.get_face_version()))
|
|
print ('Finger Version : {}'.format(conn.get_fp_version()))
|
|
print ('Old Firm compat : {}'.format(conn.get_compat_old_firmware()))
|
|
net = conn.get_network_params()
|
|
print ('IP:{} mask:{} gateway:{}'.format(net['ip'],net['mask'], net['gateway']))
|
|
now = datetime.datetime.today().replace(microsecond=0)
|
|
if args.updatetime:
|
|
print ('--- Updating Time---')
|
|
conn.set_time(now)
|
|
zk_time = conn.get_time()
|
|
dif = abs(zk_time - now).total_seconds()
|
|
print ('Time : {}'.format(zk_time))
|
|
if dif > 120:
|
|
print("WRN: TIME IS NOT SYNC!!!!!! (local: %s) use command -u to update" % now)
|
|
print ('Firmware Version : {}'.format(conn.get_firmware_version()))
|
|
print ('Platform : %s' % conn.get_platform())
|
|
print ('DeviceName : %s' % conn.get_device_name())
|
|
print ('Pin Width : %i' % conn.get_pin_width())
|
|
print ('Serial Number : %s' % conn.get_serialnumber())
|
|
print ('MAC: %s' % conn.get_mac())
|
|
print ('')
|
|
print ('--- sizes & capacity ---')
|
|
conn.read_sizes()
|
|
print (conn)
|
|
print ('')
|
|
if args.basic:
|
|
raise BasicException("Basic Info... Done!")
|
|
print ('--- Get User ---')
|
|
inicio = time.time()
|
|
users = conn.get_users()
|
|
final = time.time()
|
|
print (' took {:.3f}[s]'.format(final - inicio))
|
|
max_uid = 0
|
|
prev = None
|
|
if not args.deleteuser:
|
|
for user in users:
|
|
privilege = 'User'
|
|
if user.uid > max_uid:
|
|
max_uid = user.uid
|
|
privilege = 'User' if user.privilege == const.USER_DEFAULT else 'Admin-%s' % user.privilege
|
|
print ('-> UID #{:<5} Name : {:<27} Privilege : {}'.format(user.uid, user.name, privilege))
|
|
print (' Group ID : {:<8} User ID : {:<8} Password : {:<8} Card : {}'.format(user.group_id, user.user_id, user.password, user.card))
|
|
#print (len (user.repack73()), user.repack73().encode('hex'))
|
|
#print ('')
|
|
if args.adduser and user.uid == args.adduser:
|
|
prev = user
|
|
if args.deleteuser:
|
|
print ('')
|
|
print ('-- Delete User UID#%s ---' % args.deleteuser)
|
|
#TODO implementar luego
|
|
conn.delete_user(args.deleteuser)
|
|
users = conn.get_users() #update
|
|
for user in users:
|
|
if user.uid > max_uid:
|
|
max_uid = user.uid
|
|
privilege = 'User' if user.privilege == const.USER_DEFAULT else 'Admin-%s' % user.privilege
|
|
print ('-> UID #{:<5} Name : {:<27} Privilege : {}'.format(user.uid, user.name, privilege))
|
|
print (' Group ID : {:<8} User ID : {:<8} Password : {:<8} Card : {}'.format(user.group_id, user.user_id, user.password, user.card))
|
|
#print len (user.repack73()), user.repack73().encode('hex')
|
|
#print ''
|
|
if args.adduser and user.uid == args.adduser:
|
|
prev = user
|
|
print (' took {:.3f}[s]'.format(final - inicio))
|
|
|
|
if args.adduser:
|
|
uid = int(args.adduser)
|
|
if prev:
|
|
user = prev
|
|
privilege = 'User' if user.privilege == const.USER_DEFAULT else 'Admin-%s' % user.privilege
|
|
print ('')
|
|
print ('--- Modify User %i ---' % user.uid)
|
|
print ('-> UID #{:<5} Name : {:<27} Privilege : {}'.format(user.uid, user.name, privilege))
|
|
print (' Group ID : {:<8} User ID : {:<8} Password : {:<8} Card : {}'.format(user.group_id, user.user_id, user.password, user.card))
|
|
#discard prev
|
|
else:
|
|
print ('--- Add new User %i ---' % uid)
|
|
name = input('Name :')
|
|
admin = input('Admin (y/N):')
|
|
privilege = 14 if admin == 'y' else 0
|
|
password = input('Password :')
|
|
user_id = input('User ID2 :')
|
|
card = input('Card :')
|
|
card = int(card) if card else 0
|
|
#if prev:
|
|
# conn.delete_user(uid) #borrado previo
|
|
try:
|
|
conn.set_user(uid, name, privilege, password, '', user_id, card)
|
|
args.enrolluser = uid
|
|
except ZKErrorResponse as e:
|
|
print ("error: %s" % e)
|
|
#try new format
|
|
zk_user = User(uid, name, privilege, password, '', user_id, card)
|
|
conn.save_user_template(zk_user)# forced creation
|
|
args.enrolluser = uid
|
|
conn.refresh_data()
|
|
if args.enrolluser:
|
|
uid = int(args.enrolluser)
|
|
print ('--- Enrolling User #{} ---'.format(uid))
|
|
conn.delete_user_template(uid, args.finger)
|
|
conn.reg_event(0xFFFF) #
|
|
if conn.enroll_user(uid, args.finger):
|
|
conn.test_voice(18) # register ok
|
|
tem = conn.get_user_template(uid, args.finger)
|
|
print (tem)
|
|
else:
|
|
conn.test_voice(23) # not registered
|
|
conn.refresh_data()
|
|
#print ("Voice Test ...")
|
|
#conn.test_voice(10)
|
|
if args.templates or args.templates_raw:
|
|
print ("Read Templates...")
|
|
inicio = time.time()
|
|
templates = conn.get_templates()
|
|
final = time.time()
|
|
print (' took {:.3f}[s]'.format(final - inicio))
|
|
if args.templates:
|
|
print ('now checking individually...')
|
|
i = 0
|
|
for tem in templates:
|
|
i += 1
|
|
tem2 =conn.get_user_template(tem.uid,tem.fid)
|
|
if tem2 is None:
|
|
print ("%i: bulk! %s" % (i, tem))
|
|
elif tem == tem2: # compare with alternative method
|
|
print ("%i: OK! %s" % (i, tem))
|
|
else:
|
|
print ("%i: dif-1 %s" % (i, tem))
|
|
print ("%i: dif-2 %s" % (i, tem2))
|
|
print (' took {:.3f}[s]'.format(final - inicio))
|
|
else:
|
|
print ('template dump')
|
|
i = 0
|
|
for tem in templates:
|
|
i += 1
|
|
print ("%i: %s" % (i, tem.dump()))
|
|
print (' took {:.3f}[s]'.format(final - inicio))
|
|
|
|
if args.records:
|
|
print ("Read Records...")
|
|
inicio = time.time()
|
|
attendance = conn.get_attendance()
|
|
final = time.time()
|
|
print (' took {:.3f}[s]'.format(final - inicio))
|
|
i = 0
|
|
for att in attendance:
|
|
i += 1
|
|
print ("ATT {:>6}: uid:{:>3}, user_id:{:>8} t: {}, s:{} p:{}".format(i, att.uid, att.user_id, att.timestamp, att.status, att.punch))
|
|
print (' took {:.3f}[s]'.format(final - inicio))
|
|
print ('')
|
|
print ('--- sizes & capacity ---')
|
|
conn.read_sizes()
|
|
print (conn)
|
|
if args.open_door:
|
|
print ('')
|
|
print ('--- Opening door 10s ---')
|
|
conn.unlock(10)
|
|
print (' -- done!---')
|
|
if args.live_capture:
|
|
print ('')
|
|
print ('--- Live Capture! (press ctrl+C to break) ---')
|
|
counter = 0
|
|
for att in conn.live_capture():# using a generator!
|
|
if att is None:
|
|
#counter += 1 #enable to implemet a poorman timeout
|
|
print ("timeout {}".format(counter))
|
|
else:
|
|
print ("ATT {:>6}: uid:{:>3}, user_id:{:>8} t: {}, s:{} p:{}".format(counter, att.uid, att.user_id, att.timestamp, att.status, att.punch))
|
|
if counter >= 10:
|
|
conn.end_live_capture = True
|
|
print('')
|
|
print('--- capture End!---')
|
|
print ('')
|
|
except BasicException as e:
|
|
print (e)
|
|
except Exception as e:
|
|
print ("Process terminate : {}".format(e))
|
|
print ("Error: %s" % sys.exc_info()[0])
|
|
print ('-'*60)
|
|
traceback.print_exc(file=sys.stdout)
|
|
print ('-'*60)
|
|
finally:
|
|
if conn:
|
|
print ('Enabling device ...')
|
|
conn.enable_device()
|
|
conn.disconnect()
|
|
print ('ok bye!')
|
|
print ('')
|