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

from .models import InvoiceHistory, WithDraw, Transaction, Wallet


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


# class WithDrawAdmin(admin.ModelAdmin):
#     list_display = ('sum','created','user',)


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)