from django.contrib import admin
from contractors.models import Contractor, ContractorBill, ContractorPayment


@admin.register(Contractor)
class ContractorAdmin(admin.ModelAdmin):
    list_display = ("name", "project", "work_order_no", "contract_amount",
                     "total_billed", "total_paid", "remaining_amount", "is_active")
    list_filter = ("project", "is_active")
    search_fields = ("name", "work_order_no")


@admin.register(ContractorBill)
class ContractorBillAdmin(admin.ModelAdmin):
    list_display = ("contractor", "date", "amount", "description")
    list_filter = ("contractor__project", "date")
    date_hierarchy = "date"


@admin.register(ContractorPayment)
class ContractorPaymentAdmin(admin.ModelAdmin):
    list_display = ("contractor", "date", "amount", "payment_type")
    list_filter = ("contractor__project", "payment_type", "date")
    date_hierarchy = "date"