gestioCOF/kfet/auth/backends.py

# -*- coding: utf-8 -*-

import hashlib

from django.contrib.auth.models import User, Permission
from gestioncof.models import CofProfile
from kfet.models import Account, GenericTeamToken


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

        try:
            password_sha256 = (
                hashlib.sha256(password.encode('utf-8'))
                .hexdigest()
            )
            account = Account.objects.get(password=password_sha256)
            return account.cofprofile.user
        except Account.DoesNotExist:
            return None


class GenericTeamBackend(object):
    def authenticate(self, username=None, token=None):
        valid_token = GenericTeamToken.objects.get(token=token)
        if username == 'kfet_genericteam' and valid_token:
            # Création du user s'il n'existe pas déjà
            user, _ = User.objects.get_or_create(username='kfet_genericteam')
            profile, _ = CofProfile.objects.get_or_create(user=user)
            account, _ = Account.objects.get_or_create(
                    cofprofile=profile,
                    trigramme='GNR')

            # Ajoute la permission kfet.is_team à ce user
            perm_is_team = Permission.objects.get(codename='is_team')
            user.user_permissions.add(perm_is_team)

            return 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