from django.contrib import admin

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

@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 = ('name', 'client', 'advance', 'reference', 'date')
	search_fields = ('name', )
	list_filter = ('is_finished', )	# 'date__year',


@admin.register(Prestation)
class PrestationAdmin(admin.ModelAdmin):
	model = Prestation
	list_display = ('date', 'label', 'total_amount_htva', 'total_amount_tvac', 'contract')
	search_fields = ('label', )


@admin.register(PrestationType)
class PrestationTypeAdmin(admin.ModelAdmin):
    model = PrestationType
    list_display = ('label',)
    search_fields = ('label',)

@admin.register(DiscountContract)
class DiscountContractAdmin(admin.ModelAdmin):
    model = DiscountContract
    list_display = ('contract', 'label', 'quantity', 'unit')