kpt

【user.py】添加登录接口

  1 +from flask import Blueprint, redirect, render_template, request,Flask, session
  2 +
  3 +from utils.query import query
  4 +from utils.errResp import errorResponse
  5 +
  6 +
  7 +ub = Blueprint('user',__name__,url_prefix='/user',template_folder='templates')
  8 +
  9 +@ub.route('/login',methods=['GET','POST'])
  10 +def login():
  11 + if request.method == 'GET':
  12 + return render_template('login.html')
  13 + else:
  14 + def filter_fn(user):
  15 + return request.form['username'] in user and request.form['password'] in user
  16 + users = query('select * from user', [], 'select')
  17 + login_success = list(filter(filter_fn,users))
  18 + if not len(login_success):return errorResponse('账号或密码错误')
  19 +
  20 + session['username'] = request.form['username']
  21 + return redirect('/page/home')