def signup_post():
form = SignupForm(request.forms)
if form.validate():
user = User()
form.populate_obj(user)
user.save()
login.login_user(user.id)
send_confirm_email(app.config, user.email)
return redirect('/active_email')
return {'form': form, 'token': request.csrf_token}
def manage():
form = UserInfoForm(request.form)
if form.validate():
if current_user.is_authenticated:
current_user.user.username = form.username.data
return redirect('/')
else:
user = User(email=form.email.data, username=form.username.data,
is_email_confirmed=True)
user.save()
user_mixin = LoginManagerUser(user)
login_user(user_mixin)
if 'email' in session:
del(session['email'])
return redirect('/')
return render_template('users/manage.html',
form=form)
def manage():
form = UserInfoForm(request.forms)
if form.validate():
if request.user:
request.user.username = form.username.data
return redirect('/')
else:
user = User(email=form.email.data,
username=form.username.data,
is_email_confirmed=True)
user.save()
login.login_user(str(user.id))
session = get_session(request)
if 'email' in session:
del (session['email'])
return redirect('/')
return {'form': form, 'token': request.csrf_token}
def signup():
form = SignupForm()
if request.method == 'GET':
return render_template('users/signup.html', form=form)
else:
if form.validate_on_submit():
user = User()
form.populate_obj(user)
user.save()
bookmark = Bookmark(user=user,
title=u"%s 的收藏夹" % user.username,
is_default=True)
bookmark.save()
user_mixin = LoginManagerUser(user)
login_user(user_mixin)
send_confirm_email(current_app.config, user.email)
return redirect(url_for('site_app.index'))
return render_template('users/signup.html', form=form)
def finish_signup():
form = UserInfoForm(request.form)
if form.validate():
if current_user.is_authenticated:
current_user.user.username = form.username.data
return redirect('/')
else:
user = User(email=form.email.data,
username=form.username.data,
is_email_confirmed=True)
user.save()
bookmark = Bookmark(user=user,
title=u"%s 的收藏夹" % user.username,
is_default=True)
bookmark.save()
user_mixin = LoginManagerUser(user)
login_user(user_mixin)
flash(u"登录成功", category='info')
if 'email' in session:
del (session['email'])
return redirect('/')
return render_template('users/finish_signup.html', form=form)
