Change User to get_user_model - fix problems

This commit is contained in:
Степаненко Ольга
2021-05-06 12:43:44 +03:00
parent 9c9b4dd08b
commit f24c94914f
3 changed files with 9 additions and 11 deletions

View File

@@ -1,19 +1,19 @@
from django.contrib.auth.backends import ModelBackend
from django.contrib.auth.models import User
from django.contrib.auth import get_user_model
class EmailAuthBackend(ModelBackend):
def authenticate(self, request, username=None, password=None, **kwargs):
try:
user = User.objects.get(email=username)
user = get_user_model().objects.get(email=username)
if user.check_password(password):
return user
return None
except User.DoesNotExist:
except get_user_model().DoesNotExist:
return None
def get_user(self, user_id):
try:
return User.objects.get(pk=user_id)
except User.DoesNotExist:
return get_user_model().objects.get(pk=user_id)
except get_user_model().DoesNotExist:
return None