from django.contrib import admin

from compensation.models import Compensation, CompensationAction, CompensationState, Payment, \
    EcoAccountDeduction, EcoAccount
from konova.admin import BaseObjectAdmin


class CompensationStateAdmin(admin.ModelAdmin):
    list_display = [
        "id",
        "biotope_type",
        "surface",
    ]


class CompensationActionAdmin(admin.ModelAdmin):
    list_display = [
        "id",
        "action_type",
        "amount",
        "unit",
        "comment",
    ]


class CompensationAdmin(BaseObjectAdmin):
    list_display = [
        "id",
        "identifier",
        "title",
        "created",
        "deleted",
    ]


class EcoAccountAdmin(admin.ModelAdmin):
    list_display = [
        "id",
        "identifier",
        "title",
    ]


class PaymentAdmin(admin.ModelAdmin):
    list_display = [
        "id",
        "amount",
        "due_on",
    ]


class EcoAccountDeductionAdmin(admin.ModelAdmin):
    list_display = [
        "id",
        "account",
        "intervention",
        "surface",
    ]


admin.site.register(Compensation, CompensationAdmin)
admin.site.register(Payment, PaymentAdmin)
admin.site.register(CompensationAction, CompensationActionAdmin)
admin.site.register(CompensationState, CompensationStateAdmin)
admin.site.register(EcoAccount, EcoAccountAdmin)
admin.site.register(EcoAccountDeduction, EcoAccountDeductionAdmin)