Commit 577a3a81 authored by Mathijs de Bruin's avatar Mathijs de Bruin
Browse files

Enable ‘next’ redirection after registration.

parent 7b474efd
......@@ -244,8 +244,12 @@ class ConfirmRegisterForm(Form, RegisterFormMixin,
pass
class RegisterForm(ConfirmRegisterForm, PasswordConfirmFormMixin):
pass
class RegisterForm(ConfirmRegisterForm, PasswordConfirmFormMixin,
NextFormMixin):
def __init__(self, *args, **kwargs):
super(RegisterForm, self).__init__(*args, **kwargs)
if not self.next.data:
self.next.data = request.args.get('next', '')
class ResetPasswordForm(Form, NewPasswordFormMixin, PasswordConfirmFormMixin):
......
......@@ -234,8 +234,8 @@ def get_post_login_redirect(declared=None):
return get_post_action_redirect('SECURITY_POST_LOGIN_VIEW', declared)
def get_post_register_redirect():
return get_post_action_redirect('SECURITY_POST_REGISTER_VIEW')
def get_post_register_redirect(declared=None):
return get_post_action_redirect('SECURITY_POST_REGISTER_VIEW', declared)
def find_redirect(key):
......
......@@ -122,7 +122,7 @@ def register():
login_user(user)
if not request.json:
return redirect(get_post_register_redirect())
return redirect(get_post_register_redirect(form.next.data))
return _render_json(form, include_auth_token=True)
if request.json:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment