from django.contrib import admin
from labor.models import Worker, WorkerAttendance, WorkerPayment


@admin.register(Worker)
class WorkerAdmin(admin.ModelAdmin):
    list_display = ("name", "project", "work_type", "rate_type", "daily_rate",
                     "total_days_worked", "total_earned", "total_paid", "due_amount", "is_active")
    list_filter = ("project", "work_type", "rate_type", "is_active")
    search_fields = ("name", "mobile_number")


@admin.register(WorkerAttendance)
class WorkerAttendanceAdmin(admin.ModelAdmin):
    list_display = ("worker", "date", "days_worked")
    list_filter = ("worker__project", "date")
    date_hierarchy = "date"


@admin.register(WorkerPayment)
class WorkerPaymentAdmin(admin.ModelAdmin):
    list_display = ("worker", "date", "amount", "payment_type")
    list_filter = ("worker__project", "payment_type", "date")
    date_hierarchy = "date"