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
   9 from chercheurs.models import Chercheur, ChercheurVoir, Publication, \
  10                               GroupeChercheur, DomaineRecherche, \
  11                               AdhesionGroupe, ChercheurQuerySet, \
  12                               AdhesionCommunaute, AdhesionDomaineRecherche, \
  13                               Groupe
  14 from chercheurs.utils import export
  15 from savoirs.models import Search
  16
  17 class ChercheurAdmin(admin.ModelAdmin):
  18     list_filter = ['genre']
  19     alphabet_filter = 'nom'
  20     alphabet_filter_table = 'chercheurs_personne'
  21     DEFAULT_ALPHABET = ''
  22
  23     actions = ('remove_from_group', 'export_as_ods', 'export_as_csv')
  24     search_fields = ('nom', 'prenom')
  25
  26     def lookup_allowed(self, lookup, value):
  27         return lookup in ['genre', 'statut', 'membre_reseau_institutionnel',
  28                           'membre_instance_auf', 'discipline', 'region', 'pays',
  29                           'groupes', 'nord_sud'] or \
  30                admin.ModelAdmin.lookup_allowed(self, lookup, value)
  31
  32     def remove_from_group(self, request, queryset):
  33         groupe_id = request.GET.get('groupes__id__exact')
  34         chercheur_ids = queryset.values_list('id', flat=True)
  35         matches = AdhesionGroupe.objects.filter(groupe=groupe_id, chercheur__in=chercheur_ids)
  36         matches.delete()
  37         return HttpResponseRedirect(url('admin:chercheurs_chercheur_changelist') + '?groupes__id__exact=' + groupe_id)
  38
  39     def get_actions(self, request):
  40         actions = super(ChercheurAdmin, self).get_actions(request)
  41
  42         # Si on filtre par groupes, offrir d'en retirer les
  43         # chercheurs sélectionnés.
  44         groupe_id = request.GET.get('groupes__id__exact')
  45         if groupe_id:
  46             groupe = Groupe.objects.get(id=groupe_id)
  47             action_desc = actions['remove_from_group']
  48             actions['remove_from_group'] = (action_desc[0], action_desc[1], u'Retirer du domaine de recherche « %s »' % groupe.nom)
  49         else:
  50             del actions['remove_from_group']
  51         return actions
  52
  53     def queryset(self, request):
  54         return ChercheurAdminQuerySet(Chercheur)
  55
  56     def get_object(self, request, object_id):
  57         """On doit réimplémenter cette méthode à cause de ce qu'on fait avec "initial" dans la méthode queryset()."""
  58         try:
  59             return Chercheur.objects.get(id=object_id)
  60         except Chercheur.DoesNotExist:
  61             return None
  62
  63
  64     def export_as_csv(self, request, queryset):
  65         return export(queryset, 'csv')
  66     export_as_csv.short_description = 'Export CSV'
  67
  68     def export_as_ods(self, request, queryset):
  69         return export(queryset, 'ods')
  70     export_as_ods.short_description = 'Export ODS'
  71
  72
  73 class ChercheurVoirAdmin(ChercheurAdmin):
  74
  75     list_editable = []
  76     fields = ['salutation', 'nom', 'prenom', 'courriel', 'afficher_courriel',
  77               'fonction', 'date_naissance', 'sousfonction', 'telephone',
  78               'adresse_postale', 'genre', 'commentaire',
  79
  80               'nationalite', 'statut', 'diplome', 'etablissement',
  81               'etablissement_autre_nom', 'etablissement_autre_pays',
  82               'attestation', 'thematique', 'mots_cles', 'discipline',
  83               'theme_recherche', 'equipe_recherche', 'url_site_web',
  84               'url_blog', 'url_reseau_social',
  85               'membre_instance_auf', 'membre_instance_auf_nom',
  86               'membre_instance_auf_fonction', 'membre_instance_auf_dates',
  87               'expert_oif', 'expert_oif_details', 'expert_oif_dates',
  88               'membre_association_francophone', 'membre_association_francophone_details',
  89               'membre_reseau_institutionnel', 'membre_reseau_institutionnel_nom',
  90               'membre_reseau_institutionnel_fonction', 'membre_reseau_institutionnel_dates',
  91               'expertises_auf']
  92
  93     def __init__(self, model, admin_site):
  94         super(ChercheurVoirAdmin, self).__init__(model, admin_site)
  95         self.readonly_fields = self.fields
  96
  97
  98 admin.site.register(ChercheurVoir, ChercheurVoirAdmin)
  99
 100 class ChercheurAdminQuerySet(ChercheurQuerySet):
 101
 102     def filter(self, *args, **kwargs):
 103         """Gère des filtres supplémentaires pour l'admin.
 104
 105         C'est la seule façon que j'ai trouvée de contourner les mécanismes
 106         de recherche de l'admin."""
 107         pays = kwargs.pop('pays', None)
 108         region = kwargs.pop('region', None)
 109         nord_sud = kwargs.pop('nord_sud', None)
 110         expert = kwargs.pop('expert', None)
 111         qs = self
 112         if pays is not None:
 113             qs = qs.filter(Q(etablissement__pays=pays) |
 114                            (Q(etablissement=None) & Q(etablissement_autre_pays=pays)))
 115         elif region is not None:
 116             qs = qs.filter(Q(etablissement__pays__region=region) |
 117                            (Q(etablissement=None) & Q(etablissement_autre_pays__region=region)))
 118         elif nord_sud is not None:
 119             qs = qs.filter(Q(etablissement__pays__nord_sud=nord_sud) |
 120                            (Q(etablissement=None) & Q(etablissement_autre_pays__nord_sud=nord_sud)))
 121         if expert is not None:
 122             if expert in ['1', 1, True]:
 123                 qs = qs.exclude(expertises=None)
 124             else:
 125                 qs = qs.filter(expertises=None)
 126
 127         return super(ChercheurAdminQuerySet, qs).filter(*args, **kwargs)
 128
 129
 130 class AdhesionGroupeAdmin(admin.ModelAdmin):
 131     list_filter = ('groupe','statut')
 132     list_display = ('groupe', 'chercheur', 'statut')
 133     list_editable = ('statut',)
 134     search_fields = ('chercheur__nom', 'chercheur__prenom')
 135
 136     alphabet_filter = 'chercheur__nom'
 137     DEFAULT_ALPHABET = ''
 138
 139     actions = ['assigner_cgstatut']
 140
 141
 142     def lookup_allowed(self, lookup, value):
 143         return lookup in ['chercheur__nom__istartswith'] or \
 144                admin.ModelAdmin.lookup_allowed(self, lookup, value)
 145
 146     def queryset(self, request):
 147         qs = super(AdhesionGroupeAdmin, self).queryset(request)
 148
 149         if not request.user.is_superuser and not request.user.has_perm('chercheurs.change_adhesiongroupe'):
 150             qs = qs.filter(groupe__responsables=request.user)
 151
 152         return qs
 153
 154     def has_change_permission(self, request, obj=None):
 155
 156         if not obj:
 157             if request.user.responsable_groupe.count():
 158                 return True
 159         else:
 160             if request.user in obj.groupe.responsables.all():
 161                 return True
 162
 163         return super(AdhesionGroupeAdmin, self).has_change_permission(request, obj)
 164
 165     def assigner_cgstatut(self, request, queryset):
 166         selected = request.POST.getlist(admin.ACTION_CHECKBOX_NAME)
 167         return HttpResponseRedirect("/admin/assigner_%s?ids=%s" % ('cgstatut', ",".join(selected)))
 168     assigner_cgstatut.short_description = u'Assigner un statut'
 169
 170
 171 class AdhesionCommunauteAdmin(AdhesionGroupeAdmin):
 172     pass
 173
 174
 175 class AdhesionDomaineRechercheAdmin(AdhesionGroupeAdmin):
 176     pass
 177
 178 class BaseGroupeAdmin(admin.ModelAdmin):
 179     search_fields = ['nom']
 180     fieldsets = (
 181         (('Options générales'), {'fields': ('nom', 'url', 'liste_diffusion',
 182                                             'bulletin', 'page_accueil')}),
 183         (('Gestionnaire de communauté'), {'fields': ('responsables',)}),
 184         (('Recherches prédéfinies'), {'fields': ('recherches',)}),
 185     )
 186
 187     class Media:
 188         js = ['js/tiny_mce/tiny_mce.js', 'js/tiny_mce_textareas.js']
 189
 190     def save_model(self, request, obj, form, change):
 191         responsables = form.cleaned_data['responsables']
 192         for user in responsables:
 193             user.is_staff = True
 194             user.save()
 195
 196         if not request.user.is_superuser:
 197             recherches = Search.objects.exclude(user=request.user)
 198             form.cleaned_data['recherches'] = form.cleaned_data['recherches'] | recherches
 199
 200         super(BaseGroupeAdmin, self).save_model(request, obj, form, change)
 201
 202     def queryset(self, request):
 203         qs = super(BaseGroupeAdmin, self).queryset(request)
 204
 205         if not request.user.is_superuser and not request.user.has_perm('chercheurs.change_groupechercheur'):
 206             qs = qs.filter(responsables=request.user)
 207
 208         return qs
 209
 210     def has_change_permission(self, request, obj=None, groupe_chercheur=False):
 211
 212         if not obj:
 213             if request.user.responsable_groupe.filter(groupe_chercheur=groupe_chercheur).count():
 214                 return True
 215         else:
 216             if request.user in obj.responsables.all():
 217                 return True
 218
 219         return super(BaseGroupeAdmin, self).has_change_permission(request, obj)
 220
 221     def formfield_for_manytomany(self, db_field, request, **kwargs):
 222         if db_field.name == "recherches" and not request.user.is_superuser:
 223             kwargs["queryset"] = Search.objects.filter(user=request.user)
 224             return db_field.formfield(**kwargs)
 225         return super(BaseGroupeAdmin, self).formfield_for_manytomany(db_field, request, **kwargs)
 226
 227
 228 class GroupeChercheurAdmin(BaseGroupeAdmin):
 229
 230     def has_change_permission(self, request, obj=None):
 231         return super(GroupeChercheurAdmin, self).has_change_permission(request, obj, groupe_chercheur=True)
 232
 233
 234 class DomaineRechercheAdmin(BaseGroupeAdmin):
 235
 236     def has_change_permission(self, request, obj=None):
 237         return super(DomaineRechercheAdmin, self).has_change_permission(request, obj, groupe_chercheur=False)
 238
 239 class PublicationAdmin(admin.ModelAdmin):
 240     search_fields = ('auteurs', 'titre', 'revue', 'editeur')
 241
 242
 243 admin.site.register(Chercheur, ChercheurAdmin)
 244 admin.site.register(Publication, PublicationAdmin)
 245 admin.site.register(GroupeChercheur, GroupeChercheurAdmin)
 246 admin.site.register(DomaineRecherche, DomaineRechercheAdmin)
 247 admin.site.register(AdhesionCommunaute, AdhesionCommunauteAdmin)
 248 admin.site.register(AdhesionDomaineRecherche, AdhesionDomaineRechercheAdmin)
 249