juanboy
  1 +from flask import render_template
  2 +
  3 +def errorResponse(errorMsg):
  4 + return render_template('error.html',errorMsg=errorMsg)
  1 +import time
  2 +from flask import Blueprint, redirect, render_template, request,Flask, session
  3 +
  4 +from utils.query import query
  5 +from utils.errResp import errorResponse
  6 +
  7 +
  8 +ub = Blueprint('user',__name__,url_prefix='/user',template_folder='templates')
  9 +
  10 +@ub.route('/login',methods=['GET','POST'])
  11 +def login():
  12 + if request.method == 'GET':
  13 + return render_template('login.html')
  14 + else:
  15 + def filter_fn(user):
  16 + return request.form['username'] in user and request.form['password'] in user
  17 + users = query('select * from user', [], 'select')
  18 + login_success = list(filter(filter_fn,users))
  19 + if not len(login_success):return errorResponse('账号或密码错误')
  20 +
  21 + session['username'] = request.form['username']
  22 + return redirect('/page/home')
  23 +
  24 +@ub.route('/register',methods=['GET','POST'])
  25 +def register():
  26 + if request.method == 'GET':
  27 + return render_template('register.html')
  28 + else:
  29 + if request.form['password'] != request.form['checkPassword']:return errorResponse('两次密码不符合')
  30 + def filter_fn(user):
  31 + return request.form['username'] in user
  32 +
  33 + users = query('select * from user',[],'select')
  34 + filter_list = list(filter(filter_fn,users))
  35 + if len(filter_list):
  36 + return errorResponse('该用户名已被注册')
  37 + else:
  38 + time_tuple = time.localtime(time.time())
  39 + query('''
  40 + insert into user(username,password,createTime) values(%s,%s,%s)
  41 + ''',[request.form['username'],request.form['password'],str(time_tuple[0]) + '-' + str(time_tuple[1]) + '-' + str(time_tuple[2])])
  42 +
  43 + return redirect('/user/login')
  44 +
  45 +@ub.route('/logOut')
  46 +def logOut():
  47 + session.clear()
  48 + return redirect('/user/login')