Initial commit

bom_manager/admin.py

@@ -0,0 +1,118 @@
+from django.contrib import admin
+from mptt.admin import MPTTModelAdmin, DraggableMPTTAdmin
+from .models import Item, BOMNode, EntityType, RoutingStep, WorkCenter
+
+# Register your models here.
+
+# Создаем инлайн для отображения RoutingStep в Item
+class RoutingStepInline(admin.TabularInline):
+    model = RoutingStep    
+    extra = 1
+    # Поле tech_params в стандартной админке — это просто текстовое поле.
+    # Позже его можно будет "причесать" с помощью JS, чтобы оно выглядело красиво.
+    fields = ('order', 'operation_type', 'work_center', 'drawing_file', 'tech_params')
+
+
+# Создаем инлайн для отображения BOMNode в Item
+class BOMNodeInline(admin.TabularInline):
+    model = BOMNode
+    
+    # Указываем fk_key т
+    fk_name = 'parent'
+
+    # Указываем поля для отображения в инлайне
+    fields = ('item', 'quantity')
+
+    # Колличество пустых строк в инлайне
+    extra = 3
+
+    # добавим автокомплит для удобства
+    autocomplete_fields = ['item']
+
+    def formfield_for_foreignkey(self, db_field, request, **kwargs):
+        if db_field.name == "parent":
+            # Родителями в инлайне могут быть только Сборки/Изделия/Комплексы
+            kwargs["queryset"] = BOMNode.objects.filter(
+                item__entity_type__in=[EntityType.UNIT, EntityType.ASSEMBLY, EntityType.COMPLEX]
+            )
+        return super().formfield_for_foreignkey(db_field, request, **kwargs)
+
+@admin.register(Item)
+class ItemAdmin(admin.ModelAdmin):
+    list_display = ('get_full_name', 'entity_type', 'drawing')
+    list_filter = ('entity_type', 'is_assembly')
+    # Поиск по номеру и названию
+    search_fields = ['designation', 'title']
+
+    @admin.display(description='Наименование')
+    def get_full_name(self, obj):
+        return f"{obj.designation or ''} {obj.title}"
+    
+    inlines = [RoutingStepInline]
+    # def get_queryset(self, request):
+    #     qs = super().get_queryset(request)
+    #     return qs.filter(entity_type=EntityType.ASSEMBLY)
+
+@admin.register(BOMNode)
+class BOMNodeAdmin(DraggableMPTTAdmin):
+    # Настройки отображения древовидной структуры
+    mptt_level_indent = 40 # отступ в пикселях
+    
+    # Поля для отображения в админке
+    fields = ('parent', 'item', 'quantity')
+
+# Чтобы в списке было видно не только название связи, но и данные из Item
+    list_display = ('tree_actions', 'indented_title', 'get_designation', 'quantity')
+    list_display_links = ('indented_title',)
+
+    # Отображаем инлайн.
+    inlines = [BOMNodeInline]
+
+    # Метод для отображения децимального номера из связанной модели Item
+    def get_designation(self, obj):
+        return obj.item.designation
+    get_designation.short_description = 'Децимальный номер'
+
+    # Оптимизация запросов (чтобы админка не тормозила)
+    def get_queryset(self, request):
+        qs = super().get_queryset(request)
+        return qs.select_related('item', 'parent')
+
+    # def formfield_for_foreignkey(self, db_field, request, **kwargs):
+    #     if db_field.name == "parent":
+    #         # Фильтруем: родителями могут быть только узлы, 
+    #         # где связанный Item является сборкой или изделием
+    #         kwargs["queryset"] = BOMNode.objects.filter(
+    #             item__entity_type__in=[EntityType.UNIT, EntityType.ASSEMBLY]
+    #         )
+    #     return super().formfield_for_foreignkey(db_field, request, **kwargs)
+    
+
+    # переопределим метод get_inlines в BOMNodeAdmin. 
+    # Если узел привязан к «Детали» или «Стандартному изделию», таблица внизу просто не появится.
+    def get_inlines(self, request, obj=None):
+        # Если мы создаем новый узел (obj is None), инлайн можно показать
+        if obj is None:
+            return []
+        
+        # Разрешенные типы (те, в которые МОЖНО вкладывать)
+        allowed_types = [EntityType.UNIT, EntityType.ASSEMBLY, EntityType.COMPLEX]
+        
+        # Если тип изделия связанного узла в списке разрешенных — показываем инлайн
+        if obj.item.entity_type in allowed_types:
+            return [BOMNodeInline]
+        
+        # Для остальных (детали, стандартные) возвращаем пустой список
+        return []
+
+@admin.register(RoutingStep)  
+class RoutingStepAdmin(admin.ModelAdmin):
+    # Поля для отображения в админке
+    list_display = ('item', 'operation_type', 'work_center', 'order')
+    list_filter = ('operation_type', 'work_center')
+    search_fields = ['item__title']
+
+@admin.register(WorkCenter)
+class WorkCenterAdmin(admin.ModelAdmin):
+    list_display = ('name', 'rate_per_hour')
+    search_fields = ['name']