190 lines
7 KiB
Python
190 lines
7 KiB
Python
from django.contrib import admin
|
|
from django.core.urlresolvers import reverse
|
|
from django.http import HttpResponseRedirect
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from orchestra.admin import ChangeViewActionsMixin, ExtendedModelAdmin
|
|
from orchestra.admin.utils import admin_colored, admin_link
|
|
from orchestra.contrib.accounts.actions import list_accounts
|
|
from orchestra.contrib.accounts.admin import AccountAdminMixin, SelectAccountAdminMixin
|
|
from orchestra.plugins.admin import SelectPluginAdminMixin
|
|
|
|
from . import actions, helpers
|
|
from .methods import PaymentMethod
|
|
from .models import PaymentSource, Transaction, TransactionProcess
|
|
|
|
|
|
STATE_COLORS = {
|
|
Transaction.WAITTING_PROCESSING: 'darkorange',
|
|
Transaction.WAITTING_EXECUTION: 'magenta',
|
|
Transaction.EXECUTED: 'olive',
|
|
Transaction.SECURED: 'green',
|
|
Transaction.REJECTED: 'red',
|
|
}
|
|
|
|
|
|
class PaymentSourceAdmin(SelectPluginAdminMixin, AccountAdminMixin, admin.ModelAdmin):
|
|
list_display = ('label', 'method', 'number', 'account_link', 'is_active')
|
|
list_filter = ('method', 'is_active')
|
|
search_fields = ('account__username', 'account__full_name', 'data')
|
|
plugin = PaymentMethod
|
|
plugin_field = 'method'
|
|
|
|
|
|
class TransactionInline(admin.TabularInline):
|
|
model = Transaction
|
|
can_delete = False
|
|
extra = 0
|
|
fields = (
|
|
'transaction_link', 'bill_link', 'source_link', 'display_state',
|
|
'amount', 'currency'
|
|
)
|
|
readonly_fields = fields
|
|
|
|
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
|
|
|
|
|
|
class TransactionAdmin(SelectAccountAdminMixin, ExtendedModelAdmin):
|
|
list_display = (
|
|
'id', 'bill_link', 'account_link', 'source_link', 'display_state',
|
|
'amount', 'process_link'
|
|
)
|
|
list_filter = ('source__method', 'state')
|
|
fieldsets = (
|
|
(None, {
|
|
'classes': ('wide',),
|
|
'fields': (
|
|
'account_link',
|
|
'bill_link',
|
|
'source_link',
|
|
'display_state',
|
|
'amount',
|
|
'currency',
|
|
'process_link'
|
|
)
|
|
}),
|
|
)
|
|
add_fieldsets = (
|
|
(None, {
|
|
'classes': ('wide',),
|
|
'fields': (
|
|
'bill',
|
|
'source',
|
|
'display_state',
|
|
'amount',
|
|
'currency',
|
|
'process'
|
|
)
|
|
}),
|
|
)
|
|
change_view_actions = (
|
|
actions.process_transactions, actions.mark_as_executed, actions.mark_as_secured,
|
|
actions.mark_as_rejected,
|
|
)
|
|
actions = change_view_actions + (actions.report, list_accounts)
|
|
filter_by_account_fields = ('bill', 'source')
|
|
change_readonly_fields = ('amount', 'currency')
|
|
readonly_fields = (
|
|
'bill_link', 'display_state', 'process_link', 'account_link', 'source_link'
|
|
)
|
|
list_select_related = ('source', 'bill__account', 'process')
|
|
date_hierarchy = 'created_at'
|
|
|
|
bill_link = admin_link('bill')
|
|
source_link = admin_link('source')
|
|
process_link = admin_link('process', short_description=_("proc"))
|
|
account_link = admin_link('bill__account')
|
|
display_state = admin_colored('state', colors=STATE_COLORS)
|
|
|
|
def get_change_view_actions(self, obj=None):
|
|
actions = super(TransactionAdmin, self).get_change_view_actions()
|
|
exclude = []
|
|
if obj:
|
|
if obj.state == Transaction.WAITTING_PROCESSING:
|
|
exclude = ['mark_as_executed', 'mark_as_secured']
|
|
elif obj.state == Transaction.WAITTING_EXECUTION:
|
|
exclude = ['process_transactions', 'mark_as_secured']
|
|
if obj.state == Transaction.EXECUTED:
|
|
exclude = ['process_transactions', 'mark_as_executed']
|
|
elif obj.state in [Transaction.REJECTED, Transaction.SECURED]:
|
|
return []
|
|
return [action for action in actions if action.__name__ not in exclude]
|
|
|
|
|
|
class TransactionProcessAdmin(ChangeViewActionsMixin, admin.ModelAdmin):
|
|
list_display = ('id', 'file_url', 'display_transactions', 'created_at')
|
|
fields = ('data', 'file_url', 'created_at')
|
|
readonly_fields = ('data', 'file_url', 'display_transactions', 'created_at')
|
|
inlines = [TransactionInline]
|
|
change_view_actions = (
|
|
actions.mark_process_as_executed, actions.abort, actions.commit, actions.report
|
|
)
|
|
actions = change_view_actions + (actions.delete_selected,)
|
|
|
|
def file_url(self, process):
|
|
if process.file:
|
|
return '<a href="%s">%s</a>' % (process.file.url, process.file.name)
|
|
file_url.allow_tags = True
|
|
file_url.admin_order_field = 'file'
|
|
|
|
def display_transactions(self, process):
|
|
ids = []
|
|
lines = []
|
|
counter = 0
|
|
# Because of values_list this query doesn't benefit from prefetch_related
|
|
tx_ids = process.transactions.values_list('id', flat=True)
|
|
for tx_id in tx_ids:
|
|
ids.append(str(tx_id))
|
|
counter += 1
|
|
if counter > 10:
|
|
counter = 0
|
|
lines.append(','.join(ids))
|
|
ids = []
|
|
lines.append(','.join(ids))
|
|
transactions = '<br'.join(lines)
|
|
url = reverse('admin:payments_transaction_changelist')
|
|
url += '?processes=%i' % process.id
|
|
return '<a href="%s">%s</a>' % (url, transactions)
|
|
display_transactions.short_description = _("Transactions")
|
|
display_transactions.allow_tags = True
|
|
|
|
def has_add_permission(self, *args, **kwargs):
|
|
return False
|
|
|
|
def get_change_view_actions(self, obj=None):
|
|
actions = super(TransactionProcessAdmin, self).get_change_view_actions()
|
|
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)
|
|
response = super(TransactionProcessAdmin, self).delete_view(
|
|
request, object_id, extra_context)
|
|
if isinstance(response, HttpResponseRedirect):
|
|
helpers.post_delete_processes(self, request, related_transactions)
|
|
return response
|
|
|
|
|
|
admin.site.register(PaymentSource, PaymentSourceAdmin)
|
|
admin.site.register(Transaction, TransactionAdmin)
|
|
admin.site.register(TransactionProcess, TransactionProcessAdmin)
|