def index(): myForm = LoginForm() if myForm.validate_on_submit(): if session['recaptcha'].lower() == myForm.recaptcha.data.lower(): user = User(myForm.username.data, myForm.password.data) session['username'] = myForm.username.data if (user.isExisted()): return redirect(url_for('home')) else: flash(u'出错啦,登录失败,请检查用户名和密码是否正确!') else: flash(u'Hi, 验证码错误, 请重输!') return render_template('index.html', form=myForm)
def register(): regForm = registerForm() if regForm.validate_on_submit(): if session['recaptcha'].lower() == regForm.recaptcha.data.lower(): session['username'] = regForm.username.data query_obj = User(regForm.username.data, regForm.password.data) if (query_obj.isExisted()): flash(u'此用户名已经被使用,请重新注册!') del query_obj return render_template('register.html', form=regForm) else: user = User(regForm.username.data, regForm.password.data) user.add() flash(u'恭喜您,注册成功,您现在可以登录!') return redirect(url_for('index')) else: flash(u'Hi, 验证码错误, 请重输!') return render_template('register.html', form=regForm)