def register():
global exist
global flag
global repassword
global password_lenth
repassword = 1
password_lenth = 1
exist = 0
flag = 0
if request.method == 'POST':
new_username = request.form.get("Name")
if User.query.filter_by(Name=new_username).all():
exist = 1
# flash(u"注册失败！！用户名已存在! 换个更个性的用户名吧 -_-", category="danger")
elif request.form.get("Password") != request.form.get("repassword"):
repassword = 0
return render_template('register.html',
exist=exist,
flag=flag,
title=u"注册",
repassword=repassword)
elif len(request.form.get("Password")) < 6:
password_lenth = 0
return render_template('register.html',
exist=exist,
flag=flag,
title=u"注册",
password_lenth=password_lenth)
else:
user_forsql = User()
user_forsql.Password = request.form.get("Password")
user_forsql.Name = request.form.get("Name")
user_forsql.Gender = request.form.get("Gender")
user_forsql.Birth = request.form.get("Birth")
user_forsql.Position = request.form.get("Position")
user_forsql.Email = request.form.get("Email")
user_forsql.Tel_Number = request.form.get("Tel")
db.session.add(user_forsql)
db.session.commit()
flag = 1
# flash("恭喜您！注册成功", category="success")
return render_template('register.html',
exist=exist,
flag=flag,
title=u"注册")
def register():
"""
进行登录验证
:return:
"""
error_msg = None
success = None
if request.method == 'GET':
# 获取 GET 请求参数
phone_number = request.args.get('mobile_phone_number')
if phone_number is not None:
if sms.send_message(phone_number):
return render_template('form.html', error_msg=error_msg)
else:
error_msg = 'Failed to get the verification code!'
elif request.method == 'POST':
phone_number = request.form['phone']
code = request.form['code']
password = request.form['password']
if code == '':
error_msg = '请输入验证码'
elif User.query.filter(User.Tel_Number == phone_number).first():
error_msg = u'手机号已经注册过了'
elif sms.verify(phone_number, code):
user_forsql = User()
user_forsql.Password = password
user_forsql.Tel_Number = phone_number
db.session.add(user_forsql)
db.session.commit()
success = 1
# 自动登录
session['user_phone'] = phone_number
g.current_userphone = session['user_phone']
g.current_user = user_forsql
return render_template('form.html', success=success)
else:
error_msg = u'验证码不正确，请检查！'
return render_template('form.html', error_msg=error_msg)
return render_template('form.html', error_msg=error_msg)
