pyztk/test_machine.py

#!/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=10)
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('-ti', '--templates-index', type=int,
                    help='Get specific template', default=0)
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_index:
        print ("Read Single template... {}".format(args.templates_index))
        inicio = time.time()
        template = conn.get_user_template(args.templates_index, args.finger)
        final = time.time()
        print ('    took {:.3f}[s]'.format(final - inicio))
        print (" single! {}".format(template))
    elif 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)
    print ('')
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 ('')