from django.contrib import admin
from roster.models import (
    RollingClass,
    RollingStock,
    RollingStockImage,
    RollingStockDocument,
)


@admin.register(RollingClass)
class RollingClass(admin.ModelAdmin):
    list_display = ("__str__", "type", "company")
    list_filter = ("company", "type__category", "type")
    search_fields = list_display


class RollingStockDocInline(admin.TabularInline):
    model = RollingStockDocument
    min_num = 0
    extra = 0


class RollingStockImageInline(admin.TabularInline):
    model = RollingStockImage
    min_num = 0
    extra = 0
    readonly_fields = ("image_thumbnail",)


@admin.register(RollingStock)
class RollingStockAdmin(admin.ModelAdmin):
    inlines = (RollingStockImageInline, RollingStockDocInline)
    readonly_fields = ("creation_time", "updated_time")
    list_display = (
        "__str__",
        "address",
        "manufacturer",
        "scale",
        "sku",
        "company",
        "country",
    )
    list_filter = (
        "rolling_class__type__category",
        "rolling_class__type",
        "scale",
        "manufacturer",
    )
    search_fields = list_display

    fieldsets = (
        (
            None,
            {
                "fields": (
                    "rolling_class",
                    "road_number",
                    "manufacturer",
                    "scale",
                    "sku",
                    "decoder",
                    "address",
                    "era",
                    "production_year",
                    "purchase_date",
                    "notes",
                    "tags",
                )
            },
        ),
        (
            "Audit",
            {
                "classes": ("collapse",),
                "fields": (
                    "creation_time",
                    "updated_time",
                ),
            },
        ),
    )