def register():
if request.method == 'POST':
name = request.form['name']
email = request.form['email']
password1 = request.form['password1']
password2 = request.form['password2']
user = db.session.query(User).filter_by(email=email).first()
if user:
flash(" 此邮箱已被注册")
return render_template('register.html')
if password1 != password2:
flash("两次输入密码不一致")
return render_template('register.html')
user = User()
user.name = name
user.email = email
user.password = password1
user.role_id = 2
db.session.add(user)
db.session.commit()
log = Log()
log.user = name
log.operate = '帐号注册'
db.session.add(log)
db.session.commit()
return redirect(url_for('auth.login'))
return render_template('register.html')
def logout():
log = Log()
log.user = current_user.name
log.operate = '帐号登出'
db.session.add(log)
db.session.commit()
logout_user()
return redirect(url_for('auth.login'))
def login():
if request.method == 'POST':
email = request.form['email']
password = request.form['password']
user = db.session.query(User).filter_by(email=email).first()
if user:
if user.password == password:
login_user(user, remember=True)
log = Log()
log.user = user.name
log.operate = '帐号登录'
db.session.add(log)
db.session.commit()
return redirect(url_for('vd.index'))
else:
flash("密码错误，再试一次")
return render_template('login.html')
else:
flash("没有此用户，检查输入")
return render_template('login.html')
return render_template('login.html')
