from django.contrib import admin

from django_admin_listfilter_dropdown.filters import (
    DropdownFilter,
    ChoiceDropdownFilter,
    RelatedDropdownFilter,
)

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


@admin.register(Client)
class ClientAdmin(admin.ModelAdmin):
    model = Client

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


@admin.register(Contract)
class ContractAdmin(admin.ModelAdmin):
    model = Contract

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


@admin.register(Prestation)
class PrestationAdmin(admin.ModelAdmin):
    model = Prestation

    list_display = ("date", "label", "total_amount", "contract")
    search_fields = ("label", "contract__title")
    list_filter = (
        ("contract", RelatedDropdownFilter),
    )