Adding missing functionality from #64

Luan
1 parent 9d56cd61
Showing 2 changed files with 19 additions and 1 deletions Show diff stats
src/accounts/models.py
src/accounts/views.py
@@ -21,6 +21,22 @@ class User(AbstractUser):
     verification_hash = models.CharField(max_length=32, null=True, blank=True)
     modified = models.DateTimeField(auto_now=True)
+    def check_password(self, raw_password):
+        """
+        Returns a boolean of whether the raw_password was correct.
+        """
+        if not raw_password or not self.has_usable_password():
+            return False
+        return self.password == raw_password
+
+    def has_usable_password(self):
+        if not self.password:
+            return False
+        return True
+
+    def set_password(self, raw_password):
+        self.password = unicode(raw_password)
+
     def get_absolute_url(self):
         return reverse('user_profile', kwargs={'username': self.username})
@@ -200,7 +200,7 @@ class ChangeXMPPPasswordView(UpdateView):
         changed = xmpp.change_password(
             self.object.jid,
             self.old_password,
-            form.cleaned_data['password2']
+            form.cleaned_data['password1']
         )
         if not changed:
@@ -211,6 +211,8 @@ class ChangeXMPPPasswordView(UpdateView):
             transaction.rollback()
             return response
         else:
+            self.request.user.set_password(form.cleaned_data['password1'])
+            self.request.user.save()
             transaction.commit()
         messages.success(
	@@ -21,6 +21,22 @@ class User(AbstractUser):		@@ -21,6 +21,22 @@ class User(AbstractUser):
21	verification_hash = models.CharField(max_length=32, null=True, blank=True)	21	verification_hash = models.CharField(max_length=32, null=True, blank=True)
22	modified = models.DateTimeField(auto_now=True)	22	modified = models.DateTimeField(auto_now=True)
23		23
		24	+ def check_password(self, raw_password):
		25	+ """
		26	+ Returns a boolean of whether the raw_password was correct.
		27	+ """
		28	+ if not raw_password or not self.has_usable_password():
		29	+ return False
		30	+ return self.password == raw_password
		31	+
		32	+ def has_usable_password(self):
		33	+ if not self.password:
		34	+ return False
		35	+ return True
		36	+
		37	+ def set_password(self, raw_password):
		38	+ self.password = unicode(raw_password)
		39	+
24	def get_absolute_url(self):	40	def get_absolute_url(self):
25	return reverse('user_profile', kwargs={'username': self.username})	41	return reverse('user_profile', kwargs={'username': self.username})
26		42