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