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') 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] def display_state(self, obj): state = admin_colored('state', colors=STATE_COLORS)(obj) help_text = obj.get_state_help() state = state.replace('%s' % (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 for trans in process.transactions.only('id', 'state'): color = STATE_COLORS.get(trans.state, 'black') state = trans.get_state_display() ids.append('%i' % (color, state, trans.id)) counter += 1 if counter > 10: counter = 0 lines.append(','.join(ids)) ids = [] lines.append(','.join(ids)) transactions = '%s' % (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)