auf_savoirs_en_partage/chercheurs/admin.py

   1 # -*- coding: utf-8 -*-
   2 from django.db import models
   3 from django.db.models import Q
   4 from django.contrib import admin
   5 from django.core.urlresolvers import reverse as url
   6 from django.forms.models import BaseInlineFormSet
   7 from django.http import HttpResponseRedirect
   8 from django.utils.encoding import smart_str
   9 from django_exportateur.exportateur import exportateur
  10
  11 from chercheurs.models import Chercheur, Publication, Groupe, ChercheurGroupe, ChercheurQuerySet, These
  12
  13 class ChercheurAdmin(admin.ModelAdmin):
  14     list_filter = ['genre']
  15     list_per_page = 9999
  16     actions = ('remove_from_group', 'export_as_ods', 'export_as_csv')
  17     search_fields = ('nom', 'prenom')
  18
  19     def lookup_allowed(self, lookup, value):
  20         return lookup in ['genre', 'statut', 'membre_reseau_institutionnel',
  21                           'membre_instance_auf', 'discipline', 'region', 'pays',
  22                           'groupes', 'nord_sud', 'initial'] or \
  23                admin.ModelAdmin.lookup_allowed(self, lookup, value)
  24
  25     def remove_from_group(self, request, queryset):
  26         groupe_id = request.GET.get('groupes__id__exact')
  27         chercheur_ids = queryset.values_list('id', flat=True)
  28         matches = ChercheurGroupe.objects.filter(groupe=groupe_id, chercheur__in=chercheur_ids)
  29         matches.delete()
  30         return HttpResponseRedirect(url('admin:chercheurs_chercheur_changelist') + '?groupes__id__exact=' + groupe_id)
  31
  32     def get_actions(self, request):
  33         actions = super(ChercheurAdmin, self).get_actions(request)
  34
  35         # Si on filtre par groupe de recherche, offrir d'en retirer les
  36         # chercheurs sélectionnés.
  37         groupe_id = request.GET.get('groupes__id__exact')
  38         if groupe_id:
  39             groupe = Groupe.objects.get(id=groupe_id)
  40             action_desc = actions['remove_from_group']
  41             actions['remove_from_group'] = (action_desc[0], action_desc[1], u'Retirer du domaine de recherche « %s »' % groupe.nom)
  42         else:
  43             del actions['remove_from_group']
  44         return actions
  45
  46     def queryset(self, request):
  47         return ChercheurAdminQuerySet(Chercheur)
  48
  49     def get_object(self, request, object_id):
  50         """On doit réimplémenter cette méthode à cause de ce qu'on fait avec "initial" dans la méthode queryset()."""
  51         try:
  52             return Chercheur.objects.get(id=object_id)
  53         except Chercheur.DoesNotExist:
  54             return None
  55
  56     def changelist_view(self, request, extra_context=None):
  57         initials = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  58         current_initial = request.GET.get('initial', 'A')
  59         return super(ChercheurAdmin, self).changelist_view(request, extra_context={
  60             'initials': initials,
  61             'current_initial': current_initial
  62         })
  63
  64     def export(self, queryset, type):
  65         if queryset.count() == 0:
  66             return None
  67         obj = queryset[0]
  68         headers = ['Nom', 'Prénom', 'Genre', 'Courriel', 'Téléphone', 'Adresse postale',
  69                    'Statut', 'Diplôme', 'Établissement', 'Pays', 'Domaines de recherche',
  70                    'Thèse', 'Directeur', 'Discipline', 'Thèmes de recherche', 'Groupe de recherche', 'Mots-clés',
  71                    'Site web', 'Blog', 'Réseau social',
  72                    'Membre instance AUF', "Sollicité par l'OIF", 'Membre société francophone',
  73                    'Membre instance réseau institutionnel AUF', 'Expertises', 'Solliciter pour expertises',
  74                    'Publications']
  75         data = []
  76         for c in queryset:
  77             row = []
  78             row.append(c.nom)
  79             row.append(c.prenom)
  80             row.append(c.get_genre_display())
  81             row.append(c.courriel)
  82             row.append(c.telephone)
  83             row.append(c.adresse_postale)
  84             row.append(c.get_statut_display())
  85             row.append(c.diplome)
  86             row.append(c.nom_etablissement)
  87             row.append(c.pays)
  88             row.append(', '.join(g.nom for g in c.groupes.all()))
  89             try:
  90                 t = c.these
  91                 row.append('%s, %s, %s, %s pages.' % (t.titre, t.etablissement, t.annee, t.nb_pages))
  92                 row.append(t.directeur)
  93             except These.DoesNotExist:
  94                 row.append('')
  95                 row.append('')
  96             row.append(c.discipline.nom if c.discipline else '')
  97             row.append(c.theme_recherche)
  98             row.append(c.groupe_recherche)
  99             row.append(c.mots_cles)
 100             row.append(c.url_site_web)
 101             row.append(c.url_blog)
 102             row.append(c.url_reseau_social)
 103             if c.membre_instance_auf:
 104                 row.append(', '.join([c.membre_instance_auf_nom, c.membre_instance_auf_fonction, c.membre_instance_auf_dates]))
 105             else:
 106                 row.append('')
 107             if c.expert_oif:
 108                 row.append(', '.join([c.expert_oif_details, c.expert_oif_dates]))
 109             else:
 110                 row.append('')
 111             if c.membre_association_francophone:
 112                 row.append(c.membre_association_francophone_details)
 113             else:
 114                 row.append('')
 115             if c.membre_reseau_institutionnel:
 116                 row.append(', '.join([c.membre_reseau_institutionnel_nom, c.membre_reseau_institutionnel_fonction, c.membre_reseau_institutionnel_dates]))
 117             else:
 118                 row.append('')
 119             row.append('; '.join(', '.join([e.nom, e.date, e.organisme_demandeur]) for e in c.expertises.all()))
 120             if c.expertises_auf is None:
 121                 row.append('')
 122             else:
 123                 row.append('Oui' if c.expertises_auf else 'Non')
 124             row.append('; '.join(p.publication_affichage if p.publication_affichage else
 125                                  '%s, %s, %s, %s, %s, %s, %s pages.' %
 126                                  (p.auteurs, p.titre, p.revue, p.annee, p.editeur, p.lieu_edition, p.nb_pages)
 127                                  for p in c.publications.all()))
 128             data.append([smart_str(x) if x else '' for x in row])
 129         return exportateur(headers, data, type, filename='chercheurs.%s' % type)
 130
 131     def export_as_csv(self, request, queryset):
 132         return self.export(queryset, 'csv')
 133     export_as_csv.short_description = 'Export CSV'
 134
 135     def export_as_ods(self, request, queryset):
 136         return self.export(queryset, 'ods')
 137     export_as_ods.short_description = 'Export ODS'
 138
 139 class ChercheurAdminQuerySet(ChercheurQuerySet):
 140
 141     def filter(self, *args, **kwargs):
 142         """Gère des filtres supplémentaires pour l'admin.
 143
 144         C'est la seule façon que j'ai trouvée de contourner les mécanismes
 145         de recherche de l'admin."""
 146         pays = kwargs.pop('pays', None)
 147         region = kwargs.pop('region', None)
 148         nord_sud = kwargs.pop('nord_sud', None)
 149         expert = kwargs.pop('expert', None)
 150         initial = kwargs.pop('initial', 'A')
 151         qs = super(ChercheurAdminQuerySet, self).filter(nom__istartswith=initial)
 152         if pays is not None:
 153             qs = qs.filter(Q(etablissement__pays=pays) |
 154                            (Q(etablissement=None) & Q(etablissement_autre_pays=pays)))
 155         elif region is not None:
 156             qs = qs.filter(Q(etablissement__pays__region=region) |
 157                            (Q(etablissement=None) & Q(etablissement_autre_pays__region=region)))
 158         elif nord_sud is not None:
 159             qs = qs.filter(Q(etablissement__pays__nord_sud=nord_sud) |
 160                            (Q(etablissement=None) & Q(etablissement_autre_pays__nord_sud=nord_sud)))
 161         if expert is not None:
 162             if expert in ['1', 1, True]:
 163                 qs = qs.exclude(expertises=None)
 164             else:
 165                 qs = qs.filter(expertises=None)
 166
 167         return super(ChercheurAdminQuerySet, qs).filter(*args, **kwargs)
 168
 169 admin.site.register(Chercheur, ChercheurAdmin)
 170 admin.site.register(Publication)
 171 admin.site.register(Groupe)
 172