diff --git a/urls.py b/urls.py index c4f5856..f9e5e4f 100644 --- a/urls.py +++ b/urls.py @@ -7,13 +7,13 @@ from django.http import HttpResponseRedirect from django.views.generic import RedirectView, TemplateView from django.views.static import serve from forms import AuthenticationFormWithInactiveUsersOkay -from registration.backends.default.views import RegistrationView from snipts.api import (PublicSniptResource, PublicUserResource, PrivateSniptResource, PrivateFavoriteResource, PrivateUserProfileResource, PrivateUserResource, PublicTagResource) from snipts.views import search from tastypie.api import Api +from utils.views import SniptRegistrationView from views import (homepage, lexers, login_redirect, tags, user_api_key) @@ -54,7 +54,8 @@ urlpatterns = [ url(r'^search/$', search), url(r'^register/$', lambda x: HttpResponseRedirect('/signup/')), - url(r'^signup/$', RegistrationView.as_view()), + url(r'^signup/$', SniptRegistrationView.as_view(), + name='registration_register'), url(r'^activate/complete/$', RedirectView.as_view( url='/login-redirect/')), url(r'^login/?$', login, { diff --git a/utils/views.py b/utils/views.py new file mode 100644 index 0000000..4cf5335 --- /dev/null +++ b/utils/views.py @@ -0,0 +1,15 @@ +from registration.backends.default.views import RegistrationView +from utils.forms import SniptRegistrationForm + + +class SniptRegistrationView(RegistrationView): + """ + Custom registration view that uses our custom form. + """ + form_class = SniptRegistrationForm + + def dispatch(self, request, *args, **kwargs): + return super(RegistrationView, self).dispatch(request, *args, **kwargs) + + def get_success_url(self, request): + return '/account/activate/'