from django.contrib import admin
from unfold.admin import ModelAdmin

from core.apps.evaluation.models import DocumentModel, ValuationDocumentModel


@admin.register(ValuationDocumentModel)
class ValuationDocumentAdmin(ModelAdmin):
    list_display = (
        "id",
        "valuation",
        "document_type",
        "title",
        "uploaded_by",
        "created_at",
    )
    list_filter = ("document_type",)
    search_fields = (
        "title",
        "description",
        "valuation__conclusion_number",
    )
    readonly_fields = ("created_at", "updated_at")
    autocomplete_fields = ("valuation", "uploaded_by")
    fieldsets = (
        (
            "Hujjat",
            {
                "fields": (
                    "valuation",
                    "document_type",
                    "title",
                    "file",
                    "uploaded_by",
                ),
            },
        ),
        (
            "Qo'shimcha",
            {
                "fields": ("description",),
            },
        ),
        (
            "Tizim",
            {
                "classes": ("collapse",),
                "fields": ("created_at", "updated_at"),
            },
        ),
    )


@admin.register(DocumentModel)
class DocumentAdmin(ModelAdmin):
    list_display = (
        "id",
        "__str__",
    )