Merge branch 'hotfix/another_fix_for_comparison' into dev

[auf_rh_dae.git] / project / groups.py

# -*- encoding: utf-8 -*-

from django.core.exceptions import ObjectDoesNotExist

CORRESPONDANT_RH = 'Correspondants RH'
ADMINISTRATEURS = 'Administrateurs'
DIRECTEUR_DE_BUREAU = 'Directeurs de bureau'
DRH_NIVEAU_1 = 'DRH'
DRH_NIVEAU_2 = 'DRH-2'
ACCIOR = 'ACCIOR'
ABF = 'ABF'
FINANCE = 'Finance'
HAUTE_DIRECTION = 'Haute direction'
SERVICE_UTILISATEURS = 'Service utilisateurs'


def get_zones_from_user(user):
    from auf.django.references import models as ref
    try:
        profile = user.profile
    except ObjectDoesNotExist:
        return ref.ZoneAdministrative.objects.none()
    else:
        return profile.zones_administratives.all()


def get_employe_from_user(user):
    """
    Retourne un employé AUF à partir de son user Django.
    """
    import auf.django.references.models as ref
    try:
        employe = ref.Employe.objects.get(courriel=user.email)
    except ObjectDoesNotExist:
        raise Exception(u"L'employé avec le courriel %s n'a pas \
                été trouvé dans le référentiel." % user.email)
    return employe


def is_user_dans_services_centraux(user):
    # Set de codes de zones.
    zones = set()

    # Set de comparaison (services centraux Montreal et Paris)
    central_zones = set(['SCM', 'SCP'])

    # Ajout de zones du profil, si existantes.
    try:
        profile = user.profile
    except ObjectDoesNotExist:
        pass
    else:
        zones.update([
                x[0] for x in profile.zones_administratives
                .all().values_list('code')
                ])

    return len(central_zones.intersection(zones)) > 0


def is_user_dans_region(user):
    return not is_user_dans_services_centraux(user)