kpsul/bda/admin.py

from datetime import timedelta

from custommail.shortcuts import send_mass_custom_mail
from dal.autocomplete import ModelSelect2
from django import forms
from django.contrib import admin
from django.db.models import Count, Sum
from django.template.defaultfilters import pluralize
from django.utils import timezone

from bda.models import (
    Attribution,
    CategorieSpectacle,
    ChoixSpectacle,
    Participant,
    Quote,
    Salle,
    Spectacle,
    SpectacleRevente,
    Tirage,
)


class ReadOnlyMixin(object):
    readonly_fields_update = ()

    def get_readonly_fields(self, request, obj=None):
        readonly_fields = super().get_readonly_fields(request, obj)
        if obj is None:
            return readonly_fields
        else:
            return readonly_fields + self.readonly_fields_update


class ChoixSpectacleAdminForm(forms.ModelForm):
    class Meta:
        widgets = {
            "participant": ModelSelect2(url="bda-participant-autocomplete"),
            "spectacle": ModelSelect2(url="bda-spectacle-autocomplete"),
        }


class ChoixSpectacleInline(admin.TabularInline):
    model = ChoixSpectacle
    form = ChoixSpectacleAdminForm
    sortable_field_name = "priority"


class AttributionTabularAdminForm(forms.ModelForm):
    listing = None

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        spectacles = Spectacle.objects.select_related("location")
        if self.listing is not None:
            spectacles = spectacles.filter(listing=self.listing)
        self.fields["spectacle"].queryset = spectacles


class WithoutListingAttributionTabularAdminForm(AttributionTabularAdminForm):
    listing = False


class WithListingAttributionTabularAdminForm(AttributionTabularAdminForm):
    listing = True


class AttributionInline(admin.TabularInline):
    model = Attribution
    extra = 0
    listing = None

    def get_queryset(self, request):
        qs = super().get_queryset(request)
        if self.listing is not None:
            qs = qs.filter(spectacle__listing=self.listing)
        return qs


class WithListingAttributionInline(AttributionInline):
    exclude = ("given",)
    form = WithListingAttributionTabularAdminForm
    listing = True


class WithoutListingAttributionInline(AttributionInline):
    form = WithoutListingAttributionTabularAdminForm
    listing = False


class ParticipantAdminForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields["choicesrevente"].queryset = Spectacle.objects.select_related(
            "location"
        )


class ParticipantAdmin(ReadOnlyMixin, admin.ModelAdmin):
    inlines = [WithListingAttributionInline, WithoutListingAttributionInline]

    def get_queryset(self, request):
        return Participant.objects.annotate(
            nb_places=Count("attributions"), total=Sum("attributions__price")
        )

    def nb_places(self, obj):
        return obj.nb_places

    nb_places.admin_order_field = "nb_places"
    nb_places.short_description = "Nombre de places"

    def total(self, obj):
        tot = obj.total
        if tot:
            return "%.02f €" % tot
        else:
            return "0 €"

    total.admin_order_field = "total"
    total.short_description = "Total à payer"
    list_display = ("user", "nb_places", "total", "paid", "paymenttype", "tirage")
    list_filter = ("paid", "tirage")
    search_fields = ("user__username", "user__first_name", "user__last_name")
    actions = ["send_attribs"]
    actions_on_bottom = True
    list_per_page = 400
    readonly_fields = ("total",)
    readonly_fields_update = ("user", "tirage")
    form = ParticipantAdminForm

    def send_attribs(self, request, queryset):
        datatuple = []
        for member in queryset.all():
            attribs = member.attributions.all()
            context = {"member": member.user}
            shortname = ""
            if len(attribs) == 0:
                shortname = "bda-attributions-decus"
            else:
                shortname = "bda-attributions"
                context["places"] = attribs
            print(context)
            datatuple.append((shortname, context, "bda@ens.fr", [member.user.email]))
        send_mass_custom_mail(datatuple)
        count = len(queryset.all())
        if count == 1:
            message_bit = "1 membre a"
            plural = ""
        else:
            message_bit = "%d membres ont" % count
            plural = "s"
        self.message_user(
            request, "%s été informé%s avec succès." % (message_bit, plural)
        )

    send_attribs.short_description = "Envoyer les résultats par mail"


class AttributionAdminForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        if "spectacle" in self.fields:
            self.fields["spectacle"].queryset = Spectacle.objects.select_related(
                "location"
            )
        if "participant" in self.fields:
            self.fields["participant"].queryset = Participant.objects.select_related(
                "user", "tirage"
            )

    def clean(self):
        cleaned_data = super().clean()
        participant = cleaned_data.get("participant")
        spectacle = cleaned_data.get("spectacle")
        if participant and spectacle:
            if participant.tirage != spectacle.tirage:
                raise forms.ValidationError(
                    "Erreur : le participant et le spectacle n'appartiennent"
                    "pas au même tirage"
                )
        return cleaned_data


class AttributionAdmin(ReadOnlyMixin, admin.ModelAdmin):
    def paid(self, obj):
        return obj.participant.paid

    paid.short_description = "A payé"
    paid.boolean = True
    list_display = ("id", "spectacle", "participant", "given", "paid")
    search_fields = (
        "spectacle__title",
        "participant__user__username",
        "participant__user__first_name",
        "participant__user__last_name",
    )
    form = AttributionAdminForm
    readonly_fields_update = ("spectacle", "participant")


class ChoixSpectacleAdmin(admin.ModelAdmin):
    form = ChoixSpectacleAdminForm

    def tirage(self, obj):
        return obj.participant.tirage

    list_display = ("participant", "tirage", "spectacle", "priority", "double_choice")
    list_filter = ("double_choice", "participant__tirage")
    search_fields = (
        "participant__user__username",
        "participant__user__first_name",
        "participant__user__last_name",
        "spectacle__title",
    )


class QuoteInline(admin.TabularInline):
    model = Quote


class SpectacleAdmin(admin.ModelAdmin):
    inlines = [QuoteInline]
    model = Spectacle
    list_display = ("title", "date", "tirage", "location", "slots", "price", "listing")
    list_filter = ("location", "tirage")
    search_fields = ("title", "location__name")
    readonly_fields = ("rappel_sent",)


class TirageAdmin(admin.ModelAdmin):
    model = Tirage
    list_display = ("title", "ouverture", "fermeture", "active", "enable_do_tirage")
    readonly_fields = ("tokens",)
    list_filter = ("active",)
    search_fields = ("title",)


class SalleAdmin(admin.ModelAdmin):
    model = Salle
    search_fields = ("name", "address")


class SpectacleReventeAdminForm(forms.ModelForm):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields["confirmed_entry"].queryset = Participant.objects.select_related(
            "user", "tirage"
        )
        self.fields["seller"].queryset = Participant.objects.select_related(
            "user", "tirage"
        )
        self.fields["soldTo"].queryset = Participant.objects.select_related(
            "user", "tirage"
        )


class SpectacleReventeAdmin(admin.ModelAdmin):
    """
    Administration des reventes de spectacles
    """

    model = SpectacleRevente

    def spectacle(self, obj):
        """
        Raccourci vers le spectacle associé à la revente.
        """
        return obj.attribution.spectacle

    list_display = ("spectacle", "seller", "date", "soldTo")
    raw_id_fields = ("attribution",)
    readonly_fields = ("date_tirage",)
    search_fields = [
        "attribution__spectacle__title",
        "seller__user__username",
        "seller__user__first_name",
        "seller__user__last_name",
    ]

    actions = ["transfer", "reinit"]
    actions_on_bottom = True
    form = SpectacleReventeAdminForm

    def transfer(self, request, queryset):
        """
        Effectue le transfert des reventes pour lesquels on connaît l'acheteur.
        """
        reventes = queryset.exclude(soldTo__isnull=True).all()
        count = reventes.count()
        for revente in reventes:
            attrib = revente.attribution
            attrib.participant = revente.soldTo
            attrib.save()
        self.message_user(
            request,
            "%d attribution%s %s été transférée%s avec succès."
            % (count, pluralize(count), pluralize(count, "a,ont"), pluralize(count)),
        )

    transfer.short_description = "Transférer les reventes sélectionnées"

    def reinit(self, request, queryset):
        """
        Réinitialise les reventes.
        """
        count = queryset.count()
        for revente in queryset.filter(
            attribution__spectacle__date__gte=timezone.now()
        ):
            revente.reset(new_date=timezone.now() - timedelta(hours=1))
        self.message_user(
            request,
            "%d attribution%s %s été réinitialisée%s avec succès."
            % (count, pluralize(count), pluralize(count, "a,ont"), pluralize(count)),
        )

    reinit.short_description = "Réinitialiser les reventes sélectionnées"


admin.site.register(CategorieSpectacle)
admin.site.register(Spectacle, SpectacleAdmin)
admin.site.register(Salle, SalleAdmin)
admin.site.register(Participant, ParticipantAdmin)
admin.site.register(Attribution, AttributionAdmin)
admin.site.register(ChoixSpectacle, ChoixSpectacleAdmin)
admin.site.register(Tirage, TirageAdmin)
admin.site.register(SpectacleRevente, SpectacleReventeAdmin)