django-orchestra/orchestra/contrib/payments/admin.py

246 lines
9.3 KiB
Python
Raw Normal View History

2014-07-23 16:24:56 +00:00
from django.contrib import admin
from django.urls import reverse
from django.http import HttpResponseRedirect
from django.utils.html import format_html
from django.utils.safestring import mark_safe
2014-07-29 20:10:37 +00:00
from django.utils.translation import ugettext_lazy as _
2014-07-23 16:24:56 +00:00
2014-11-12 16:33:40 +00:00
from orchestra.admin import ChangeViewActionsMixin, ExtendedModelAdmin
2016-04-07 11:14:44 +00:00
from orchestra.admin.utils import admin_colored, admin_link, admin_date
from orchestra.contrib.accounts.actions import list_accounts
2015-04-05 10:46:24 +00:00
from orchestra.contrib.accounts.admin import AccountAdminMixin, SelectAccountAdminMixin
from orchestra.plugins.admin import SelectPluginAdminMixin
2014-07-24 09:53:34 +00:00
from . import actions, helpers
from .methods import PaymentMethod
from .models import PaymentSource, Transaction, TransactionProcess
2014-07-23 16:24:56 +00:00
2014-07-24 09:53:34 +00:00
STATE_COLORS = {
Transaction.WAITTING_PROCESSING: 'darkorange',
2014-09-18 15:07:39 +00:00
Transaction.WAITTING_EXECUTION: 'magenta',
2014-09-16 17:14:24 +00:00
Transaction.EXECUTED: 'olive',
Transaction.SECURED: 'green',
2014-07-24 09:53:34 +00:00
Transaction.REJECTED: 'red',
}
PROCESS_STATE_COLORS = {
TransactionProcess.CREATED: 'blue',
TransactionProcess.EXECUTED: 'olive',
TransactionProcess.ABORTED: 'red',
TransactionProcess.COMMITED: 'green',
}
2014-07-24 09:53:34 +00:00
2016-05-26 08:15:34 +00:00
class PaymentSourceAdmin(SelectPluginAdminMixin, AccountAdminMixin, ExtendedModelAdmin):
2014-09-18 15:07:39 +00:00
list_display = ('label', 'method', 'number', 'account_link', 'is_active')
list_filter = ('method', 'is_active')
2016-05-26 08:15:34 +00:00
change_readonly_fields = ('method',)
2014-11-27 19:17:26 +00:00
search_fields = ('account__username', 'account__full_name', 'data')
2014-09-26 19:21:09 +00:00
plugin = PaymentMethod
plugin_field = 'method'
2014-09-18 15:07:39 +00:00
class TransactionInline(admin.TabularInline):
model = Transaction
can_delete = False
extra = 0
2014-09-16 17:14:24 +00:00
fields = (
'transaction_link', 'bill_link', 'source_link', 'display_state',
'amount', 'currency'
)
readonly_fields = fields
2015-04-02 16:14:55 +00:00
transaction_link = admin_link('__str__', short_description=_("ID"))
bill_link = admin_link('bill')
source_link = admin_link('source')
display_state = admin_colored('state', colors=STATE_COLORS)
class Media:
css = {
'all': ('orchestra/css/hide-inline-id.css',)
}
def has_add_permission(self, *args, **kwargs):
return False
2016-02-23 11:49:10 +00:00
def get_queryset(self, *args, **kwargs):
qs = super().get_queryset(*args, **kwargs)
return qs.select_related('source', 'bill')
2014-10-21 15:29:36 +00:00
class TransactionAdmin(SelectAccountAdminMixin, ExtendedModelAdmin):
2014-07-24 09:53:34 +00:00
list_display = (
'id', 'bill_link', 'account_link', 'source_link', 'display_created_at',
'display_modified_at', 'display_state', 'amount', 'process_link'
2014-07-24 09:53:34 +00:00
)
list_filter = ('source__method', 'state')
2014-10-21 15:29:36 +00:00
fieldsets = (
(None, {
'classes': ('wide',),
'fields': (
'account_link',
'bill_link',
'source_link',
'display_state',
'amount',
'currency',
'process_link'
)
}),
2016-04-07 11:14:44 +00:00
(_("Dates"), {
'classes': ('wide',),
'fields': ('display_created_at', 'display_modified_at'),
}),
2014-10-21 15:29:36 +00:00
)
add_fieldsets = (
(None, {
'classes': ('wide',),
'fields': (
'bill',
'source',
'display_state',
'amount',
'currency',
)
}),
)
2015-06-03 12:49:30 +00:00
change_view_actions = (
actions.process_transactions, actions.mark_as_executed, actions.mark_as_secured,
actions.mark_as_rejected, actions.reissue
2014-09-16 17:14:24 +00:00
)
search_fields = ('bill__number', 'bill__account__username', 'id')
actions = change_view_actions + (actions.report, list_accounts,)
2014-10-21 15:29:36 +00:00
filter_by_account_fields = ('bill', 'source')
2015-07-14 10:19:45 +00:00
readonly_fields = (
2016-04-07 11:14:44 +00:00
'bill_link', 'display_state', 'process_link', 'account_link', 'source_link',
'display_created_at', 'display_modified_at'
2015-07-14 10:19:45 +00:00
)
list_select_related = ('source', 'bill__account', 'process')
2015-06-22 14:14:16 +00:00
date_hierarchy = 'created_at'
2014-07-24 09:53:34 +00:00
bill_link = admin_link('bill')
2014-09-04 15:55:43 +00:00
source_link = admin_link('source')
process_link = admin_link('process', short_description=_("proc"))
2014-07-24 09:53:34 +00:00
account_link = admin_link('bill__account')
2016-04-07 11:14:44 +00:00
display_created_at = admin_date('created_at', short_description=_("Created"))
display_modified_at = admin_date('modified_at', short_description=_("Modified"))
def has_delete_permission(self, *args, **kwargs):
return False
def get_actions(self, request):
actions = super().get_actions(request)
if 'delete_selected' in actions:
del actions['delete_selected']
return actions
2016-05-18 14:08:12 +00:00
def get_change_readonly_fields(self, request, obj):
if obj.state in (Transaction.WAITTING_PROCESSING, Transaction.WAITTING_EXECUTION):
return ()
return ('amount', 'currency')
2014-09-16 17:14:24 +00:00
def get_change_view_actions(self, obj=None):
actions = super(TransactionAdmin, self).get_change_view_actions()
2014-09-18 15:07:39 +00:00
exclude = []
2014-09-16 17:14:24 +00:00
if obj:
2014-09-19 14:47:25 +00:00
if obj.state == Transaction.WAITTING_PROCESSING:
exclude = ['mark_as_executed', 'mark_as_secured', 'reissue']
2014-09-19 14:47:25 +00:00
elif obj.state == Transaction.WAITTING_EXECUTION:
exclude = ['process_transactions', 'mark_as_secured', 'reissue']
2014-09-16 17:14:24 +00:00
if obj.state == Transaction.EXECUTED:
exclude = ['process_transactions', 'mark_as_executed', 'reissue']
elif obj.state == Transaction.REJECTED:
exclude = ['process_transactions', 'mark_as_executed', 'mark_as_secured', 'mark_as_rejected']
elif obj.state == Transaction.SECURED:
2014-09-19 14:47:25 +00:00
return []
2014-09-18 15:07:39 +00:00
return [action for action in actions if action.__name__ not in exclude]
@mark_safe
2015-09-29 12:35:22 +00:00
def display_state(self, obj):
state = admin_colored('state', colors=STATE_COLORS)(obj)
help_text = obj.get_state_help()
state = state.replace('<span ', '<span title="%s" ' % help_text)
return state
display_state.admin_order_field = 'state'
display_state.short_description = _("State")
2014-09-18 15:07:39 +00:00
class TransactionProcessAdmin(ChangeViewActionsMixin, admin.ModelAdmin):
list_display = (
'id', 'file_url', 'display_transactions', 'display_state', 'display_created_at',
)
list_filter = ('state',)
2014-10-11 16:21:51 +00:00
fields = ('data', 'file_url', 'created_at')
search_fields = ('transactions__bill__number', 'transactions__bill__account__username', 'id')
2014-10-21 15:29:36 +00:00
readonly_fields = ('data', 'file_url', 'display_transactions', 'created_at')
2016-02-19 10:11:28 +00:00
list_prefetch_related = ('transactions',)
inlines = [TransactionInline]
2015-06-03 12:49:30 +00:00
change_view_actions = (
actions.mark_process_as_executed, actions.abort, actions.commit, actions.report
)
actions = change_view_actions
display_state = admin_colored('state', colors=PROCESS_STATE_COLORS)
2016-04-07 11:14:44 +00:00
display_created_at = admin_date('created_at', short_description=_("Created"))
2014-07-29 20:10:37 +00:00
def file_url(self, process):
if process.file:
return format_html('<a href="{}">{}</a>', process.file.url, process.file.name)
2014-07-29 20:10:37 +00:00
file_url.admin_order_field = 'file'
@mark_safe
2014-07-29 20:10:37 +00:00
def display_transactions(self, process):
ids = []
lines = []
counter = 0
2016-02-19 10:11:28 +00:00
for trans in process.transactions.all():
color = STATE_COLORS.get(trans.state, 'black')
state = trans.get_state_display()
ids.append('<span style="color:%s" title="%s">%i</span>' % (color, state, trans.id))
2016-04-07 11:14:44 +00:00
counter += 1 + len(str(trans.id))
if counter > 100:
counter = 0
lines.append(','.join(ids))
ids = []
lines.append(','.join(ids))
2016-04-06 19:00:16 +00:00
transactions = '<br>'.join(lines)
url = reverse('admin:payments_transaction_changelist')
2016-02-19 10:11:28 +00:00
url += '?process_id=%i' % process.id
2014-10-11 16:21:51 +00:00
return '<a href="%s">%s</a>' % (url, transactions)
2014-07-29 20:10:37 +00:00
display_transactions.short_description = _("Transactions")
2014-09-22 15:59:53 +00:00
def has_add_permission(self, *args, **kwargs):
return False
2014-09-18 15:07:39 +00:00
def get_change_view_actions(self, obj=None):
2016-02-19 10:11:28 +00:00
actions = super().get_change_view_actions()
2014-09-18 15:07:39 +00:00
exclude = []
if obj:
if obj.state == TransactionProcess.EXECUTED:
exclude.append('mark_process_as_executed')
elif obj.state == TransactionProcess.COMMITED:
exclude = ['mark_process_as_executed', 'abort', 'commit']
elif obj.state == TransactionProcess.ABORTED:
exclude = ['mark_process_as_executed', 'abort', 'commit']
return [action for action in actions if action.__name__ not in exclude]
def delete_view(self, request, object_id, extra_context=None):
queryset = self.model.objects.filter(id=object_id)
related_transactions = helpers.pre_delete_processes(self, request, queryset)
2016-02-19 10:11:28 +00:00
response = super().delete_view(request, object_id, extra_context)
if isinstance(response, HttpResponseRedirect):
helpers.post_delete_processes(self, request, related_transactions)
return response
2014-07-29 20:10:37 +00:00
def delete_queryset(self, request, queryset):
# override default admin action delete behaviour
related_transactions = helpers.pre_delete_processes(self, request, queryset)
super().delete_queryset(self, request, queryset)
helpers.post_delete_processes(self, request, related_transactions)
admin.site.register(PaymentSource, PaymentSourceAdmin)
2014-07-24 09:53:34 +00:00
admin.site.register(Transaction, TransactionAdmin)
admin.site.register(TransactionProcess, TransactionProcessAdmin)