gestioCOF/kfet/auth/backends.py

# -*- coding: utf-8 -*-
from django.contrib.auth import get_user_model
from kfet.models import Account, GenericTeamToken

from .utils import get_kfet_generic_user

User = get_user_model()


class BaseKFetBackend:
    def get_user(self, user_id):
        """
        Add extra select related up to Account.
        """
        try:
            return (
                User.objects
                .select_related('profile__account_kfet')
                .get(pk=user_id)
            )
        except User.DoesNotExist:
            return None


class KFetBackend(BaseKFetBackend):
    def authenticate(self, request):
        password = request.POST.get('KFETPASSWORD', '')
        password = request.META.get('HTTP_KFETPASSWORD', password)
        if not password:
            return None

        try:
            return Account.objects.get_by_password(password).user
        except Account.DoesNotExist:
            return None


class GenericTeamBackend(BaseKFetBackend):
    def authenticate(self, username=None, token=None):
        valid_token = GenericTeamToken.objects.get(token=token)
        if username == 'kfet_genericteam' and valid_token:
            return get_kfet_generic_user()
        return None

    def get_user(self, user_id):
        try:
            return (
                User.objects
                .select_related('profile__account_kfet')
                .get(pk=user_id)
            )
        except User.DoesNotExist:
            return None