from django.contrib import admin

from .models import CountReason, CountTask, CountTaskDetail


@admin.register(CountReason)
class CountReasonAdmin(admin.ModelAdmin):
    list_display = ("code", "description", "is_active", "sort", "update_time")
    list_filter = ("is_active",)
    search_fields = ("code", "description")
    ordering = ("sort", "code")


class CountTaskDetailInline(admin.TabularInline):
    model = CountTaskDetail
    extra = 0
    readonly_fields = (
        "container_code",
        "goods_code",
        "book_qty",
        "initial_count_qty",
        "recount_qty",
        "final_qty",
        "variance_qty",
        "status",
    )


@admin.register(CountTask)
class CountTaskAdmin(admin.ModelAdmin):
    list_display = (
        "doc_no",
        "task_type",
        "status",
        "total_details",
        "counted_details",
        "variance_details",
        "create_time",
    )
    list_filter = ("task_type", "status")
    search_fields = ("doc_no",)
    readonly_fields = ("total_details", "counted_details", "variance_details")
    inlines = (CountTaskDetailInline,)


@admin.register(CountTaskDetail)
class CountTaskDetailAdmin(admin.ModelAdmin):
    list_display = (
        "task",
        "container_code",
        "goods_code",
        "book_qty",
        "initial_count_qty",
        "recount_qty",
        "variance_qty",
        "status",
    )
    list_filter = ("status",)
    search_fields = ("container_code", "goods_code", "task__doc_no")