cloned from 'bitbucket', 2013/08/13

1 files changed, 25 insertions, 0 deletions

diff --git a/97suifangqa/apps/profile/backends.py b/97suifangqa/apps/profile/backends.py
new file mode 100644
index 0000000..8903805
--- /dev/null
+++ b/97suifangqa/apps/profile/backends.py
@@ -0,0 +1,25 @@
+#-*- coding: utf-8 -*-
+
+from django.contrib.auth.models import User
+
+class EmailOrUsernameModelBackend(object):
+
+	def authenticate(self, username=None, password=None):
+		if '@' in username:
+			kwargs = {'email': username}
+		else:
+			kwargs = {'username': username}
+		try:
+			user = User.objects.get(**kwargs)
+			if user.check_password(password):
+				return user
+		except User.DoesNotExist:
+			pass
+		return None
+
+	def get_user(self, user_id):
+		try:
+			return User.objects.get(pk=user_id)
+		except User.DoesNotExist:
+			pass
+		return None
\ No newline at end of file