src/club/admin.py

   1 # This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
   3 #
   4 import json
   5 from django.contrib import admin
   6 from django.db.models.functions import Now
   7 from django.db.models import Q
   8 from django import forms
   9 from django.utils.html import conditional_escape
  10 from django.utils.safestring import mark_safe
  11 from fnpdjango.actions import export_as_csv_action
  12 from modeltranslation.admin import TranslationAdmin
  13 from wolnelektury.utils import YesNoFilter
  14 from . import models
  15
  16
  17 class SingleAmountInline(admin.TabularInline):
  18     model = models.SingleAmount
  19
  20
  21 class MonthlyAmountInline(admin.TabularInline):
  22     model = models.MonthlyAmount
  23
  24
  25 @admin.register(models.Club)
  26 class ClubAdmin(admin.ModelAdmin):
  27     inlines = [
  28         SingleAmountInline,
  29         MonthlyAmountInline
  30     ]
  31
  32
  33 class PayUOrderInline(admin.TabularInline):
  34     model = models.PayUOrder
  35     fields = ['order_id', 'status', 'customer_ip']
  36     readonly_fields = fields
  37     extra = 0
  38     show_change_link = True
  39     can_delete = False
  40
  41     def has_add_permission(self, request, obj):
  42         return False
  43
  44
  45 class PayUCardTokenInline(admin.TabularInline):
  46     model = models.PayUCardToken
  47     fields = ['created_at', 'disposable_token', 'reusable_token']
  48     readonly_fields = fields
  49     extra = 0
  50     show_change_link = True
  51     can_delete = False
  52     show_change_link = True
  53
  54     def has_add_permission(self, request, obj):
  55         return False
  56
  57
  58 class PayedFilter(YesNoFilter):
  59     title = 'płatność zakończona'
  60     parameter_name = 'payed'
  61     q = ~Q(payed_at=None)
  62
  63
  64 class ExpiredFilter(YesNoFilter):
  65     title = 'harmonogram przedawniony'
  66     parameter_name = 'expired'
  67     q = Q(expires_at__isnull=False, expires_at__lt=Now())
  68
  69
  70 class ActiveFilter(YesNoFilter):
  71     title = 'płatność aktualna'
  72     parameter_name = 'active'
  73     q = Q(expires_at__gt=Now())
  74
  75
  76 class ScheduleForm(forms.ModelForm):
  77     def __init__(self, *args, **kwargs):
  78         super().__init__(*args, **kwargs)
  79         self.fields['email'].required = False
  80         self.fields['method'].required = False
  81         self.fields['consent'].required = False
  82
  83     class Meta:
  84         model = models.Schedule
  85         fields = '__all__'
  86
  87
  88 class SourceFilter(admin.SimpleListFilter):
  89     title = 'Źródło' # display title
  90     parameter_name = 'source'
  91     template = "admin/long_filter.html"
  92
  93     def lookups(self, request, model_admin):
  94         lookups = [
  95             (m, m) for m in
  96             model_admin.model.objects.exclude(source='').values_list('source', flat=True).distinct()[:10]
  97         ]
  98         print(lookups)
  99         return lookups
 100
 101     def queryset(self, request, queryset):
 102         return queryset
 103
 104     #field_name = 'source' # name of the foreign key field
 105
 106
 107
 108
 109 class ScheduleAdmin(admin.ModelAdmin):
 110     form = ScheduleForm
 111
 112     list_display = [
 113         'email', 'started_at', 'payed_at', 'expires_at', 'amount', 'monthly', 'yearly', 'is_cancelled',
 114         'method'
 115     ]
 116     list_display_links = ['email', 'started_at']
 117     search_fields = ['email', 'source']
 118     list_filter = [
 119         'is_cancelled', 'monthly', 'yearly', 'method',
 120         PayedFilter, ActiveFilter, ExpiredFilter,
 121         SourceFilter,
 122     ]
 123     filter_horizontal = ['consent']
 124     date_hierarchy = 'started_at'
 125     raw_id_fields = ['membership']
 126     inlines = [PayUOrderInline, PayUCardTokenInline]
 127     actions = [export_as_csv_action()]
 128
 129 admin.site.register(models.Schedule, ScheduleAdmin)
 130
 131
 132 class ScheduleInline(admin.TabularInline):
 133     model = models.Schedule
 134     fields = ['email', 'amount', 'is_cancelled', 'started_at', 'payed_at', 'expires_at', 'email_sent']
 135     readonly_fields = fields
 136     extra = 0
 137     show_change_link = True
 138     can_delete = False
 139
 140     def has_add_permission(self, request, obj):
 141         return False
 142
 143
 144 class MembershipAdmin(admin.ModelAdmin):
 145     list_display = ['user', 'manual', 'updated_at', 'notes']
 146     list_filter = ['manual']
 147     date_hierarchy = 'updated_at'
 148     raw_id_fields = ['user']
 149     search_fields = ['user__username', 'user__email', 'schedule__email', 'notes']
 150     inlines = [ScheduleInline]
 151
 152 admin.site.register(models.Membership, MembershipAdmin)
 153
 154
 155 admin.site.register(models.ReminderEmail, TranslationAdmin)
 156
 157
 158 class PayUNotificationAdmin(admin.ModelAdmin):
 159     list_display = ['received_at', 'order']
 160     fields = ['received_at', 'order', 'body_']
 161     readonly_fields = ['received_at', 'body_']
 162     raw_id_fields = ['order']
 163
 164     def body_(self, obj):
 165         return mark_safe(
 166                 "<pre>" +
 167                 conditional_escape(json.dumps(json.loads(obj.body), indent=4))
 168                 + "</pre>")
 169
 170
 171 admin.site.register(models.PayUNotification, PayUNotificationAdmin)
 172
 173
 174 class PayUNotificationInline(admin.TabularInline):
 175     model = models.PayUNotification
 176     fields = ['received_at', 'body_']
 177     readonly_fields = fields
 178     extra = 0
 179     show_change_link = True
 180     can_delete = False
 181
 182     def body_(self, obj):
 183         return mark_safe(
 184                 "<pre>" +
 185                 conditional_escape(json.dumps(json.loads(obj.body), indent=4))
 186                 + "</pre>")
 187
 188     def has_add_permission(self, request, obj):
 189         return False
 190
 191
 192 class PayUOrderAdmin(admin.ModelAdmin):
 193     list_display = ['schedule']
 194     raw_id_fields = ['schedule']
 195     inlines = [PayUNotificationInline]
 196
 197
 198 admin.site.register(models.PayUOrder, PayUOrderAdmin)
 199
 200
 201 admin.site.register(models.Ambassador)
 202
 203
 204
 205
 206 @admin.register(models.Consent)
 207 class ConsentAdmin(admin.ModelAdmin):
 208     list_display = ['text', 'order', 'active', 'required']
 209
 210     def get_readonly_fields(self, request, obj=None):
 211         if obj:
 212             return ['text']
 213         else:
 214             return []