from django.contrib import admin
from django.db.models import Sum

from .models import PaymentRequest, PNA, Fake


class PNAAdmin(admin.ModelAdmin):
    list_display = ['pna_status_code', 'pna_status_message', 'jajali_datetime', 'payment_request']
    list_filter = ['pna_status']
    search_fields = ['pna_status_code', 'pna_status_message']


class PaymentRequestAdmin(admin.ModelAdmin):
    list_display = ['state', 'uuid', 'splited_amount', 'user_id', 'jajali_datetime', 'application', 'gateway']
    search_fields = ['application__uuid','application__client_id', 'amount', 'user__uuid']
    list_filter = ['application', 'state', 'gateway']

    def changelist_view(self, request, extra_context=None):
        extra_context = extra_context or {}

        cl = self.get_changelist_instance(request)
        queryset = cl.result_list
        totals = queryset.aggregate(
            total_amount=Sum('amount'),
        )
        extra_context['totals'] = totals
        return super().changelist_view(request, extra_context=extra_context)


class FakeAdmin(admin.ModelAdmin):
    list_display = [ 'created_at']

admin.site.register(PaymentRequest, PaymentRequestAdmin)
admin.site.register(PNA, PNAAdmin)
admin.site.register(Fake, FakeAdmin)