from django.contrib import admin

from .models import BOM, ProductEntity, RouteStub


@admin.register(RouteStub)
class RouteStubAdmin(admin.ModelAdmin):
    list_display = ('name',)
    search_fields = ('name',)


class BOMChildInline(admin.TabularInline):
    """Состав изделия/сборки (строки BOM) прямо в карточке ProductEntity."""

    model = BOM
    fk_name = 'parent'
    fields = ('child', 'quantity')
    autocomplete_fields = ('child',)
    extra = 10


@admin.register(ProductEntity)
class ProductEntityAdmin(admin.ModelAdmin):
    list_display = (
        'drawing_number',
        'name',
        'entity_type',
        'planned_material',
        'blank_area_m2',
        'blank_length_mm',
    )
    list_filter = ('entity_type', 'planned_material__category')
    search_fields = ('drawing_number', 'name', 'planned_material__name', 'planned_material__full_name')
    autocomplete_fields = ('planned_material', 'route')
    inlines = (BOMChildInline,)


@admin.register(BOM)
class BOMAdmin(admin.ModelAdmin):
    list_display = ('parent', 'child', 'quantity')
    search_fields = ('parent__name', 'parent__drawing_number', 'child__name', 'child__drawing_number')
    list_filter = ('parent',)
    autocomplete_fields = ('parent', 'child')