from django.contrib import admin
from .models import Client, Brigade, Service, Order, CompletedWork


class CompletedWorkInline(admin.TabularInline):
    model = CompletedWork
    extra = 1

@admin.register(Client)
class ClientAdmin(admin.ModelAdmin):
    list_display = ['full_name', 'phone', 'address', 'preferred_time']
    search_fields = ['full_name', 'phone', 'address']
    list_filter = ['preferred_time']

@admin.register(Brigade)
class BrigadeAdmin(admin.ModelAdmin):
    list_display = ['name', 'supervisor', 'employee_count', 'specialization']
    search_fields = ['name', 'supervisor', 'specialization']


@admin.register(Service)
class ServiceAdmin(admin.ModelAdmin):
    list_display = ['name', 'price_per_sqm', 'duration_per_sqm']
    search_fields = ['name']

@admin.register(Order)
class OrderAdmin(admin.ModelAdmin):
    list_display = ['id', 'client', 'brigade', 'date', 'start_time', 'end_time']
    search_fields = ['client__full_name', 'brigade__name']
    list_filter = ['date', 'brigade']
    inlines = [CompletedWorkInline]

@admin.register(CompletedWork)
class CompletedWorkAdmin(admin.ModelAdmin):
    list_display = ['order', 'service', 'area']
    search_fields = ['order__client__full_name', 'service__name']