# -*- coding: utf-8 -*- from __future__ import division from __future__ import print_function from __future__ import unicode_literals from django import forms from django.contrib import admin from gestioncof.models import SurveyQuestionAnswer, SurveyQuestion, \ CofProfile, EventOption, EventOptionChoice, Event, Club, CustomMail, \ Survey, EventCommentField, EventRegistration from gestioncof.petits_cours_models import PetitCoursDemande, \ PetitCoursSubject, PetitCoursAbility, PetitCoursAttribution, \ PetitCoursAttributionCounter from django.contrib.auth.models import User from django.contrib.auth.admin import UserAdmin from django.core.urlresolvers import reverse from django.utils.safestring import mark_safe import django.utils.six as six import autocomplete_light def add_link_field(target_model='', field='', link_text=six.text_type, desc_text=six.text_type): def add_link(cls): reverse_name = target_model or cls.model.__name__.lower() def link(self, instance): app_name = instance._meta.app_label reverse_path = "admin:%s_%s_change" % (app_name, reverse_name) link_obj = getattr(instance, field, None) or instance if not link_obj.id: return "" url = reverse(reverse_path, args=(link_obj.id,)) return mark_safe("<a href='%s'>%s</a>" % (url, link_text(link_obj))) link.allow_tags = True link.short_description = desc_text(reverse_name + ' link') cls.link = link cls.readonly_fields =\ list(getattr(cls, 'readonly_fields', [])) + ['link'] return cls return add_link class SurveyQuestionAnswerInline(admin.TabularInline): model = SurveyQuestionAnswer @add_link_field(desc_text=lambda x: "Réponses", link_text=lambda x: "Éditer les réponses") class SurveyQuestionInline(admin.TabularInline): model = SurveyQuestion class SurveyQuestionAdmin(admin.ModelAdmin): search_fields = ('survey__title', 'answer') inlines = [ SurveyQuestionAnswerInline, ] class SurveyAdmin(admin.ModelAdmin): search_fields = ('title', 'details') inlines = [ SurveyQuestionInline, ] class EventOptionChoiceInline(admin.TabularInline): model = EventOptionChoice @add_link_field(desc_text=lambda x: "Choix", link_text=lambda x: "Éditer les choix") class EventOptionInline(admin.TabularInline): model = EventOption class EventCommentFieldInline(admin.TabularInline): model = EventCommentField class EventOptionAdmin(admin.ModelAdmin): search_fields = ('event__title', 'name') inlines = [ EventOptionChoiceInline, ] class EventAdmin(admin.ModelAdmin): search_fields = ('title', 'location', 'description') inlines = [ EventOptionInline, EventCommentFieldInline, ] class CofProfileInline(admin.StackedInline): model = CofProfile inline_classes = ("collapse open",) class FkeyLookup(object): def __init__(self, fkeydecl, short_description=None, admin_order_field=None): self.fk, fkattrs = fkeydecl.split('__', 1) self.fkattrs = fkattrs.split('__') self.short_description = short_description or self.fkattrs[-1] self.admin_order_field = admin_order_field or fkeydecl def __get__(self, obj, klass): if obj is None: """ hack required to make Django validate (if obj is None, then we're a class, and classes are callable <wink>) """ return self item = getattr(obj, self.fk) for attr in self.fkattrs: item = getattr(item, attr) return item def ProfileInfo(field, short_description, boolean=False): def getter(self): try: return getattr(self.profile, field) except CofProfile.DoesNotExist: return "" getter.short_description = short_description getter.boolean = boolean return getter User.profile_login_clipper = FkeyLookup("profile__login_clipper", "Login clipper") User.profile_num = FkeyLookup("profile__num", "Numéro") User.profile_phone = ProfileInfo("phone", "Téléphone") User.profile_occupation = ProfileInfo("occupation", "Occupation") User.profile_departement = ProfileInfo("departement", "Departement") User.profile_mailing_cof = ProfileInfo("mailing_cof", "ML COF", True) User.profile_mailing_bda = ProfileInfo("mailing_bda", "ML BDA", True) User.profile_mailing_bda_revente = ProfileInfo("mailing_bda_revente", "ML BDA-R", True) class UserProfileAdmin(UserAdmin): def is_buro(self, obj): try: return obj.profile.is_buro except CofProfile.DoesNotExist: return False is_buro.short_description = 'Membre du Buro' is_buro.boolean = True def is_cof(self, obj): try: return obj.profile.is_cof except CofProfile.DoesNotExist: return False is_cof.short_description = 'Membre du COF' is_cof.boolean = True list_display = ('profile_num',) + UserAdmin.list_display \ + ('profile_login_clipper', 'profile_phone', 'profile_occupation', 'profile_mailing_cof', 'profile_mailing_bda', 'profile_mailing_bda_revente', 'is_cof', 'is_buro', ) list_display_links = ('username', 'email', 'first_name', 'last_name') list_filter = UserAdmin.list_filter \ + ('profile__is_cof', 'profile__is_buro', 'profile__mailing_cof', 'profile__mailing_bda') search_fields = UserAdmin.search_fields + ('profile__phone',) inlines = [ CofProfileInline, ] # FIXME: This is absolutely horrible. def user_unicode(self): if self.first_name and self.last_name: return "%s %s (%s)" % (self.first_name, self.last_name, self.username) else: return self.username if six.PY2: User.__unicode__ = user_unicode else: User.__str__ = user_unicode class EventRegistrationAdmin(admin.ModelAdmin): form = autocomplete_light.modelform_factory(EventRegistration, exclude=[]) list_display = ('__unicode__' if six.PY2 else '__str__', 'event', 'user', 'paid') list_filter = ('paid',) search_fields = ('user__username', 'user__first_name', 'user__last_name', 'user__email', 'event__title') class PetitCoursAbilityAdmin(admin.ModelAdmin): list_display = ('user', 'matiere', 'niveau', 'agrege') search_fields = ('user__username', 'user__first_name', 'user__last_name', 'user__email', 'matiere__name', 'niveau') list_filter = ('matiere', 'niveau', 'agrege') class PetitCoursAttributionAdmin(admin.ModelAdmin): list_display = ('user', 'demande', 'matiere', 'rank', ) search_fields = ('user__username', 'matiere__name') class PetitCoursAttributionCounterAdmin(admin.ModelAdmin): list_display = ('user', 'matiere', 'count', ) list_filter = ('matiere',) search_fields = ('user__username', 'user__first_name', 'user__last_name', 'user__email', 'matiere__name') actions = ['reset', ] actions_on_bottom = True def reset(self, request, queryset): queryset.update(count=0) reset.short_description = "Remise à zéro du compteur" class PetitCoursDemandeAdmin(admin.ModelAdmin): list_display = ('name', 'email', 'agrege_requis', 'niveau', 'created', 'traitee', 'processed') list_filter = ('traitee', 'niveau') search_fields = ('name', 'email', 'phone', 'lieu', 'remarques') class CustomMailAdmin(admin.ModelAdmin): search_fields = ('shortname', 'title') class ClubAdminForm(forms.ModelForm): def clean(self): cleaned_data = super(ClubAdminForm, self).clean() respos = cleaned_data.get('respos') members = cleaned_data.get('membres') for respo in respos.all(): if respo not in members: raise forms.ValidationError( "Erreur : le respo %s n'est pas membre du club." % respo.get_full_name()) return cleaned_data class ClubAdmin(admin.ModelAdmin): list_display = ['name'] form = ClubAdminForm admin.site.register(Survey, SurveyAdmin) admin.site.register(SurveyQuestion, SurveyQuestionAdmin) admin.site.register(Event, EventAdmin) admin.site.register(EventOption, EventOptionAdmin) admin.site.unregister(User) admin.site.register(User, UserProfileAdmin) admin.site.register(CofProfile) admin.site.register(Club, ClubAdmin) admin.site.register(CustomMail) admin.site.register(PetitCoursSubject) admin.site.register(PetitCoursAbility, PetitCoursAbilityAdmin) admin.site.register(PetitCoursAttribution, PetitCoursAttributionAdmin) admin.site.register(PetitCoursAttributionCounter, PetitCoursAttributionCounterAdmin) admin.site.register(PetitCoursDemande, PetitCoursDemandeAdmin) admin.site.register(EventRegistration, EventRegistrationAdmin)