plateforme-tfjm2/registration/admin.py

# Copyright (C) 2020 by Animath
# SPDX-License-Identifier: GPL-3.0-or-later

from django.contrib import admin
from django.contrib.admin import ModelAdmin
from django.utils.translation import gettext_lazy as _
from polymorphic.admin import PolymorphicChildModelAdmin, PolymorphicChildModelFilter, PolymorphicParentModelAdmin

from .models import CoachRegistration, Payment, ParticipantRegistration, Registration, \
    StudentRegistration, VolunteerRegistration


@admin.register(Registration)
class RegistrationAdmin(PolymorphicParentModelAdmin):
    child_models = (StudentRegistration, CoachRegistration, VolunteerRegistration,)
    list_display = ('user', 'first_name', 'last_name', 'type', 'email_confirmed',)
    list_filter = (PolymorphicChildModelFilter, 'email_confirmed',)
    search_fields = ('user__first_name', 'user__last_name', 'user__email',)
    polymorphic_list = True

    @admin.display(description=_('first name'), ordering='user__first_name')
    def first_name(self, record):
        return record.user.first_name

    @admin.display(description=_('last name'), ordering='user__last_name')
    def last_name(self, record):
        return record.user.last_name

@admin.register(ParticipantRegistration)
class ParticipantRegistrationAdmin(PolymorphicChildModelAdmin):
    list_display = ('user', 'first_name', 'last_name', 'type', 'team', 'email_confirmed',)
    list_filter = ('email_confirmed',)
    search_fields = ('user__first_name', 'user__last_name', 'user__email',)
    autocomplete_fields = ('user', 'team',)

    @admin.display(description=_('first name'), ordering='user__first_name')
    def first_name(self, record):
        return record.user.first_name

    @admin.display(description=_('last name'), ordering='user__last_name')
    def last_name(self, record):
        return record.user.last_name


@admin.register(StudentRegistration)
class StudentRegistrationAdmin(PolymorphicChildModelAdmin):
    list_display = ('user', 'first_name', 'last_name', 'team', 'email_confirmed',)
    list_filter = ('email_confirmed',)
    search_fields = ('user__first_name', 'user__last_name', 'user__email',)
    autocomplete_fields = ('user', 'team',)

    @admin.display(description=_('first name'), ordering='user__first_name')
    def first_name(self, record):
        return record.user.first_name

    @admin.display(description=_('last name'), ordering='user__last_name')
    def last_name(self, record):
        return record.user.last_name


@admin.register(CoachRegistration)
class CoachRegistrationAdmin(PolymorphicChildModelAdmin):
    list_display = ('user', 'first_name', 'last_name', 'team', 'email_confirmed',)
    list_filter = ('email_confirmed',)
    search_fields = ('user__first_name', 'user__last_name', 'user__email',)
    autocomplete_fields = ('user', 'team',)

    @admin.display(description=_('first name'), ordering='user__first_name')
    def first_name(self, record):
        return record.user.first_name

    @admin.display(description=_('last name'), ordering='user__last_name')
    def last_name(self, record):
        return record.user.last_name


@admin.register(VolunteerRegistration)
class VolunteerRegistrationAdmin(PolymorphicChildModelAdmin):
    list_display = ('user', 'first_name', 'last_name', 'tournaments', 'email_confirmed',)
    list_filter = ('organized_tournaments', 'email_confirmed',)
    search_fields = ('user__first_name', 'user__last_name', 'user__email',)
    autocomplete_fields = ('user',)

    @admin.display(description=_('first name'), ordering='user__first_name')
    def first_name(self, record):
        return record.user.first_name

    @admin.display(description=_('last name'), ordering='user__last_name')
    def last_name(self, record):
        return record.user.last_name

    @admin.display(description=_("tournaments"))
    def tournaments(self, record):
        return ', '.join(tr.name for tr in record.interesting_tournaments) or None


@admin.register(Payment)
class PaymentAdmin(ModelAdmin):
    list_display = ('registration', 'registration_type', 'type', 'valid', )
    search_fields = ('registration__user__last_name', 'registration__user__first_name', 'registration__user__email',)
    list_filter = ('registration__team__participation__valid', 'type', 'type', 'valid',)
    autocomplete_fields = ('registration',)

    @admin.display(description=_('registration type'), ordering='registration__polymorphic_ctype')
    def registration_type(self, record: Payment):
        return record.registration.get_real_instance().type