21 lines
583 B
Python
21 lines
583 B
Python
|
from django.contrib.auth.backends import ModelBackend
|
||
|
|
|
||
|
|
from .models import CustomUser
|
||
|
|
|
||
|
|
|
||
|
|
class CustomUserBackend(ModelBackend):
|
||
|
|
|
||
|
|
def authenticate(self, username=None, password=None):
|
||
|
|
try:
|
||
|
|
user = CustomUser.custom_objects.get_by_natural_key(username)
|
||
|
|
if user.check_password(password):
|
||
|
|
return user
|
||
|
|
except CustomUser.DoesNotExist:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def get_user(self, user_id):
|
||
|
|
try:
|
||
|
|
return CustomUser.custom_objects.get(pk=user_id)
|
||
|
|
except CustomUser.DoesNotExist:
|
||
|
|
return None
|