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 = ('name', 'client', 'advance', 'reference', 'date')
	search_fields = ('name', )
	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)