You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
685 B
23 lines
685 B
from django.contrib.auth import get_user_model
|
|
|
|
from .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 = get_user_model().objects.get(**kwargs)
|
|
if user.check_password(password):
|
|
return user
|
|
except User.DoesNotExist:
|
|
return None
|
|
|
|
def get_user(self, username):
|
|
try:
|
|
return get_user_model().objects.get(pk=username)
|
|
except get_user_model().DoesNotExist:
|
|
return None
|
|
|