from django.contrib.auth.backends import BaseBackend
from django.core.exceptions import ObjectDoesNotExist
from .models import Users


class PhoneNumberBackend(BaseBackend):
    def authenticate(self, request, phone_number=None, email=None, password=None):
        try:
            if email:
                user = Users.objects.get(email=email)
                if user.check_password(password) and user.user_type != "Customer":  # type: ignore
                    return user
            elif phone_number:
                user = Users.objects.get(phone_number=phone_number)
                return user
        except ObjectDoesNotExist:
            return None

    def get_user(self, user_id):
        try:
            return Users.objects.get(pk=user_id)
        except ObjectDoesNotExist:
            return None