djadhere/banking/admin.py

from django.contrib import admin
from django.contrib.contenttypes.admin import GenericTabularInline
from django.contrib.contenttypes.models import ContentType
from django.db.models import Q

from services.models import ServiceType
from .models import Payment


class PaymentMixin:
    def get_fields(self, request, obj=None):
        fields = ('amount', 'period', 'payment_method', 'date',)
        if request.user.is_superuser or request.user.has_perm('banking.validate_payment'):
            fields += ('validated',)
        return fields

    def get_queryset(self, request):
        qs = super().get_queryset(request)
        if request.user.is_superuser or request.user.has_perm('banking.validate_payment'):
            return qs
        # Show only adherent contribution and payment related to a service for which the user is in managment group.
        user_ctype = ContentType.objects.get(app_label='adhesions', model='adherent')
        return qs.filter(Q(reason_type=user_ctype) | Q(service__service_type__group__in=request.user.groups.all()))


class PaymentInlineMixin(PaymentMixin, GenericTabularInline):
    model = Payment
    ct_field = 'reason_type'
    ct_fk_field = 'reason_id'


class ValidatedPaymentInline(PaymentInlineMixin):
    extra = 0
    verbose_name_plural = 'Paiements validés'

    def get_readonly_fields(self, request, obj=None):
        return self.get_fields(request, obj)

    def has_add_permission(self, request):
        return False

    def has_delete_permission(self, request, obj=None):
        return request.user.has_perm('banking.validate_payment')

    def get_queryset(self, request):
        return super().get_queryset(request).filter(validated=True)


class PendingOrNewPaymentInline(PaymentInlineMixin):
    extra = 1
    verbose_name_plural = 'Paiements en attente de validation et nouveaux paiements'

    def get_queryset(self, request):
        return super().get_queryset(request).filter(validated=False)


class PaymentTypeFilter(admin.SimpleListFilter):
    title = 'type de paiement'
    parameter_name = 'type'

    def lookups(self, request, model_admin):
        choices = [
            ('membership', 'Cotisation'),
            ('service', 'Service'),
        ]
        service_types = ServiceType.objects.all()
        if not request.user.is_superuser:
            service_types = service_types.filter(group__in=request.user.groups.all())
        for stype in service_types:
            choices.append((stype.pk, 'Service (%s)' % stype.name))
        return choices

    def queryset(self, request, queryset):
        if self.value() == 'membership':
            return queryset.filter(reason_type__app_label='adhesions',
                                   reason_type__model='adherent')
        if self.value() == 'service':
            return queryset.filter(reason_type__app_label='services',
                                   reason_type__model='service')
        try:
            service_type = ServiceType.objects.get(pk=int(self.value()))
        except (ValueError, TypeError, ServiceType.DoesNotExist,):
            return queryset
        else:
            return queryset.filter(service__service_type=service_type)


class PaymentAdmin(PaymentMixin, admin.ModelAdmin):
    list_display = ('get_adherent', 'payment_type_verbose', 'amount',
                    'period', 'payment_method', 'validated',)
    list_filter = (PaymentTypeFilter, 'payment_method', 'validated',)

    def has_change_permission(self, request, obj=None):
        if obj and not request.user.is_superuser:
            return not obj.validated
        return True

    def has_add_permission(self, request):
        return False


admin.site.register(Payment, PaymentAdmin)