src/pz/admin.py

   1 from django.contrib import admin
   2 from django.contrib.admin.filters import FieldListFilter, SimpleListFilter
   3 from django.contrib import messages
   4 from django.db.models import Q
   5 from django.shortcuts import get_object_or_404, redirect
   6 from django.urls import path, reverse
   7 from django.utils.safestring import mark_safe
   8 from django.utils.translation import gettext_lazy as _
   9 from django.utils.timezone import now
  10 from fnpdjango.actions import export_as_csv_action
  11 from . import bank
  12 from . import models
  13
  14
  15 admin.site.register(models.Fundraiser)
  16 admin.site.register(models.Campaign)
  17
  18
  19 # Backport from Django 3.1.
  20 class EmptyFieldListFilter(FieldListFilter):
  21     def __init__(self, field, request, params, model, model_admin, field_path):
  22         self.lookup_kwarg = '%s__isempty' % field_path
  23         self.lookup_val = params.get(self.lookup_kwarg)
  24         super().__init__(field, request, params, model, model_admin, field_path)
  25
  26     def queryset(self, request, queryset):
  27         if self.lookup_kwarg not in self.used_parameters:
  28             return queryset
  29         if self.lookup_val not in ('0', '1'):
  30             raise IncorrectLookupParameters
  31
  32         lookup_condition = Q(**{'%s__isnull' % self.field_path: True})
  33         if self.lookup_val == '1':
  34             return queryset.filter(lookup_condition)
  35         return queryset.exclude(lookup_condition)
  36
  37     def expected_parameters(self):
  38         return [self.lookup_kwarg]
  39
  40     def choices(self, changelist):
  41         for lookup, title in (
  42             (None, _('All')),
  43             ('1', _('Empty')),
  44             ('0', _('Not empty')),
  45         ):
  46             yield {
  47                 'selected': self.lookup_val == lookup,
  48                 'query_string': changelist.get_query_string({self.lookup_kwarg: lookup}),
  49                 'display': title,
  50             }
  51
  52
  53 class PayedListFilter(SimpleListFilter):
  54     title = 'pobrane'
  55     parameter_name = 'payed'
  56     def lookups(self, request, model_admin):
  57         return (
  58                 ('yes', 'tak'),
  59                 ('no', 'nie'),
  60                 )
  61
  62     def queryset(self, request, queryset):
  63         if self.value() == 'yes':
  64             return queryset.filter(payment__is_dd=True, payment__realised=True).distinct()
  65         if self.value() == 'no':
  66             return queryset.exclude(payment__is_dd=True, payment__realised=True).distinct()
  67
  68
  69
  70 class BankExportFeedbackLineInline(admin.TabularInline):
  71     model = models.BankExportFeedbackLine
  72     extra = 0
  73
  74 class BankPaymentInline(admin.TabularInline):
  75     model = models.Payment
  76     extra = 0
  77
  78 @admin.register(models.DirectDebit)
  79 class DirectDebitAdmin(admin.ModelAdmin):
  80     list_display = [
  81         'payment_id', 'acquisition_date',
  82         'iban_valid',
  83         'latest_status',
  84         'bank_submission_date',
  85         'bank_acceptance_date',
  86         'amount', 'first_name', 'last_name',
  87     ]
  88     date_hierarchy = 'acquisition_date'
  89     search_fields = [
  90         'payment_id', 'first_name', 'last_name', 'street', 'building', 'town', 'flat',
  91         'phone', 'email', 'iban',
  92         'notes',
  93         'fundraiser_bill',
  94     ]
  95     list_filter = [
  96         'iban_valid',
  97         'agree_fundraising',
  98         'agree_newsletter',
  99         'fundraiser',
 100         'campaign',
 101         ('cancelled_at', EmptyFieldListFilter),
 102         'latest_status',
 103         'needs_redo',
 104         'optout',
 105         'amount',
 106         'sex',
 107         'is_consumer',
 108         ('fundraiser_commission', EmptyFieldListFilter),
 109         ('fundraiser_bonus', EmptyFieldListFilter),
 110         PayedListFilter,
 111     ]
 112     fieldsets = [
 113         (None, {
 114             "fields": [
 115                 ('first_name', 'sex', 'date_of_birth'),
 116                 'last_name',
 117                 ('street', 'building'),
 118                 ('town', 'flat'),
 119                 ('postal_code', 'phone'),
 120                 'email',
 121                 ('iban', 'iban_valid', 'iban_warning'),
 122                 ('payment_id', 'latest_status'),
 123                 'agree_contact',
 124                 'agree_fundraising',
 125                 'agree_newsletter',
 126                 ('acquisition_date', 'amount'),
 127                 'is_consumer',
 128                 'fundraiser',
 129                 'campaign',
 130             ]
 131         }),
 132         (_('Processing'), {"fields": [
 133             ('cancelled_at', 'needs_redo', 'optout'),
 134             'submission_date',
 135             'fundraiser_commission',
 136             'fundraiser_bonus',
 137             'fundraiser_bill',
 138             'bank_submission_date',
 139             'bank_acceptance_date',
 140             'notes',
 141             ]
 142         })
 143     ]
 144     readonly_fields = ['agree_contact', 'iban_valid', 'iban_warning', 'latest_status']
 145     inlines = [BankExportFeedbackLineInline, BankPaymentInline]
 146
 147     def set_bank_submission(m, r, q):
 148         q.update(bank_submission_date=now())
 149
 150     def create_bank_order(m, request, queryset):
 151         bo = models.BankOrder.objects.create()
 152         bo.debits.set(queryset)
 153         messages.info(request, mark_safe(
 154             '<a href="{}">Bank order</a> created.'.format(
 155                 reverse('admin:pz_bankorder_change', args=[bo.pk])
 156             )
 157         ))
 158
 159
 160     actions = [
 161         bank.bank_export,
 162         set_bank_submission,
 163         create_bank_order,
 164         export_as_csv_action(),
 165     ]
 166
 167     def agree_contact(self, obj):
 168         return _('obligatory')
 169     agree_contact.short_description = _('agree contact')
 170
 171     def get_changeform_initial_data(self, request):
 172         return {
 173             'payment_id': models.DirectDebit.get_next_payment_id(),
 174         }
 175
 176
 177 @admin.register(models.BankExportFeedback)
 178 class BankExportFeedbackAdmin(admin.ModelAdmin):
 179     inlines = [BankExportFeedbackLineInline, BankPaymentInline]
 180
 181
 182
 183 @admin.register(models.BankOrder)
 184 class BankOrderAdmin(admin.ModelAdmin):
 185     fields = ['payment_date', 'debits', 'sent', 'download']
 186     filter_horizontal = ['debits']
 187
 188     def get_readonly_fields(self, request, obj=None):
 189         fields = ['download']
 190         if obj is not None and obj.sent:
 191             fields += ['debits', 'payment_date']
 192         return fields
 193
 194     def download(self, obj):
 195         if obj is not None and obj.pk:
 196             return mark_safe('<a href="{}">Download</a>'.format(
 197                 reverse('admin:pz_bankorder_download', args=[obj.pk])
 198             ))
 199         else:
 200             return ''
 201
 202     def get_urls(self):
 203         urls = super().get_urls()
 204         my_urls = [
 205             path(
 206                 '<int:pk>/download/',
 207                 self.admin_site.admin_view(self.download_view),
 208                 name='pz_bankorder_download',
 209             ),
 210         ]
 211         return my_urls + urls
 212
 213     def download_view(self, request, pk):
 214         order = get_object_or_404(
 215             models.BankOrder, pk=pk)
 216         try:
 217             return bank.bank_order(
 218                 order.payment_date,
 219                 order.sent,
 220                 order.debits.all()
 221             )
 222         except Exception as e:
 223             messages.error(request, str(e))
 224             return redirect('admin:pz_bankorder_change', pk)