added register next

tschellenbach · Apr 18, 2011 · 33a4c2e · 33a4c2e
1 parent e4bd1d7
commit 33a4c2e
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/django_facebook/models.py b/django_facebook/models.py
@@ -25,13 +25,14 @@ def __unicode__(self):
     class Meta:
         abstract = True
 
-    def post_facebook_registration(self):
+    def post_facebook_registration(self, request):
         '''
         Behaviour after registering with facebook
         '''
-        url = reverse('facebook_connect')
-        response = HttpResponseRedirect(url)
-        response.set_cookie('fresh_registration', self.user.id)
+        from django_facebook.utils import next_redirect
+        default_url = reverse('facebook_connect')
+        response = next_redirect(request, default=default_url, next_key='register_next')
+        response.set_cookie('fresh_registration', user.id)
         return response
 
 

diff --git a/django_facebook/tests/test.py b/django_facebook/tests/test.py
@@ -20,6 +20,7 @@ class UserConnectTest(FacebookTest):
     - Retry on facebook connection errors
     - Taken username errors
     - Next param op mystyle
+    - unique fb ids
     '''
     fixtures = ['users.json']
 

diff --git a/django_facebook/views.py b/django_facebook/views.py
@@ -56,7 +56,7 @@ def connect(request):
             if action is CONNECT_ACTIONS.CONNECT:
                 messages.info(request, _("You have connected your account to %s's facebook profile") % facebook_data['name'])
             elif action is CONNECT_ACTIONS.REGISTER:
-                response = user.get_profile().post_facebook_registration()
+                response = user.get_profile().post_facebook_registration(request)
                 return response
         else:
             return next_redirect(request, additional_params=dict(fb_error_or_cancel=1), next_key=['error_next', 'next'])