Adding email lookup to forgot password flow.

This commit is contained in:
Samuel Clay 2013-01-05 10:24:26 -08:00
parent cbd99b7462
commit 6c124c2035

View file

@ -15,7 +15,7 @@ class Command(BaseCommand):
if username: if username:
try: try:
user = User.objects.get(username__icontains=username) user = User.objects.get(username__icontains=username)
except User.MultipleObjectsFound: except User.MultipleObjectsReturned:
user = User.objects.get(username__iexact=username) user = User.objects.get(username__iexact=username)
except User.DoesNotExist: except User.DoesNotExist:
user = User.objects.get(email__iexact=username) user = User.objects.get(email__iexact=username)
@ -24,9 +24,9 @@ class Command(BaseCommand):
elif email: elif email:
try: try:
user = User.objects.get(email__icontains=email) user = User.objects.get(email__icontains=email)
except User.MultipleObjectsFound: except User.MultipleObjectsReturned:
user = User.objects.get(email__iexact=email) user = User.objects.get(email__iexact=email)
except User.MultipleObjectsFound: except User.MultipleObjectsReturned:
users = User.objects.filter(email__iexact=email) users = User.objects.filter(email__iexact=email)
user = users[0] user = users[0]
except User.DoesNotExist: except User.DoesNotExist: