Ajouté les filtres les plus faciles.

[auf_savoirs_en_partage_django.git] / auf_savoirs_en_partage / chercheurs / admin.py

# -*- encoding: utf-8 -*-
from chercheurs.models import Chercheur, Publication, Groupe, ChercheurGroupe
from django.db import models
from django.contrib import admin
from django.core.urlresolvers import reverse as url
from django.forms.models import BaseInlineFormSet
from django.http import HttpResponseRedirect

class ChercheurAdmin(admin.ModelAdmin):
    list_filter = ('genre', 'statut', 'membre_reseau_institutionnel', 'membre_instance_auf', 'discipline', 'groupes')
    list_per_page = 25
    actions = ('remove_from_group',)
    search_fields = ('nom', 'prenom')

    def lookup_allowed(self, lookup):
        return lookup in ['groupes__id__exact', 'discipline__id__exact'] or \
               admin.ModelAdmin.lookup_allowed(self, lookup)

    def remove_from_group(self, request, queryset):
        groupe_id = request.GET.get('groupes__id__exact')
        chercheur_ids = queryset.values_list('id', flat=True)
        matches = ChercheurGroupe.objects.filter(groupe=groupe_id, chercheur__in=chercheur_ids)
        matches.delete()
        return HttpResponseRedirect(url('admin:chercheurs_chercheur_changelist') + '?groupes__id__exact=' + groupe_id)

    def get_actions(self, request):
        actions = super(ChercheurAdmin, self).get_actions(request)

        # Si on filtre par groupe de recherche, offrir d'en retirer les
        # chercheurs sélectionnés.
        groupe_id = request.GET.get('groupes__id__exact')
        if groupe_id:
            groupe = Groupe.objects.get(id=groupe_id)
            action_desc = actions['remove_from_group']
            actions['remove_from_group'] = (action_desc[0], action_desc[1], u'Retirer du domaine de recherche « %s »' % groupe.nom)
        else:
            del actions['remove_from_group']
        return actions

admin.site.register(Chercheur, ChercheurAdmin)
admin.site.register(Publication)
admin.site.register(Groupe)