nk20/apps/note/admin.py

164 lines
4.3 KiB
Python
Raw Normal View History

2020-02-18 20:30:26 +00:00
# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
2019-07-16 13:42:31 +00:00
# SPDX-License-Identifier: GPL-3.0-or-later
2019-07-08 12:30:58 +00:00
from django.contrib import admin
2019-07-17 10:48:03 +00:00
from django.utils.translation import gettext_lazy as _
from polymorphic.admin import PolymorphicChildModelAdmin, \
PolymorphicChildModelFilter, PolymorphicParentModelAdmin
2019-07-08 12:30:58 +00:00
2019-07-17 09:17:50 +00:00
from .models.notes import Alias, Note, NoteClub, NoteSpecial, NoteUser
2020-02-04 00:18:03 +00:00
from .models.transactions import Transaction, TransactionCategory, TransactionTemplate
2019-07-16 07:05:30 +00:00
2019-07-16 12:38:52 +00:00
class AliasInlines(admin.TabularInline):
"""
Define user and club aliases when editing their note
"""
extra = 0
model = Alias
2019-07-17 10:14:23 +00:00
@admin.register(Note)
class NoteAdmin(PolymorphicParentModelAdmin):
2019-07-16 12:38:52 +00:00
"""
2019-07-17 10:14:23 +00:00
Parent regrouping all note types as children
"""
child_models = (NoteClub, NoteSpecial, NoteUser)
2020-02-18 11:31:15 +00:00
list_filter = (
PolymorphicChildModelFilter,
'is_active',
)
2019-07-17 10:14:23 +00:00
# Use a polymorphic list
2019-07-24 20:03:07 +00:00
list_display = ('pretty', 'balance', 'is_active')
2019-07-17 10:14:23 +00:00
polymorphic_list = True
# Organize notes by registration date
date_hierarchy = 'created_at'
ordering = ['-created_at']
# Search by aliases
search_fields = ['alias__name']
@admin.register(NoteClub)
class NoteClubAdmin(PolymorphicChildModelAdmin):
"""
Child for a club note, see NoteAdmin
2019-07-16 12:38:52 +00:00
"""
2020-02-18 11:31:15 +00:00
inlines = (AliasInlines, )
2019-07-16 12:38:52 +00:00
2019-07-17 09:17:50 +00:00
# We can't change club after creation or the balance
readonly_fields = ('club', 'balance')
2020-02-18 11:31:15 +00:00
search_fields = ('club', )
2019-07-17 07:49:59 +00:00
def has_add_permission(self, request):
"""
A club note should not be manually added
"""
return False
def has_delete_permission(self, request, obj=None):
"""
A club note should not be manually removed
"""
return False
2019-07-16 12:38:52 +00:00
2019-07-17 10:14:23 +00:00
@admin.register(NoteSpecial)
class NoteSpecialAdmin(PolymorphicChildModelAdmin):
2019-07-16 12:38:52 +00:00
"""
2019-07-17 10:14:23 +00:00
Child for a special note, see NoteAdmin
2019-07-16 12:38:52 +00:00
"""
2020-02-18 11:31:15 +00:00
readonly_fields = ('balance', )
2019-07-16 12:38:52 +00:00
2019-07-17 10:14:23 +00:00
@admin.register(NoteUser)
class NoteUserAdmin(PolymorphicChildModelAdmin):
2019-07-16 12:38:52 +00:00
"""
2019-07-17 10:14:23 +00:00
Child for an user note, see NoteAdmin
2019-07-16 12:38:52 +00:00
"""
2020-02-18 11:31:15 +00:00
inlines = (AliasInlines, )
2019-07-16 12:38:52 +00:00
2019-07-17 09:17:50 +00:00
# We can't change user after creation or the balance
readonly_fields = ('user', 'balance')
2019-07-17 07:49:59 +00:00
def has_add_permission(self, request):
"""
An user note should not be manually added
"""
return False
def has_delete_permission(self, request, obj=None):
"""
An user note should not be manually removed
"""
return False
2019-07-16 12:38:52 +00:00
2019-07-17 10:48:03 +00:00
@admin.register(Transaction)
class TransactionAdmin(admin.ModelAdmin):
"""
Admin customisation for Transaction
"""
list_display = ('created_at', 'poly_source', 'poly_destination',
'quantity', 'amount', 'transaction_type', 'valid')
list_filter = ('transaction_type', 'valid')
2020-02-18 11:31:15 +00:00
autocomplete_fields = (
'source',
'destination',
)
2019-07-17 10:48:03 +00:00
def poly_source(self, obj):
"""
Force source to resolve polymorphic object
"""
return str(obj.source)
poly_source.short_description = _('source')
def poly_destination(self, obj):
"""
Force destination to resolve polymorphic object
"""
return str(obj.destination)
poly_destination.short_description = _('destination')
2019-07-19 13:00:44 +00:00
def get_readonly_fields(self, request, obj=None):
"""
Only valid can be edited after creation
Else the amount of money would not be transferred
"""
if obj: # user is editing an existing object
return 'created_at', 'source', 'destination', 'quantity',\
'amount', 'transaction_type'
return []
2019-07-17 10:48:03 +00:00
2019-07-17 10:14:23 +00:00
@admin.register(TransactionTemplate)
2019-07-16 13:42:31 +00:00
class TransactionTemplateAdmin(admin.ModelAdmin):
"""
Admin customisation for TransactionTemplate
"""
2019-07-17 10:48:03 +00:00
list_display = ('name', 'poly_destination', 'amount', 'template_type')
2020-02-18 11:31:15 +00:00
list_filter = ('template_type', )
autocomplete_fields = ('destination', )
2019-07-16 13:42:31 +00:00
2019-07-17 10:48:03 +00:00
def poly_destination(self, obj):
"""
Force destination to resolve polymorphic object
"""
return str(obj.destination)
2019-07-16 13:42:31 +00:00
2019-07-17 10:48:03 +00:00
poly_destination.short_description = _('destination')
2020-02-04 00:18:03 +00:00
@admin.register(TransactionCategory)
class TransactionCategoryAdmin(admin.ModelAdmin):
"""
Admin customisation for TransactionTemplate
"""
2020-02-18 11:31:15 +00:00
list_display = ('name', )
list_filter = ('name', )