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)