Added proper email templates

[auf_rh_dae.git] / project / rh / admin.py

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

import datetime
import reversion
import itertools
from ajax_select import make_ajax_form
from auf.django.references import models as ref
from django.core.mail import send_mail, EmailMessage
from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.models import User
from django import forms
from django.core.urlresolvers import reverse
from django.contrib import admin
from django.contrib.contenttypes.models import ContentType
from django.conf import settings
from django.db.models import Q, Count
from django.template.defaultfilters import date
from django.template import Context, loader
from django.utils.formats import date_format

from project import groups
from project.decorators import in_drh_or_admin
from project.rh import models as rh
from project.permissions import user_gere_obj_de_sa_region, \
        user_can_list_obj, \
        user_can_add_obj, \
        user_can_change_obj, \
        user_can_delete_obj

from project.rh.forms import ContratForm, AyantDroitForm, EmployeAdminForm, \
        AjaxSelect, DossierForm, ResponsableInlineForm, \
        ClassementHistoriqueForm

from project.rh.change_list import ChangeList


def listing_par_defaut(model, request):
    """
    Teste si la requete provient de la même page.
    """
    if not 'HTTP_REFERER' in request.META.keys():
        return False
    referer = request.META['HTTP_REFERER']
    referer = "/".join(referer.split('/')[3:])
    referer = "/%s" % referer.split('?')[0]
    change_list_view = 'admin:%s_%s_changelist' % (
            model._meta.app_label,
            model.__name__.lower(),)
    return referer != reverse(change_list_view)


class BaseAdmin(admin.ModelAdmin):

    class Media:
        css = {'screen': (
            'css/admin_custom.css',
            'jquery-autocomplete/jquery.autocomplete.css',
        )}
        js = (
            'https://ajax.googleapis.com/ajax/libs/jquery/1.7.2/jquery.min.js',
            'jquery-autocomplete/jquery.autocomplete.min.js',
        )


# Admin pour reversion

class ArchivableAdmin(admin.ModelAdmin):
    """
    Admin pour les modèles archivables
    """
    list_filter = ('archive', )

    def queryset(self, request):
        return self.model.avec_archives.all()

    def _archive(self, obj):
        if obj.archive:
            return "oui"
        else:
            return "non"
    _archive.short_description = u'Archivé'
    _archive.admin_order_field = 'archive'


class RegionProxy(ref.Region):
    """ Proxy utilisé pour les organigrammes par région """
    class Meta:
        managed = False
        proxy = True
        verbose_name = u"Organigramme par région"
        verbose_name_plural = u"Organigramme par région"


class ImplantationProxy(ref.Implantation):
    """ Proxy utilisé pour les organigrammes par implantation """
    class Meta:
        managed = False
        proxy = True
        verbose_name = u"Organigramme par implantations"
        verbose_name_plural = u"Organigramme par implantations"


class ServiceProxy(rh.Service):
    """ Proxy utilisé pour les organigrammes par service """

    class Meta:
        managed = False
        proxy = True
        verbose_name = u"Organigramme par services"
        verbose_name_plural = u"Organigramme par services"


class EmployeProxy(rh.Employe):
    """ Proxy utilisé pour les organigrammes des employés """
    class Meta:
        managed = False
        proxy = True
        verbose_name = u"Organigramme des employés"
        verbose_name_plural = u"Organigramme des employés"


class DateRangeMixin(object):
    prefixe_recherche_temporelle = ""

    def get_changelist(self, request, **kwargs):
        """
        On filtre par défaut sur les items 'actifs'.
        Le changelist plug le filtrage temporel.
        """
        if listing_par_defaut(self.model, request):
            params = request.GET.copy()
            params.update({'statut': 'Actif'})
            request.GET = params
        return ChangeList


# Override of the InlineModelAdmin to support the link in the tabular inline
class LinkedInline(admin.options.InlineModelAdmin):
    template = "admin/linked.html"
    admin_model_path = None

    def __init__(self, *args):
        super(LinkedInline, self).__init__(*args)
        if self.admin_model_path is None:
            self.admin_model_path = self.model.__name__.lower()


class ProtectRegionMixin(object):

    def changelist_view(self, request, extra_context=None):
        """
        On filtre par défaut sur la ZA du user connecté
        """
        if listing_par_defaut(self.model, request):
            if user_gere_obj_de_sa_region(request.user):
                params = request.GET.copy()
                zones = groups.get_zones_from_user(request.user)
                prefix_za = "%s__in" % self.model.prefix_implantation
                params.update({prefix_za: zones})
                request.GET = params
        return super(ProtectRegionMixin, self) \
                .changelist_view(request, extra_context)

    def queryset(self, request):
        qs = super(ProtectRegionMixin, self).queryset(request)

        if in_drh_or_admin(request.user):
            return qs

        if user_gere_obj_de_sa_region(request.user):
            zones = groups.get_zones_from_user(request.user)
            qkey = '%s__in' % self.model.prefix_implantation
            q = Q(**{qkey: zones})
            qs = qs.filter(q).distinct()
            return qs
        return qs.none()

    def has_add_permission(self, request):
        return user_can_add_obj(request.user)

    def has_change_permission(self, request, obj=None):
        if obj is None:
            return user_can_list_obj(request.user)
        else:
            return user_can_change_obj(request.user, obj)

    def has_delete_permission(self, request, obj=None):
        return user_can_delete_obj(request.user, obj) if obj else True


class DerniereModificationAdmin(admin.ModelAdmin):

    def queryset(self, request):
        qs = super(DerniereModificationAdmin, self).queryset(request)
        ct = ContentType.objects.get_for_model(self.model)
        db_table = self.model._meta.db_table
        pk = self.model._meta.pk.column
        return qs.extra(select={
            'date_modification':
            "SELECT action_time FROM django_admin_log "
            "WHERE content_type_id = %d AND object_id = %s.%s "
            "ORDER BY action_time DESC "
            "LIMIT 1" % (ct.id, db_table, pk),
            'user_modification':
            "SELECT u.username "
            "FROM auth_user u "
            "INNER JOIN django_admin_log l ON l.user_id = u.id "
            "WHERE l.content_type_id = %d AND object_id = %s.%s "
            "ORDER BY action_time DESC "
            "LIMIT 1" % (ct.id, db_table, pk),
        })

    def derniere_modification(self, obj):
        text = ''
        if obj.date_modification:
            text += obj.date_modification.strftime('%d-%m-%Y %H:%M')
        if obj.user_modification:
            text += ' par ' + obj.user_modification
        return text
    derniere_modification.short_description = u'dernière modification'
    derniere_modification.admin_order_field = 'date_modification'


# Inlines

class CommentaireInlineForm(forms.ModelForm):

    def save(self, commit=True):

        # Hack: reversion.VersionAdmin ne sauvegarde pas les champs qui ne
        # sont pas explicitement dans le formulaire. Il plante cependant
        # leur valeur dans `self.initial`. Ceci est un peu fragile. Si
        # c'est possible, il serait plus approprié que Reversion se rende
        # compte qu'il manque des champs.
        instance = super(CommentaireInlineForm, self).save(commit=False)
        if instance.owner_id is None and 'owner' in self.initial:
            instance.owner_id = self.initial['owner']
        if instance.date_creation is None and 'date_creation' in self.initial:
            instance.date_creation = self.initial['date_creation']
        if commit:
            instance.save()
            self.save_m2m()
        return instance


class ReadOnlyInlineMixin(object):

    def get_readonly_fields(self, request, obj=None):
        return [f.name for f in self.model._meta.fields]


class AyantDroitInline(admin.StackedInline):
    model = rh.AyantDroit
    form = AyantDroitForm
    extra = 0

    fieldsets = (
        (None, {
            'fields': (
                ('nom', 'prenom'),
                ('nom_affichage', 'genre'),
                'nationalite',
                'date_naissance',
                'lien_parente',
                )}),
    )


class AyantDroitCommentaireInline(admin.TabularInline):
    readonly_fields = ('owner',)
    model = rh.AyantDroitCommentaire
    extra = 1
    form = CommentaireInlineForm


class ContratInline(admin.TabularInline):
    form = ContratForm
    model = rh.Contrat
    extra = 1


class DossierROInline(ReadOnlyInlineMixin, LinkedInline):
    template = "admin/rh/dossier/linked.html"
    model = rh.Dossier
    extra = 0
    can_delete = False
    fields = ('poste', 'date_debut', 'date_fin', )

    def has_add_permission(self, request=None):
        return False

    def has_change_permission(self, request, obj=None):
        return False

    def has_delete_permission(self, request, obj=None):
        return False


class DossierClassementRecordInline(admin.TabularInline):
    def queryset(self, request):
        return (super(DossierClassementRecordInline, self)
                .queryset(request).order_by('-date_debut', '-id'))
    model = rh.RHDossierClassementRecord
    form = ClassementHistoriqueForm
    extra = 0


class DossierCommentaireInline(admin.TabularInline):
    readonly_fields = ('owner',)
    model = rh.DossierCommentaire
    extra = 1
    form = CommentaireInlineForm


class DossierPieceInline(admin.TabularInline):
    model = rh.DossierPiece
    extra = 4


class EmployeInline(admin.TabularInline):
    model = rh.Employe


class EmployeCommentaireInline(admin.TabularInline):
    readonly_fields = ('owner',)
    model = rh.EmployeCommentaire
    extra = 1
    form = CommentaireInlineForm


class EmployePieceInline(admin.TabularInline):
    model = rh.EmployePiece
    extra = 4


class PosteCommentaireInline(admin.TabularInline):
    readonly_fields = ('owner',)
    model = rh.PosteCommentaire
    extra = 1
    form = CommentaireInlineForm


class PosteFinancementInline(admin.TabularInline):
    model = rh.PosteFinancement


class PostePieceInline(admin.TabularInline):
    model = rh.PostePiece


class RemunerationInline(admin.TabularInline):
    model = rh.Remuneration
    extra = 1


class RemunerationROInline(ReadOnlyInlineMixin, RemunerationInline):
    pass


class TypePosteInline(admin.TabularInline):
    model = rh.TypePoste


class PosteComparaisonInline(admin.TabularInline):
    model = rh.PosteComparaison


class ClassementAdmin(reversion.VersionAdmin,
                      ArchivableAdmin,
                      DerniereModificationAdmin,
                      BaseAdmin):
    ignore_duplicate_revisions = True
    list_display = ('_classement', 'derniere_modification', '_archive')
    list_filter = ('archive', )
    fieldsets = (
        (None, {'fields': (
                    'type', 'echelon',
                    'degre', 'coefficient', 'archive')}),
    )

    def _classement(self, obj):
        return unicode(obj)
    _classement.short_description = u"Classement"


class DeviseAdmin(reversion.VersionAdmin, ArchivableAdmin,
                  DerniereModificationAdmin, BaseAdmin):
    ignore_duplicate_revisions = True
    list_display = (
        'code', 'nom', 'derniere_modification', '_archive',
    )
    list_filter = ('archive', )
    fieldsets = (
        (None, {'fields': ('code', 'nom', 'archive', )}),
    )


class DossierAdmin(DateRangeMixin, ProtectRegionMixin, reversion.VersionAdmin,
                   AjaxSelect, DerniereModificationAdmin, BaseAdmin):
    change_list_template = "admin/rh/dossier/change_list.html"
    ignore_duplicate_revisions = True
    alphabet_filter = 'employe__nom'
    search_fields = (
        'id',
        'employe__id',
        'poste__id',
        'employe__nom',
        'employe__prenom',
        'poste__nom',
        'poste__nom_feminin',
        'poste__implantation__nom',
    )
    list_display = (
        '_id',
        '_apercu',
        '_nom',
        '_employe',
        '_poste',
        '_zone_administrative',
        '_implantation',
        '_date_debut',
        '_date_fin',
        'derniere_modification',
        '_dae',
    )
    list_display_links = ('_nom',)
    list_filter = (
        'poste__implantation__zone_administrative',
        'poste__implantation',
        'poste__type_poste__categorie_emploi',
        'poste__type_poste',
        'rh_contrats__type_contrat',
        'principal',
    )
    inlines = (DossierPieceInline,
               DossierClassementRecordInline,
               ContratInline,
               RemunerationInline,
               DossierCommentaireInline,
              )
    fieldsets = (
        (None, {
            'fields': (
                'employe',
                'poste',
                'principal',
                'est_cadre',
                'statut',
                'organisme_bstg',)}),
        ('Recrutement', {
            'fields': (
                'statut_residence',
                'remplacement',
                'remplacement_de', )}),
        ('Rémunération', {
            'fields': (
                'classement',
                ('regime_travail', 'regime_travail_nb_heure_semaine'),)}),
        ('Occupation du Poste par cet Employe', {
            'fields': (('date_debut', 'date_fin'), )}
            ),
    )
    form = make_ajax_form(rh.Dossier, {
        'employe': 'employes',
        'poste': 'postes',
        'remplacement_de': 'dossiers',
    }, superclass=DossierForm)

    def lookup_allowed(self, key, value):
        if key in (
            'employe__nom__istartswith',
            'poste__implantation__zone_administrative__code__exact',
            'poste__implantation__zone_administrative__in',
            'poste__implantation__id__exact',
            'poste__type_poste__id__exact',
            'poste__type_poste__categorie_emploi__id__exact',
            'rh_contrats__type_contrat__id__exact',
            'principal__exact',
            'principal__isnull',
            ):
            return True

    def _id(self, obj):
        return obj.id
    _id.short_description = u"#"
    _id.admin_order_field = "id"

    def _apercu(self, d):
        apercu_link = u"""<a title="Aperçu du dossier"
                             onclick="return showAddAnotherPopup(this);"
                             href='%s'>
                             <img src="%simg/dossier-apercu.png" />
                             </a>""" % \
                (reverse('dossier_apercu', args=(d.id,)),
                 settings.STATIC_URL,
                 )
        return apercu_link
    _apercu.allow_tags = True
    _apercu.short_description = u""

    def _nom(self, obj):
        return "Dossier"
    _nom.allow_tags = True
    _nom.short_description = u"Dossier"

    def _employe(self, obj):
        employe = obj.employe
        view_link = reverse('employe_apercu', args=(employe.id,))
        edit_link = reverse('admin:rh_employe_change', args=(employe.id,))
        style = ""
        view = u"""<a href="%s"
                      title="Aperçu l'employé"
                      onclick="return showAddAnotherPopup(this);">
                      <img src="%simg/employe-apercu.png" />
                    </a>""" % (view_link, settings.STATIC_URL,)
        return u"""%s<a href='%s' style="%s;">%s</a>""" % \
        (view, edit_link, style, employe)
    _employe.allow_tags = True
    _employe.short_description = u"Employé"
    _employe.admin_order_field = "employe__nom"

    def _poste(self, dossier):
        link = u"""<a title="Aperçu du poste"
                      onclick="return showAddAnotherPopup(this);"
                      href='%s'><img src="%simg/poste-apercu.png" />
                    </a>
                    <a href="%s" title="Modifier le poste">%s [%d]</a>""" % \
                (reverse('poste_apercu', args=(dossier.poste.id,)),
                 settings.STATIC_URL,
                 reverse('admin:rh_poste_change', args=(dossier.poste.id,)),
                 dossier.poste.nom,
                 dossier.poste.id,
                 )
        return link
    _poste.allow_tags = True
    _poste.short_description = u'Poste'
    _poste.admin_order_field = 'poste__nom'

    def _zone_administrative(self, obj):
        return obj.poste.implantation.zone_administrative.code
    _zone_administrative.short_description = u"Zone administrative"
    _zone_administrative.admin_order_field = \
            'poste__implantation__zone_administrative__code'

    def _implantation(self, obj):
        return obj.poste.implantation.nom
    _implantation.short_description = u"Implantation"
    _implantation.admin_order_field = 'poste__implantation__nom'

    def _date_debut(self, obj):
        return date(obj.date_debut)

    _date_debut.short_description = u'Début'
    _date_debut.admin_order_field = 'date_debut'

    def _date_fin(self, obj):
        return date(obj.date_fin)
    _date_fin.short_description = u'Fin'
    _date_fin.admin_order_field = 'date_fin'

    def _date_modification(self, obj):
        return date(obj.date_modification) \
                if obj.date_modification is not None else "(aucune)"
    _date_modification.short_description = u'date modification'
    _date_modification.admin_order_field = 'date_modification'

    def _dae(self, d):
        apercu_link = ""
        dossiers_dae = d.dossiers_dae.all()
        if len(dossiers_dae) > 0:
            dossier_dae = dossiers_dae[0]
            apercu_link = u"""<a title="Aperçu du dossier"
                                 onclick="return showAddAnotherPopup(this);"
                                 href='%s'>
                                 <img src="%simg/loupe.png" />
                                 </a>""" % \
                    (reverse('embauche_consulter', args=(dossier_dae.id,)),
                     settings.STATIC_URL,
                     )
        return apercu_link
    _dae.allow_tags = True
    _dae.short_description = u"DAE"

    def save_formset(self, request, form, formset, change):
        instances = formset.save(commit=False)
        for instance in instances:
            if instance.__class__ == rh.DossierCommentaire:
                instance.owner = request.user
                instance.date_creation = datetime.datetime.now()
            instance.save()


class EmployeAdminBase(DateRangeMixin, ProtectRegionMixin,
                       DerniereModificationAdmin, BaseAdmin):
    prefixe_recherche_temporelle = "rh_dossiers__"
    alphabet_filter = 'nom'
    DEFAULT_ALPHABET = u'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
    search_fields = (
        'id', 'nom', 'prenom', 'nom_affichage',
        'rh_dossiers__poste__nom',
        'rh_dossiers__poste__nom_feminin'
    )
    ordering = ('nom', )
    form = EmployeAdminForm
    list_display = (
        '_id', '_apercu', '_nom', '_dossiers_postes',
        #'_zone_administrative',
        #'_implantation',
        'date_entree',
        'derniere_modification'
    )
    list_display_links = ('_nom',)
    list_filter = (
        'rh_dossiers__poste__implantation__zone_administrative',
        'rh_dossiers__poste__implantation', 'nb_postes'
    )
    inlines = (
        AyantDroitInline, DossierROInline, EmployePieceInline,
        EmployeCommentaireInline
    )
    fieldsets = (
        ('Identification', {
            'fields': (
                ('nom', 'prenom'),
                ('nom_affichage', 'genre'),
                'nationalite',
                'date_naissance',
            )
        }),
        ('Informations personnelles', {
            'fields': ('situation_famille', 'date_entree', )
        }),
        ('Coordonnées personnelles', {
            'fields': (
                ('tel_domicile', 'tel_cellulaire'),
                ('adresse', 'ville'),
                ('code_postal', 'province'),
                'pays',
                'courriel_perso'
            )
        }),
    )

    def _id(self, obj):
        return obj.id
    _id.short_description = u"#"
    _id.admin_order_field = "id"

    def _apercu(self, obj):
        return u"""<a title="Aperçu de l'employé"
                      onclick="return showAddAnotherPopup(this);"
                      href='%s'>
                      <img src="%simg/employe-apercu.png" />
                    </a>""" % \
            (reverse('employe_apercu', args=(obj.id,)), settings.STATIC_URL)
    _apercu.allow_tags = True
    _apercu.short_description = u""

    def _nom(self, obj):
        edit_link = reverse('admin:rh_employe_change', args=(obj.id,))
        return u"""<a href='%s'><strong>%s</strong></a>""" % \
        (edit_link, "%s %s" % (obj.nom.upper(), obj.prenom))
    _nom.allow_tags = True
    _nom.short_description = u"Employé"
    _nom.admin_order_field = "nom"

    def _zone_administrative(self, obj):
        try:
            d = rh.Dossier.objects.filter(employe=obj.id, principal=True)[0]
            zone = d.poste.implantation.zone_administrative.code
        except:
            zone = None
        return zone
    _zone_administrative.short_description = u"Zone administrative"

    def _implantation(self, obj):
        try:
            d = rh.Dossier.objects.filter(employe=obj.id, principal=True)[0]
            implantation = d.poste.implantation.nom
        except:
            implantation = None
        return implantation
    _implantation.short_description = u"Implantation"

    def _dossiers_postes(self, obj):
        l = []
        for d in obj.rh_dossiers.all().order_by('-date_debut'):
            link_style = u''
            list_style = u''
            if d.date_fin is not None and d.date_fin < datetime.date.today():
                link_style = u' style="color:#666;"'
                list_style = u' style="color:grey;"'

            dossier = u"""<a title="Aperçu du dossier"
                             href="%s"
                             onclick="return showAddAnotherPopup(this);"
                             title="Aperçu du dossier">
                             <img src="%simg/dossier-apercu.png" />
                          </a>
                          <a href="%s"%s>Dossier</a>
                          &nbsp;""" % \
                (reverse('dossier_apercu', args=(d.id,)),
                 settings.STATIC_URL,
                 reverse('admin:rh_dossier_change', args=(d.id,)),
                 link_style,)

            poste = u"""<a title="Aperçu du poste"
                           href="%s"
                           onclick="return showAddAnotherPopup(this);"
                           title="Aperçu du poste">
                           <img src="%simg/poste-apercu.png" />
                        </a>
                        <a href="%s"%s>%s [%d]</a>
                        &nbsp;""" % \
                (reverse('poste_apercu', args=(d.poste.id,)),
                 settings.STATIC_URL,
                 reverse('admin:rh_poste_change', args=(d.poste.id,)),
                 link_style,
                 d.poste.nom,
                 d.poste.id)

            link = u"""<li%s>%s %s</li>""" % \
                (list_style, dossier, poste)

            l.append(link)
        return "<ul>%s</ul>" % "\n".join(l)
    _dossiers_postes.allow_tags = True
    _dossiers_postes.short_description = u"Dossiers et postes"

    def _date_modification(self, obj):
        return date(obj.date_modification) \
                if obj.date_modification is not None else "(aucune)"
    _date_modification.short_description = u'date modification'
    _date_modification.admin_order_field = 'date_modification'

    def queryset(self, request):
        qs = super(EmployeAdminBase, self).queryset(request)
        return qs.select_related(depth=1).order_by('nom')

    def save_formset(self, request, form, formset, change):
        instances = formset.save(commit=False)
        for instance in instances:
            if instance.__class__ == rh.EmployeCommentaire:
                instance.owner = request.user
                instance.date_creation = datetime.datetime.now()
            instance.save()


class EmployeAdmin(reversion.VersionAdmin, EmployeAdminBase):
    change_list_template = "admin/rh/employe/change_list.html"
    ignore_duplicate_revisions = True


class EmployeProxyAdmin(EmployeAdminBase):
    list_display = ('_id', '_apercu', '_nom', '_organigramme')
    list_per_page = 500
    actions = None

    def __init__(self, *args, **kwargs):
        super(EmployeProxyAdmin, self).__init__(*args, **kwargs)
        self.list_display_links = (None, )

    def queryset(self, request):
        qs = super(ProtectRegionMixin, self).queryset(request)

        if in_drh_or_admin(request.user) or \
            user_gere_obj_de_sa_region(request.user):
            return qs

        return qs.none()

    def has_add_permission(self, obj):
        return False

    def has_change_permission(self, request, obj=None):
        user_groups = [g.name for g in request.user.groups.all()]
        if groups.CORRESPONDANT_RH in user_groups or \
           groups.ADMINISTRATEURS in user_groups or \
           groups.DIRECTEUR_DE_BUREAU in user_groups or \
           in_drh_or_admin(request.user):
            return True
        return False

    def _organigramme(self, obj):
        l = []
        for d in rh.Dossier.objects.filter(
            Q(date_fin__gt=datetime.date.today()) | Q(date_fin=None),
            Q(date_debut__lt=datetime.date.today()) | Q(date_debut=None),
            employe=obj.id
        ):
            organigramme = \
                    u'Organigramme, niveau: ' \
                    u'<input type="text" id="level_%s" ' \
                    u'style="width:30px;height:15px;" /> ' \
                    u'<input type="button" value="Générer" ' \
                    u"""onclick="window.location='%s' + """ \
                    u"""document.getElementById('level_%s').value" />""" % (
                        d.poste.id,
                        reverse('rho_employe_sans_niveau', args=(d.poste.id,)),
                        d.poste.id
                    )
            link = u"""<li>%s [%s]:<br />%s</li>""" % (
                    d.poste.nom,
                    d.poste.id,
                    organigramme
            )
            l.append(link)
        return "<ul>%s</ul>" % "\n".join(l)

    _organigramme.allow_tags = True
    _organigramme.short_description = "Organigramme"


class CategorieEmploiAdmin(reversion.VersionAdmin,
                           DerniereModificationAdmin, BaseAdmin):
    ignore_duplicate_revisions = True
    list_display = ('nom', 'derniere_modification')
    inlines = (TypePosteInline,)
    fieldsets = (
        (None, {'fields': ('nom', )}),
    )


class OrganismeBstgAdmin(reversion.VersionAdmin, DerniereModificationAdmin,
                         BaseAdmin):
    ignore_duplicate_revisions = True
    search_fields = ('nom',)
    list_display = ('nom', 'type', 'pays', 'derniere_modification')
    list_filter = ('type', )
    inlines = (DossierROInline,)
    fieldsets = (
        (None, {'fields': ('nom', 'type', 'pays',)}),
    )


class PosteAdmin(DateRangeMixin, ProtectRegionMixin, reversion.VersionAdmin,
                 AjaxSelect, DerniereModificationAdmin, BaseAdmin):
    change_list_template = "admin/rh/poste/change_list.html"
    ignore_duplicate_revisions = True
    form = make_ajax_form(rh.Poste, {
        'implantation': 'implantations',
        'type_poste': 'typepostes',
        'responsable': 'postes',
        'valeur_point_min': 'valeurpoints',
        'valeur_point_max': 'valeurpoints',
    })
    alphabet_filter = 'nom'
    search_fields = (
        'id',
        'nom',
        'implantation__nom',
        'implantation__zone_administrative__code',
        'implantation__zone_administrative__nom',
        'rh_dossiers__employe__id',
        'rh_dossiers__employe__nom',
        'rh_dossiers__employe__prenom',
    )
    list_display = (
        '_id', '_apercu', '_nom', '_occupe_par', 'implantation', '_service',
        '_responsable', 'date_debut', 'date_fin', 'derniere_modification',
        '_dae'
    )
    list_filter = (
        'implantation__zone_administrative',
        'implantation',
        'service',
        'type_poste',
        'type_poste__categorie_emploi',
        'type_poste__famille_professionnelle',
        'vacant',
    )
    list_display_links = ('_nom',)
    fieldsets = (
        (None, {'fields': (
                ('nom', 'nom_feminin'),
                'implantation',
                'type_poste',
                'service',
                'responsable',
                )}
                ),
        ('Contrat', {
            'fields': ((
                'regime_travail',
                'regime_travail_nb_heure_semaine'),
                )}
            ),
        ('Recrutement', {
            'fields': (('local', 'expatrie', 'mise_a_disposition', 'appel'),)}
            ),
        ('Rémunération', {
            'fields': (('classement_min',
                        'valeur_point_min',
                        'devise_min',
                        'salaire_min',
                        'indemn_min',
                        'autre_min',),
                       ('classement_max',
                        'valeur_point_max',
                        'devise_max',
                        'salaire_max',
                        'indemn_max',
                        'autre_max',),
                        )}),
        ('Comparatifs de rémunération', {
            'fields': ('devise_comparaison',
                       ('comp_locale_min', 'comp_locale_max'),
                       ('comp_universite_min', 'comp_universite_max'),
                       ('comp_fonctionpub_min', 'comp_fonctionpub_max'),
                       ('comp_ong_min', 'comp_ong_max'),
                       ('comp_autre_min', 'comp_autre_max'))}
            ),
        ('Justification', {
            'fields': ('justification',)}
            ),
        ('Autres Méta-données', {
            'fields': ('date_debut', 'date_fin')}
            ),
    )

    inlines = (PosteFinancementInline,
               PostePieceInline,
               DossierROInline,
               PosteComparaisonInline,
               PosteCommentaireInline, )

    def lookup_allowed(self, key, value):
        return key in (
            'date_debut__gte', 'date_debut__isnull', 'date_fin__lte',
            'date_fin__isnull',
            'implantation__zone_administrative__code__exact',
            'implantation__id__exact', 'type_poste__id__exact',
            'type_poste__categorie_emploi__id__exact', 'service__id__exact',
            'service__isnull', 'vacant__exact', 'vacant__isnull'
        ) or super(PosteAdmin, self).lookup_allowed(key, value)

    def _id(self, obj):
        return "%s" % obj.id
    _id.short_description = '#'
    _id.admin_order_field = 'id'

    def _apercu(self, poste):
        view_link = u"""<a onclick="return showAddAnotherPopup(this);"
                           title="Aperçu du poste"
                           href='%s'>
                        <img src="%simg/poste-apercu.png" />
                        </a>""" % \
                (reverse('poste_apercu', args=(poste.id,)),
                 settings.STATIC_URL,)
        return view_link
    _apercu.allow_tags = True
    _apercu.short_description = ''

    def _nom(self, poste):
        return """<a href="%s">%s</a>""" % \
                (reverse('admin:rh_poste_change', args=(poste.id,)),
                poste.nom)
    _nom.allow_tags = True
    _nom.short_description = u'Poste'
    _nom.admin_order_field = 'nom'

    def _occupe_par(self, obj):
        """Formatte la méthode Poste.occupe_par() pour l'admin"""
        output = u"Vacant"
        if obj.date_fin is not None and obj.date_fin < datetime.date.today():
            return u"s/o"
        employes = obj.occupe_par()
        if employes:
            l = []
            for e in employes:
                link = u"""<a href='%s'
                              title='Aperçu de l\'employé'
                              onclick='return showAddAnotherPopup(this)'>
                              <img src='%simg/employe-apercu.png' />
                            </a>
                            <a href='%s'>%s</a>""" % \
                     (reverse('employe_apercu', args=(e.id,)),
                     settings.STATIC_URL,
                     reverse('admin:rh_employe_change', args=(e.id,)),
                     e)
                l.append(link)
            output = "\n<br />".join(l)
        return output
    _occupe_par.allow_tags = True
    _occupe_par.short_description = "Occupé par"

    def _zone_administrative(self, poste):
        return poste.implantation.zone_administrative.code
    _zone_administrative.short_description = 'Zone administrative'
    _zone_administrative.admin_order_field = \
            'implantation__zone_administrative__code'

    def _implantation(self, poste):
        return poste.implantation.nom
    _implantation.short_description = 'Implantation'
    _implantation.admin_order_field = 'implantation'

    def _service(self, obj):
        return obj.service
    _service.short_description = 'Service'
    _service.allow_tags = True

    def _responsable(self, obj):
        try:
            responsable = u"""<a href="%s"
                                 onclick="return showAddAnotherPopup(this)">
                                <img src="%simg/poste-apercu.png"
                                     title="Aperçu du poste" />
                              </a>
                              <a href="%s">%s [%d]</a>
                              <br />""" % \
                (reverse('poste_apercu', args=(obj.responsable.id,)),
                settings.STATIC_URL,
                reverse('admin:rh_poste_change', args=(obj.responsable.id,)),
                obj.responsable.nom,
                obj.responsable.id)
        except:
            responsable = ''

        try:
            dossier = obj.responsable.rh_dossiers.all() \
                    .order_by('-date_debut')[0]
            employe_id = dossier.employe.id
            employe_html = u"""<br />
                    <a href="%s"
                       onclick="return showAddAnotherPopup(this)">
                       <img src="%simg/employe-apercu.png"
                            title="Aperçu de l'employé">
                    </a>
                    <a href="%s">%s</a>""" % \
                    (reverse('employe_apercu', args=(employe_id,)),
                     settings.STATIC_URL,
                     reverse('admin:rh_employe_change', args=(employe_id,)),
                     dossier.employe)
        except:
            employe_html = ""

        return "%s %s" % (responsable, employe_html)
    _responsable.short_description = 'Responsable'
    _responsable.allow_tags = True

    def _date_debut(self, obj):
        return date_format(obj.date_debut)
    _date_debut.short_description = u'Début'
    _date_debut.admin_order_field = 'date_debut'

    def _date_fin(self, obj):
        return date_format(obj.date_fin)
    _date_fin.short_description = u'Fin'
    _date_fin.admin_order_field = 'date_fin'

    def _dae(self, poste):
        apercu_link = ""
        postes_dae = poste.postes_dae.all()
        if len(postes_dae) > 0:
            poste_dae = postes_dae[0]
            apercu_link = \
                    u'<a title="Aperçu du dossier" href="%s" ' \
                    u'onclick="return showAddAnotherPopup(this);">' \
                    u'<img src="%simg/loupe.png" /></a>' % (reverse(
                        'poste_consulter', args=("dae-%s" % poste_dae.id,)
                    ), settings.STATIC_URL)
        return apercu_link
    _dae.allow_tags = True
    _dae.short_description = u"DAE"

    def save_formset(self, request, form, formset, change):
        instances = formset.save(commit=False)
        for instance in instances:
            if instance.__class__ == rh.PosteCommentaire:
                instance.owner = request.user
                instance.date_creation = datetime.datetime.now()
            instance.save()
        formset.save_m2m()


class ResponsableInline(admin.TabularInline):
    model = rh.ResponsableImplantation
    extra = 0
    fk_name = "implantation"
    form = ResponsableInlineForm


class ResponsableImplantationAdmin(BaseAdmin):
    actions = None
    fields = ('nom', )
    inlines = (ResponsableInline, )
    list_filter = ('zone_administrative', 'statut', )
    list_display = ('_zone_administrative', '_nom', 'statut', '_responsable', )
    list_display_links = ('_nom',)
    list_per_page = 500
    readonly_fields = ('nom', )
    search_fields = (
            'nom',
            'responsable__employe__id',
            'responsable__employe__nom',
            'responsable__employe__prenom',
            )
    ordering = ('nom',)
    inlines = (ResponsableInline, )

    def _zone_administrative(self, obj):
        return obj.zone_administrative.code
    _zone_administrative.short_description = u"Zone administrative"
    _zone_administrative.admin_order_field = 'zone_administrative__code'

    def _nom(self, obj):
        return obj.nom
    _nom.short_description = u"Implantation"
    _nom.admin_order_field = 'nom'

    def _responsable(self, obj):
        try:
            employe = employe = obj.responsable.employe
        except Exception, e:
            return (
                u"<span style='color: red;'>"
                u"Pas de responsable</span><!-- %s -->" % e
            )
        try:
            dossiers = employe.dossiers_encours()
            if len(dossiers) == 0:
                return u"<span style='color: red;'>%s %s </span>" % (
                    employe, u"sans dossier actif")
            else:
                return employe
        except Exception, e:
            return u"<!-- %s -->" % e
    _responsable.allow_tags = True
    _responsable.short_description = u"Responsable"
    _responsable.admin_order_field = 'responsable__employe__nom'

    def has_add_permission(self, request=None):
        return False

    def has_change_permission(self, request, obj=None):
        return in_drh_or_admin(request.user)

    def has_delete_permission(self, request, obj=None):
        return False


class ServiceAdminBase(ArchivableAdmin, DerniereModificationAdmin, BaseAdmin):
    list_display = ('nom', 'derniere_modification', '_archive')
    list_filter = ('archive', )
    fieldsets = (
        (None, {'fields': ('nom', 'archive')}),
    )


class ServiceAdmin(reversion.VersionAdmin, ServiceAdminBase):
    ignore_duplicate_revisions = True


class ServiceProxyAdmin(ServiceAdminBase):
    list_display = ('nom', '_organigramme', '_archive', )
    actions = None

    def __init__(self, *args, **kwargs):
        super(ServiceProxyAdmin, self).__init__(*args, **kwargs)
        self.list_display_links = (None, )

    def queryset(self, request):
        return super(ServiceProxyAdmin, self).queryset(request) \
                .annotate(num_postes=Count('rh_postes')) \
                .filter(num_postes__gt=0)

    def has_add_permission(self, obj):
        return False

    def has_change_permission(self, request, obj=None):
        user_groups = [g.name for g in request.user.groups.all()]
        if groups.CORRESPONDANT_RH in user_groups or \
           groups.ADMINISTRATEURS in user_groups or \
           groups.DIRECTEUR_DE_BUREAU in user_groups or \
           in_drh_or_admin(request.user):
            return True
        return False

    def _organigramme(self, obj):
        return """<a href="%s"><strong>Organigramme</strong></a>""" % \
                (reverse('rho_service', args=(obj.id,)))
    _organigramme.allow_tags = True
    _organigramme.short_description = "Organigramme"


class StatutAdmin(reversion.VersionAdmin,
                  ArchivableAdmin,
                  DerniereModificationAdmin,
                  BaseAdmin):
    ignore_duplicate_revisions = True
    list_display = ('code', 'nom', 'derniere_modification', '_archive')
    list_filter = ('archive', )
    fieldsets = (
        (None, {
            'fields': ('code', 'nom', 'archive'),
        }),
    )


class TauxChangeAdmin(reversion.VersionAdmin, DerniereModificationAdmin,
                      BaseAdmin):
    ignore_duplicate_revisions = True
    list_display = ('taux', 'devise', 'annee', 'derniere_modification')
    list_filter = ('devise',)
    fieldsets = (
        (None, {
            'fields': ('taux', 'devise', 'annee', ),
        }),
    )


class TypeContratAdmin(reversion.VersionAdmin,
                       ArchivableAdmin,
                       DerniereModificationAdmin,
                       BaseAdmin):
    ignore_duplicate_revisions = True
    list_display = ('nom', 'nom_long', 'derniere_modification', '_archive')
    list_filter = ('archive', )
    fieldsets = (
        (None, {
            'fields': ('nom', 'nom_long', 'archive'),
        }),
    )


class TypePosteAdmin(reversion.VersionAdmin,
                     ArchivableAdmin,
                     DerniereModificationAdmin,
                     BaseAdmin):
    ignore_duplicate_revisions = True
    search_fields = ('nom', 'nom_feminin', )
    list_display = ('nom', 'categorie_emploi',
                     'derniere_modification', '_archive',)
    list_filter = ('categorie_emploi', 'famille_professionnelle', 'archive')
    fieldsets = (
        (None, {
            'fields': (
                'nom', 'nom_feminin', 'is_responsable', 'categorie_emploi',
                'famille_professionnelle',
                'archive',
            )
        }),
    )


class TypeRemunerationAdmin(reversion.VersionAdmin, ArchivableAdmin,
                            DerniereModificationAdmin, BaseAdmin):
    ignore_duplicate_revisions = True
    list_display = (
        'nom', 'type_paiement', 'nature_remuneration',
        'derniere_modification', '_archive'
    )
    list_filter = ('archive', )
    fieldsets = (
        (None, {
            'fields': (
                'nom', 'type_paiement', 'nature_remuneration', 'archive'
            )
        }),
    )


class TypeRevalorisationAdmin(reversion.VersionAdmin,
                              ArchivableAdmin,
                              DerniereModificationAdmin,
                              BaseAdmin):
    ignore_duplicate_revisions = True
    list_display = ('nom', 'derniere_modification', '_archive')
    list_filter = ('archive', )
    fieldsets = (
        (None, {'fields': ('nom', 'archive')}),
    )


class ValeurPointAdmin(reversion.VersionAdmin,
                       DerniereModificationAdmin,
                       BaseAdmin):
    ignore_duplicate_revisions = True
    list_display = (
        '_devise_code', '_devise_nom', 'annee', 'implantation',
        'valeur', 'derniere_modification'
    )
    list_filter = ('annee', 'devise', 'implantation__zone_administrative', )
    fieldsets = (
        (None, {'fields': ('valeur', 'devise', 'implantation', 'annee')}),
    )

    def queryset(self, request):
        return super(ValeurPointAdmin, self).queryset(request) \
                .select_related('devise', 'implantation')

    def _devise_code(self, obj):
        return obj.devise.code
    _devise_code.short_description = "Code de la devise"

    def _devise_nom(self, obj):
        return obj.devise.nom
    _devise_nom.short_description = "Nom de la devise"


class ImplantationProxyAdmin(BaseAdmin):
    list_display = ('nom', '_organigramme')
    actions = None

    def __init__(self, *args, **kwargs):
        super(ImplantationProxyAdmin, self).__init__(*args, **kwargs)
        self.list_display_links = (None, )

    def has_add_permission(self, obj):
        return False

    def has_change_permission(self, request, obj=None):
        user_groups = [g.name for g in request.user.groups.all()]
        if groups.CORRESPONDANT_RH in user_groups or \
           groups.ADMINISTRATEURS in user_groups or \
           groups.DIRECTEUR_DE_BUREAU in user_groups or \
           in_drh_or_admin(request.user):
            return True
        return False

    def _organigramme(self, obj):
        return '<a href="%s"><strong>Organigramme</strong></a>' % (
            reverse('rho_implantation', args=(obj.id,))
        )
    _organigramme.allow_tags = True
    _organigramme.short_description = "Organigramme"


class RegionProxyAdmin(BaseAdmin):
    list_display = ('nom', '_organigramme')
    actions = None

    def __init__(self, *args, **kwargs):
        super(RegionProxyAdmin, self).__init__(*args, **kwargs)
        self.list_display_links = (None, )

    def has_add_permission(self, obj):
        return False

    def has_change_permission(self, request, obj=None):
        user_groups = [g.name for g in request.user.groups.all()]
        if groups.CORRESPONDANT_RH in user_groups or \
           groups.ADMINISTRATEURS in user_groups or \
           groups.DIRECTEUR_DE_BUREAU in user_groups or \
           in_drh_or_admin(request.user):
            return True
        return False

    def _organigramme(self, obj):
        return """<a href="%s"><strong>Organigramme</strong></a>""" % (
            reverse('rho_region', args=(obj.id,))
        )
    _organigramme.allow_tags = True
    _organigramme.short_description = "Organigramme"


class ProfileInline(admin.StackedInline):
    model = rh.UserProfile
    

class RHUserAdmin(UserAdmin):
    inlines = list(UserAdmin.inlines) + [ProfileInline]
    

def _invalidate(modaladmin, req, qs):
    qs.update(valide=False)
_invalidate.short_description = 'Invalider'
    

def _communique(modaladmin, req, qs):
    url_prefix = ('https://' if req.is_secure() else 'http://') + (
        req.META['SERVER_NAME'] if 'SERVER_NAME' in req.META else
        'localhost') + (':%s' % (req.META['SERVER_PORT']) if 'SERVER_PORT'
        in req.META else '')

    types = [x[0] for x in rh.TYPES_CHANGEMENT]

    # make a list of all possible email types. It is reversed so that
    # we start with the largest combinations.
    combs = list(itertools.chain(*map(
            lambda x: itertools.combinations(types, x),
            range(1, len(types)+1))))
    
    combs = reversed(combs)

    recipient_list = ref.Employe.objects.filter(
        changement_notifications__in=
        rh.ChangementPersonnelNotifications.objects.all()).distinct()

    thead_colors = {
        'NO': 'ff99ff',
        'MO': 'ccccff',
        'DE': 'ff99ff',
        }


    for comb in combs:
        # qf c'est pour trouver les recipients.
        qf = reduce(
            lambda x, y: x & y,
            [Q(**{'changement_notifications__type': t}) for t in comb]
            )
        recipients = recipient_list.filter(qf)

        recipient_list = recipient_list.exclude(
            id__in=recipients.values_list('id', flat=True))

        types_dict = dict(rh.TYPES_CHANGEMENT)

        if len(recipients):
            ctx = {
                'types': [
                    ]
                }

            for t in comb:
                changements = []
                ctx['types'].append({
                        'bgcolor': thead_colors[t],
                        'name': types_dict[t],
                        'changements': changements,
                        })

                for ch in qs.filter(type__in=[t]):
                    changements.append(ch)
                    

            template = loader.get_template('email/mouvement_employe.html')
            content = template.render(Context(ctx))

            sujet = '[SGRH] ARRIVÉES - DÉPARTS - MOBILITÉS'

            msg = EmailMessage(
                sujet,
                content,
                settings.SERVER_EMAIL,
                [x.courriel for x in recipients],
                )
            msg.content_subtype = "html"
            msg.send()
        
    qs.update(
        communique=True,
        date_communication=datetime.datetime.now()
        )

_communique.short_description = u'Envoyer aux desinataires prévus.'


class ChangementPersonnelAdmin(admin.ModelAdmin):

    actions = (
        _invalidate,
        _communique,
        )

    def _dossier(obj):
        return obj.dossier.__unicode__()
    _dossier.short_description = u'Dossier'

    list_display = (
        'type', _dossier, 'valide', 'communique', 'date_communication'
        )

    list_filter = (
        'type',
        'valide',
        'communique',
        )
    

admin.site.unregister(User)
admin.site.register(User, RHUserAdmin)

admin.site.register(rh.Classement, ClassementAdmin)
admin.site.register(rh.Devise, DeviseAdmin)
admin.site.register(rh.Dossier, DossierAdmin)
admin.site.register(EmployeProxy, EmployeProxyAdmin)
admin.site.register(ServiceProxy, ServiceProxyAdmin)
admin.site.register(rh.Employe, EmployeAdmin)
admin.site.register(rh.CategorieEmploi, CategorieEmploiAdmin)
admin.site.register(rh.FamilleProfessionnelle)
admin.site.register(rh.OrganismeBstg, OrganismeBstgAdmin)
admin.site.register(rh.Poste, PosteAdmin)
admin.site.register(
    rh.ResponsableImplantationProxy, ResponsableImplantationAdmin
)
admin.site.register(rh.Service, ServiceAdmin)
admin.site.register(rh.Statut, StatutAdmin)
admin.site.register(rh.TauxChange, TauxChangeAdmin)
admin.site.register(rh.TypeContrat, TypeContratAdmin)
admin.site.register(rh.TypePoste, TypePosteAdmin)
admin.site.register(rh.TypeRemuneration, TypeRemunerationAdmin)
admin.site.register(rh.TypeRevalorisation, TypeRevalorisationAdmin)
admin.site.register(rh.ValeurPoint, ValeurPointAdmin)
admin.site.register(ImplantationProxy, ImplantationProxyAdmin)
admin.site.register(RegionProxy, RegionProxyAdmin)
admin.site.register(rh.ChangementPersonnelNotifications)
admin.site.register(rh.ChangementPersonnel, ChangementPersonnelAdmin)