Ticket #18170: confirm_password_reset-1.3.1.diff

new file django/contrib/auth/utils.py

@@ +1 @@
+from django.contrib.auth.models import User
+from django.contrib.auth.tokens import default_token_generator
+from django.utils.http import base36_to_int
+
+def confirm_password_reset(uidb36, token, token_generator=default_token_generator):
+    try:
+        uid_int = base36_to_int(uidb36)
+        user = User.objects.get(id=uid_int)
+    except (ValueError, User.DoesNotExist):
+        user = None
+
+    return user, token_generator.check_token(user, token) if user else False
+

django/contrib/auth/views.py

@@ -5 +5 @@
 from django.http import HttpResponseRedirect, QueryDict
 from django.shortcuts import render_to_response
 from django.template import RequestContext
-from django.utils.http import base36_to_int
 from django.utils.translation import ugettext as _
 from django.views.decorators.cache import never_cache
 from django.views.decorators.csrf import csrf_protect
@@ -14 +13 @@
 from django.contrib.auth import REDIRECT_FIELD_NAME, login as auth_login, logout as auth_logout
 from django.contrib.auth.decorators import login_required
 from django.contrib.auth.forms import AuthenticationForm, PasswordResetForm, SetPasswordForm, PasswordChangeForm
-from django.contrib.auth.models import User
 from django.contrib.auth.tokens import default_token_generator
+from django.contrib.auth.utils import confirm_password_reset
 from django.contrib.sites.models import get_current_site
 
 
@@ -186 +185 @@
     assert uidb36 is not None and token is not None # checked by URLconf
     if post_reset_redirect is None:
         post_reset_redirect = reverse('django.contrib.auth.views.password_reset_complete')
-    try:
-        uid_int = base36_to_int(uidb36)
-        user = User.objects.get(id=uid_int)
-    except (ValueError, User.DoesNotExist):
-        user = None
-
-    if user is not None and token_generator.check_token(user, token):
-        validlink = True
+    user, validlink = confirm_password_reset(uidb36, token, token_generator=token_generator)
+
+    if user is not None and validlink:
         if request.method == 'POST':
             form = set_password_form(user, request.POST)
             if form.is_valid():
@@ -202 +196 @@
         else:
             form = set_password_form(None)
     else:
-        validlink = False
         form = None
     context = {
         'form': form,