195 lines
7 KiB
Python
195 lines
7 KiB
Python
from django import forms
|
|
from django.contrib import admin
|
|
from django.db import models
|
|
|
|
from stages.models import (
|
|
Teacher, Student, Section, Level, Klass, Referent, Corporation,
|
|
CorpContact, Domain, Period, Availability, Training, Course,
|
|
)
|
|
|
|
|
|
class ArchivedListFilter(admin.BooleanFieldListFilter):
|
|
"""
|
|
Default filter that shows by default unarchived elements.
|
|
"""
|
|
def __init__(self, request, params, *args, **kwargs):
|
|
super().__init__(request, params, *args, **kwargs)
|
|
if self.lookup_val is None:
|
|
self.lookup_val = '0'
|
|
|
|
def choices(self, cl):
|
|
# Removing the "all" choice
|
|
return list(super().choices(cl))[1:]
|
|
|
|
def queryset(self, request, queryset):
|
|
if not self.used_parameters:
|
|
self.used_parameters[self.lookup_kwarg] = '0'
|
|
return super().queryset(request, queryset)
|
|
|
|
|
|
class KlassAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'section')
|
|
ordering = ('name',)
|
|
list_filter = ('section', 'level',)
|
|
|
|
|
|
class TeacherAdmin(admin.ModelAdmin):
|
|
list_display = ('__str__', 'abrev', 'email')
|
|
|
|
|
|
class StudentAdmin(admin.ModelAdmin):
|
|
list_display = ('__str__', 'pcode', 'city', 'klass', 'archived')
|
|
ordering = ('last_name', 'first_name')
|
|
list_filter = (('archived', ArchivedListFilter), 'klass')
|
|
search_fields = ('last_name', 'first_name', 'pcode', 'city', 'klass__name')
|
|
fields = (('last_name', 'first_name', 'ext_id'), ('street', 'pcode', 'city', 'district'),
|
|
('email', 'tel', 'mobile'), ('avs', 'birth_date'),
|
|
('dispense_ecg', 'dispense_eps', 'soutien_dys'), ('klass', 'archived'),
|
|
('corporation', 'instructor'))
|
|
readonly_fields = ('ext_id',)
|
|
actions = ['archive']
|
|
|
|
def archive(self, request, queryset):
|
|
queryset.update(archived=True)
|
|
archive.short_description = "Marquer les étudiants sélectionnés comme archivés"
|
|
|
|
'''def get_readonly_fields(self, request, obj=None):
|
|
if 'edit' not in request.GET:
|
|
return self.fields
|
|
else:
|
|
return self.readonly_fields
|
|
'''
|
|
|
|
|
|
class ReferentAdmin(admin.ModelAdmin):
|
|
list_display = ('__str__', 'abrev', 'email')
|
|
list_filter = (('archived', ArchivedListFilter),)
|
|
|
|
|
|
class CorpContactAdmin(admin.ModelAdmin):
|
|
list_display = ('__str__', 'corporation', 'role')
|
|
list_filter = (('archived', ArchivedListFilter),)
|
|
ordering = ('last_name', 'first_name')
|
|
search_fields = ('last_name', 'first_name', 'role')
|
|
fields = (('corporation',), ('title', 'last_name', 'first_name'),
|
|
('sections', 'is_main', 'always_cc', 'archived'),
|
|
('role', 'ext_id'), ('tel', 'email'))
|
|
formfield_overrides = {
|
|
models.ManyToManyField: {'widget': forms.CheckboxSelectMultiple},
|
|
}
|
|
|
|
def get_form(self, *args, **kwargs):
|
|
form = super().get_form(*args, **kwargs)
|
|
form.base_fields['sections'].widget.can_add_related = False
|
|
return form
|
|
|
|
|
|
class ContactInline(admin.StackedInline):
|
|
model = CorpContact
|
|
fields = (('title', 'last_name', 'first_name'),
|
|
('sections', 'is_main', 'always_cc', 'archived'),
|
|
('role', 'tel', 'email'))
|
|
extra = 1
|
|
formfield_overrides = {
|
|
models.ManyToManyField: {'widget': forms.CheckboxSelectMultiple},
|
|
}
|
|
|
|
|
|
class CorporationAdmin(admin.ModelAdmin):
|
|
list_display = ('name', 'short_name', 'pcode', 'city')
|
|
list_editable = ('short_name',) # Temporarily?
|
|
list_filter = (('archived', ArchivedListFilter),)
|
|
search_fields = ('name', 'pcode', 'city')
|
|
ordering = ('name',)
|
|
fields = (('name', 'short_name'), 'parent', ('sector', 'typ', 'ext_id'),
|
|
'street', ('pcode', 'city'), ('tel', 'email'), 'web', 'archived')
|
|
inlines = [ContactInline]
|
|
|
|
|
|
class AvailabilityAdminForm(forms.ModelForm):
|
|
"""
|
|
Custom avail form to create several availabilities at once when inlined in
|
|
the PeriodAdmin interface
|
|
"""
|
|
num_avail = forms.IntegerField(label="Nombre de places", initial=1, required=False)
|
|
|
|
class Media:
|
|
js = ('js/avail_form.js',)
|
|
|
|
class Meta:
|
|
model = Availability
|
|
fields = '__all__'
|
|
widgets = {
|
|
'num_avail': forms.TextInput(attrs={'size': 3}),
|
|
}
|
|
|
|
def __init__(self, data=None, files=None, **kwargs):
|
|
super(AvailabilityAdminForm, self).__init__(data=data, files=files, **kwargs)
|
|
if self.instance.pk is not None:
|
|
# Hide num_avail on existing instances
|
|
self.fields['num_avail'].widget = forms.HiddenInput()
|
|
# Limit CorpContact objects to contacts of chosen corporation
|
|
if data is None and self.instance.corporation_id:
|
|
self.fields['contact'].queryset = self.instance.corporation.corpcontact_set
|
|
|
|
def save(self, **kwargs):
|
|
instance = super(AvailabilityAdminForm, self).save(**kwargs)
|
|
# Create supplementary availabilities depending on num_avail
|
|
num_avail = self.cleaned_data.get('num_avail', 1) or 1
|
|
for i in range(1, num_avail):
|
|
Availability.objects.create(
|
|
corporation=instance.corporation,
|
|
period=instance.period,
|
|
domain=instance.domain,
|
|
contact=instance.contact,
|
|
comment=instance.comment)
|
|
return instance
|
|
|
|
class AvailabilityInline(admin.StackedInline):
|
|
model = Availability
|
|
form = AvailabilityAdminForm
|
|
ordering = ('corporation__name',)
|
|
extra = 1
|
|
formfield_overrides = {
|
|
models.TextField: {'widget': forms.Textarea(attrs={'rows':2, 'cols':40})},
|
|
}
|
|
|
|
|
|
class PeriodAdmin(admin.ModelAdmin):
|
|
list_display = ('title', 'dates', 'section', 'level')
|
|
list_filter = ('section', 'level')
|
|
inlines = [AvailabilityInline]
|
|
|
|
|
|
class AvailabilityAdmin(admin.ModelAdmin):
|
|
list_display = ('corporation', 'period', 'domain')
|
|
list_filter = ('period',)
|
|
fields = (('corporation', 'period'), 'domain', 'contact', 'priority', 'comment')
|
|
form = AvailabilityAdminForm
|
|
|
|
def formfield_for_foreignkey(self, db_field, request, **kwargs):
|
|
if db_field.name == "corporation":
|
|
kwargs["queryset"] = Corporation.objects.filter(archived=False).order_by('name')
|
|
if db_field.name == "contact":
|
|
kwargs["queryset"] = CorpContact.objects.filter(archived=False)
|
|
return super(AvailabilityAdmin, self).formfield_for_foreignkey(db_field, request, **kwargs)
|
|
|
|
|
|
class TrainingAdmin(admin.ModelAdmin):
|
|
search_fields = ('student__first_name', 'student__last_name', 'availability__corporation__name')
|
|
raw_id_fields = ('availability',)
|
|
|
|
|
|
admin.site.register(Section)
|
|
admin.site.register(Level)
|
|
admin.site.register(Klass, KlassAdmin)
|
|
admin.site.register(Student, StudentAdmin)
|
|
admin.site.register(Referent, ReferentAdmin)
|
|
admin.site.register(Teacher, TeacherAdmin)
|
|
admin.site.register(Course)
|
|
admin.site.register(Corporation, CorporationAdmin)
|
|
admin.site.register(CorpContact, CorpContactAdmin)
|
|
admin.site.register(Domain)
|
|
admin.site.register(Period, PeriodAdmin)
|
|
admin.site.register(Availability, AvailabilityAdmin)
|
|
admin.site.register(Training, TrainingAdmin)
|