98 lines
3.6 KiB
Python
98 lines
3.6 KiB
Python
"""
|
|
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.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)
|