from django.contrib import admin
from import_export.admin import ImportExportModelAdmin

from .models import InvoiceHistory, WithDraw, Transaction, Wallet, PayFromScore


class InvoiceHistoryAdmin(admin.ModelAdmin):
    list_display = ('comment', 'sum', 'user', 'balance',)


# class WithDrawResource(resources.ModelResource):
#
#     class Meta:
#         model = WithDraw
#         export_order = ('id', 'sum', 'yandex_card', 'user')


class WithDrawAdmin(ImportExportModelAdmin):
    list_display = ('sum', 'created', 'yandex_card', 'user', 'complete',)


class TransactionAdmin(admin.ModelAdmin):
    list_display = ('customer', 'complete',)


admin.site.register(InvoiceHistory, InvoiceHistoryAdmin)
admin.site.register(WithDraw, WithDrawAdmin)
admin.site.register(Transaction, TransactionAdmin)
admin.site.register(Wallet)
admin.site.register(PayFromScore)