From 036d813de43b92b5afeda74415cfa56e99b87a8f Mon Sep 17 00:00:00 2001
From: Nick Sergeant <nick@nicksergeant.com>
Date: Tue, 1 Nov 2016 15:34:10 -0400
Subject: [PATCH] Auth stuff.

---
 urls.py        |  5 +++--
 utils/views.py | 15 +++++++++++++++
 2 files changed, 18 insertions(+), 2 deletions(-)
 create mode 100644 utils/views.py

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/'