From 258675f9eb77d3ab154fe3503f281c5c0a1a4b7d Mon Sep 17 00:00:00 2001
From: j <0x006A@0x2620.org>
Date: Sat, 25 Dec 2010 15:23:10 +0530
Subject: [PATCH] request token takes username or email now

---
 pandora/user/views.py | 88 ++++++++++++++++++++++---------------------
 1 file changed, 45 insertions(+), 43 deletions(-)

diff --git a/pandora/user/views.py b/pandora/user/views.py
index e5a269e5..769bd076 100644
--- a/pandora/user/views.py
+++ b/pandora/user/views.py
@@ -226,66 +226,68 @@ def resetPassword(request):
     return render_to_json_response(response)
 actions.register(resetPassword)
 
-class RecoverForm(forms.Form):
-    username_or_email = forms.TextInput()
-
 def requestToken(request):
     '''
         param data {
-            username_or_email: username
+            username: username,
+            email: email
         }
         
         return {
             status: {'code': int, 'text': string}
             data: {
                 errors: {
-                    username_or_email: 'Username or email address not found'
+                    username: 'Unknown Username'
+                    email: 'Unknown Email'
                 }
                 username: user
             }
         }
     '''
     data = json.loads(request.POST['data'])
-    form = RegisterForm(data, request.FILES)
-    if form.is_valid():
-        username = data['username_or_email']
-        user = None
-        q = models.User.objects.filter(username=username)
-        if q.count() > 0:
-            user = q[0]
-        else:
-            q = models.User.objects.filter(email=username)
-            if q.count() > 0:
-                user = q[0]
-        if user:
-            while True:
-                token = ox.to32(random.randint(32768, 1048575))
-                if models.UserProfile.objects.filter(reset_token=token).count() == 0:
-                    break
-            user_profile = user.get_profile()
-            user_profile.reset_token = token
-            user_profile.save()
+    user = None
+    if 'username' in data:
+        try:
+            user = models.User.objects.get(username=data['username'])
+        except models.User.DoesNotExist:
+            user = None
+    elif 'email' in data:
+        try:
+            user = models.User.objects.get(email=data['email'])
+        except models.User.DoesNotExist:
+            user = None
+    if user:
+        while True:
+            token = ox.to32(random.randint(32768, 1048575))
+            if models.UserProfile.objects.filter(reset_token=token).count() == 0:
+                break
+        user_profile = user.get_profile()
+        user_profile.reset_token = token
+        user_profile.save()
 
-            template = loader.get_template('password_reset_email.txt')
-            context = RequestContext(request, {
-                'url': request.build_absolute_uri("/"),
-                'token': token,
-                'sitename': settings.SITENAME,
-            })
-            message = template.render(context)
-            subject = '%s password reset' % settings.SITENAME
-            user.email_user(subject, message)
-            response = json_response({
-                'username': user.username
-            }, text='recover email sent')
-        else:
-            response = json_response({
-                'errors': {
-                    'username_or_email': 'Username or email address not found'
-                }
-            })
+        template = loader.get_template('password_reset_email.txt')
+        context = RequestContext(request, {
+            'url': request.build_absolute_uri("/"),
+            'token': token,
+            'sitename': settings.SITENAME,
+        })
+        message = template.render(context)
+        subject = '%s password reset' % settings.SITENAME
+        user.email_user(subject, message)
+        response = json_response({
+            'username': user.username
+        }, text='password reset email sent')
     else:
-        response = json_response(status=400, text='invalid data')
+        response = json_response({
+            'errors': {
+            }
+        })
+        if 'username' in data:
+            response['data']['errors']['username'] = 'Unknown Username'
+        elif 'email' in data:
+            response['data']['errors']['email'] = 'Unknown Email'
+        else:
+            response = json_response(status=400, text='invalid data')
     return render_to_json_response(response)
 actions.register(requestToken)