ernestophone.ens.fr/gestion/admin.py

"""
Administration minimaliste du site de l'Ernestophone.
Les chefs peuvent modifier les utilisateurs et les événements.
Les super-utilisateurs peuvent en plus ajouter ou supprimer ces objets et gérer
les permissions.
"""

from django.contrib import admin
from django.contrib.auth.models import User, Group, Permission
from django.contrib.auth.admin import UserAdmin

from gestion.models import ErnestoUser
from calendrier.models import Event


class UserProfileInline(admin.StackedInline):
    """Affichage du profil ernestophone dans la vue UserAdmin"""
    model = ErnestoUser


def ProfileInfo(field, short_description, boolean=False):
    def getter(self):
        try:
            return getattr(self.profile, field)
        except ErnestoUser.DoesNotExist:
            return ""
    getter.short_description = short_description
    getter.boolean = boolean
    return getter

User.profile_phone = ProfileInfo("phone", "Telephone")
User.profile_instru = ProfileInfo("instru", "Instrument joué")
User.profile_is_ern = ProfileInfo("is_ernesto", "Ernestophoniste")
User.profile_is_chef = ProfileInfo("is_chef", "Chef Fanfare")
User.profile_get_mails = ProfileInfo("mails", "Recevoir les mails")


class UserProfileAdmin(UserAdmin):
    """
    Management des utilisateurs de l'ernestosite.
    Lors de la sauvegarde, les chefs se voient octroyées les permissions
    d'édition des comptes et événements mais il ne peuvent pas en créer via
    l'interface admin.
    """
    list_display = ['username', 'first_name', 'last_name', 'email',
                    'profile_phone', 'profile_instru', 'profile_is_ern',
                    'profile_is_chef']
    list_display_links = ['username', 'email', 'first_name', 'last_name']
    list_filter = ['profile__instru']
    ordering = ['username']
    search_fields = ['username', 'first_name', 'last_name', 'profile__phone',
                     'profile__instru']
    inlines = [UserProfileInline]

    staff_fieldsets = [
        ('Général', {'fields': ['username', 'password', 'first_name',
                                'last_name']}),
        ('Permissions', {'fields': ['is_active']})
    ]

    def get_fieldsets(self, request, user=None):
        """
        Les super utilisateurs ont toutes les infos.
        Les autres (e.g. les chefs) ne peuvent modifier qu'un nombre restreints
        de choses.
        """
        if not request.user.is_superuser:
            return self.staff_fieldsets
        return super(UserProfileAdmin, self).get_fieldsets(request, user)

    def save_model(self, request, user, form, change):
        """
        Ajoute le chef au group Chef.
        Le groupe est créé si nécessaire.
        """
        chef_group, created = Group.objects.get_or_create(name="Chef")
        if created:
            # Si le groupe vient d'être créé :
            # On associe les bonnes permissions au groupe Chef
            perms = [('change_user', 'auth', 'user'),
                     ('change_event', 'calendrier', 'event'),
                     ('change_ernestouser', 'gestion', 'ernestouser')]
            for nat_key in perms:
                perm = Permission.objects.get_by_natural_key(*nat_key)
                chef_group.permissions.add(perm)
            # On met tous les chef dans le groupe
            chef_group.user_set = User.objects.filter(profile__is_chef=True)
        # Les chefs sont dans le groupe Chef
        if user.profile.is_chef:
            print("J'aime la choucroute")
            user.is_staff = True
            user.groups.add(chef_group)
        user.save()


admin.site.unregister(User)
admin.site.register(User, UserProfileAdmin)
admin.site.register(Event)