def register():
error = None
if request.method == 'POST':
username = request.form.get('username',None)
email = request.form.get('email','')
password = request.form.get('password','')
if not username or not email or not password:
error = u'请填写下面所有信息'
if not ut.is_email(email):
error = u'您输入的邮箱不合法'
elif f.get_user(username):
error = u'您的用户名“'+username+u'”已经被注册，请更换用户名'
elif f.get_user(email):
error = u'您的邮箱“'+email+u'”已经被注册，请更换邮箱'
else:
user = User(user_id = ut.create_user_id(),
username=username, email=email,
pw_hash=generate_password_hash(password),
created=int(time.time()*1000), updated=int(time.time()*1000))
m.session.add(user)
m.session.commit()
session['user_id'] = user.user_id
if error:
return render_template('register.html',error = error)
else:
return redirect(request.referrer)
if g.user:
return redirect(url_for('main.index'))
return render_template('register.html',error = error)
def login():
"""login"""
error = None
if request.method == 'POST':
username = request.form.get('username',None)
remember = request.form.get('remember', None) == 'on'
user = f.get_user(username)
if user is None:
error = u'邮箱或者手机号不正确'
elif not check_password_hash(user.pw_hash,
request.form.get('password',None)):
error = u'密码不正确'
else:
session['user_id'] = user.user_id
if remember:
session.permanent = True
if error:
return render_template('login.html',error = error)
else:
return redirect(request.referrer)
if g.user:
return redirect(url_for('main.index'))
return render_template('login.html',error = error)
