from django.contrib import admin

from .models import (
    Client,
    Contract,
    Prestation,
)


class ClientAdmin(admin.ModelAdmin):
    model = Client

    list_display = ("name", "address", "postal_code", "city", "contact")
    search_fields = ("name", "adress", "city")


class ContractAdmin(admin.ModelAdmin):
    model = Contract

    list_display = ("title", "client", "advance", "reference", "date")
    search_fields = ("title",)
    list_filter = ("is_finished",)  # 'date__year',


class PrestationAdmin(admin.ModelAdmin):
    model = Prestation

    list_display = ("date", "label", "total_amount", "contract")
    search_fields = ("label",)


admin.site.register(Client, ClientAdmin)
admin.site.register(Contract, ContractAdmin)
admin.site.register(Prestation, PrestationAdmin)