人脸识别考勤系统，用 Python 实现人脸识别考勤系统

import time import warnings import functools import base64 from PIL import Image import numpy as np import sqlite3 from io import BytesIO from flaskr.facenet_keras import ( face_encodings ) import click import time from flask import ( Blueprint, flash, g, redirect, render_template, request, session, url_for ) from werkzeug.security import check_password_hash, generate_password_hash from flaskr.db import get_db, init_db bp = Blueprint('admin', __name__, url_prefix='/admin') @bp.route('/') def root() : return redirect('/admin/index') @bp.route('/index') def index(): # session 中没有记录，重定向到登陆页面 if session.get('is_admin') is None : return render_template('/admin/login.html') # session 中的权限不足，给出提示并重定向 if session.get('is_admin') < 1 : return render_template('/showmsg.html', title='无权访问！', msg='权限不足，请重新登陆。', jump='/admin/login') db = get_db() usertab = db.execute( 'SELECT id, username, is_admin, last_punch FROM user' ) js_code = 'let datas = ' \ + str(np.load('face_features.npy').tolist()) + ';' try : return render_template('admin/index.html', username=session['username'], usertab=usertab, js_code=js_code) # 当 session 不存在 'username' 字段时，重定向到登陆页面 except KeyError : session.clear() return redirect('/admin/login') # 登陆页面 @bp.route('/login', methods=('GET', 'POST')) def login() : return render_template('/admin/login.html') # 发送登陆请求 @bp.route('/login/login', methods=('GET', 'POST')) def login_login(): if request.method == 'POST' : session.clear() username = request.form['username'] password = request.form['password'] db = get_db() try : user = db.execute( 'SELECT * FROM user WHERE username = ?', (username,) ).fetchone() except sqlite3.OperationalError : init_db() return render_template('/showmsg.html', title='登陆失败！', msg='数据库尚未初始化，请稍后再试。', jump='/admin/login') if user is None: return render_template('/showmsg.html', title='登陆失败！', msg='用户不存在，请检查后重试。', jump='/admin/login') elif not check_password_hash(user['password'], password): return render_template('/showmsg.html', title='登陆失败！', msg='密码错误。', jump='/admin/login') elif not user['is_admin'] >= 1 : return render_template('/showmsg.html', title='登陆失败！', msg='权限不足。', jump='/admin/login') session['id'] = user['id'] session['username'] = user['username'] session['is_admin'] = user['is_admin'] return redirect('/admin/index') return render_template('/showmsg.html', title='登陆失败！', msg='request.method 应为 POST。', jump='/admin/login') # 删除人脸数据 @bp.route('/delete', methods=('GET', 'POST')) def delete() : if request.method == 'POST' : idx = request.json['idx'] db = get_db() user = db.execute( 'SELECT * FROM user' ).fetchall() features = np.load('face_features.npy') features = np.delete(features, idx, axis=0) np.save('face_features.npy', features) # 数据库中的 id 到 np.ndarray 中的 index 的转换 idx = user[idx]['id'] db.execute( 'DELETE FROM user WHERE id = ?', (idx,)) db.commit() return '{"msg" : "delete success"}' return render_template('/showmsg.html', title='删除失败！', msg='request.method 应为 POST。', jump='/admin/index') # 登出 @bp.route('/logout', methods=('GET', 'POST')) def logout(): session.clear() return redirect('/admin/login') @bp.before_app_request def load_logged_in_user(): is_admin = session.get('is_admin') if is_admin is None: g.is_admin = None else: g.is_admin = is_admin def login_required(view): @functools.wraps(view) def wrapped_view(**kwargs): if g.is_admin is None or g.is_admin <= 0: return redirect('/admin/login') return view(**kwargs) return wrapped_view # 注册新用户 @bp.route('/register', methods=('GET', 'POST')) @login_required def register() : return render_template('/admin/register.html') # 主页页面的基本信息 @bp.route('/register/basic_info', methods=('GET', 'POST')) @login_required def register_basic_info() : if request.method == 'POST' : session['basic_info'] = { 'username' : request.form['username'], 'is_admin' : request.form['is_admin'], # 不储存密码明文 'password' : generate_password_hash(request.form['password']), # 当前日期与时间 'last_punch' : time.strftime("%Y/%m/%d %H:%M:%S", time.localtime()) } return render_template('/admin/face.html') return render_template('/showmsg.html', title='注册失败！', msg='request.method 应为 POST。', jump='/admin/register') @bp.before_app_request def load_basic_info() : basic_info = session.get('basic_info') if basic_info is None : g.basic_info = None else : g.basic_info = basic_info def basic_info_registered(view) : @functools.wraps(view) def wrapped_view(**kwargs): if g.basic_info is None : return redirect('/admin/index') return view(**kwargs) return wrapped_view # 注册页面的人脸录入 @bp.route('/register/recognize', methods=('GET', 'POST')) @login_required @basic_info_registered def register_recognize(): if request.method == 'POST' : data = request.json['data'] string_b64 = request.json['imgbase64'] if not data or not string_b64 : return render_template('/showmsg.html', title='注册失败！', msg='图片解析错误，请检查后重试。', jump='/admin/index') img_data = base64.b64decode(string_b64[22:]) img = Image.open(BytesIO(img_data)).convert('RGB') img = np.asarray(img) try : feature = face_encodings(img) except IndexError : return render_template('/showmsg.html', title='检测失败！', msg='未检测到人脸，请重试。', jump='/admin/index') feature = np.expand_dims(feature, axis=0) features = np.load('face_features.npy') features = np.concatenate([features, feature]) np.save('face_features.npy', features) db = get_db() db.execute( "INSERT INTO user (username, password, is_admin, last_punch) \ VALUES (?, ?, ?, ?)", ( session['basic_info']['username'], session['basic_info']['password'], session['basic_info']['is_admin'], session['basic_info']['last_punch'],)) try : db.commit() except sqlite3.OperationalError : return render_template('/showmsg.html', title='注册失败！', msg='数据库被锁，请稍后再试。', jump='/admin/index') return render_template('/showmsg.html', title='注册成功！', msg='您的账户已经注册成功。', jump='/admin/index') return render_template('/showmsg.html', title='注册失败！', msg='request.method 应为 POST。', jump='/admin/index') # 修改密码 @bp.route('/change_password', methods=('GET', 'POST')) @login_required def change_password() : return render_template('admin/change_password.html') # 修改密码的请求 @bp.route('/change_password/change_password', methods=('GET', 'POST')) def change_password_change_password() : if request.method == 'POST' : idx = session['id'] password = request.form['password'] db =