Поменял модели операций с JSON на полиморфные
This commit is contained in:
@@ -1,118 +1,159 @@
|
||||
from django.contrib import admin
|
||||
from mptt.admin import MPTTModelAdmin, DraggableMPTTAdmin
|
||||
from .models import Item, BOMNode, EntityType, RoutingStep, WorkCenter
|
||||
from mptt.admin import DraggableMPTTAdmin
|
||||
from polymorphic.admin import (
|
||||
PolymorphicChildModelAdmin,
|
||||
PolymorphicParentModelAdmin,
|
||||
PolymorphicChildModelFilter,
|
||||
PolymorphicInlineSupportMixin,
|
||||
StackedPolymorphicInline
|
||||
)
|
||||
from .models import (
|
||||
Item, BOMNode, EntityType, WorkCenter,
|
||||
BaseOperation, LaserCutSheet, LaserCutTube,
|
||||
Turning, Weld, Paint, Coating
|
||||
)
|
||||
from django import forms
|
||||
|
||||
# Register your models here.
|
||||
# создаем форму с фильтром для dxf и dwg файлов
|
||||
class LaserCutSheetForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = LaserCutSheet
|
||||
fields = '__all__'
|
||||
widgets = {
|
||||
'dxf_file': forms.FileInput(attrs={'accept': '.dxf,.DXF'}),
|
||||
}
|
||||
|
||||
# Создаем инлайн для отображения RoutingStep в Item
|
||||
class RoutingStepInline(admin.TabularInline):
|
||||
model = RoutingStep
|
||||
extra = 1
|
||||
# Поле tech_params в стандартной админке — это просто текстовое поле.
|
||||
# Позже его можно будет "причесать" с помощью JS, чтобы оно выглядело красиво.
|
||||
fields = ('order', 'operation_type', 'work_center', 'drawing_file', 'tech_params')
|
||||
# создаем форму с фильтром для iges и igs файлов
|
||||
class LaserCutTubeForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = LaserCutTube
|
||||
fields = '__all__'
|
||||
widgets = {
|
||||
'iges_file': forms.FileInput(attrs={'accept': '.iges,.IGS'}),
|
||||
}
|
||||
|
||||
|
||||
# Создаем инлайн для отображения BOMNode в Item
|
||||
# --- 1. ИНЛАЙНЫ ДЛЯ ОПЕРАЦИЙ (ДЛЯ КАРТОЧКИ ITEM) ---
|
||||
|
||||
class LaserCutSheetInline(StackedPolymorphicInline.Child):
|
||||
model = LaserCutSheet
|
||||
form = LaserCutSheetForm
|
||||
|
||||
class LaserCutTubeInline(StackedPolymorphicInline.Child):
|
||||
model = LaserCutTube
|
||||
form = LaserCutTubeForm
|
||||
|
||||
class TurningInline(StackedPolymorphicInline.Child):
|
||||
model = Turning
|
||||
|
||||
class WeldInline(StackedPolymorphicInline.Child):
|
||||
model = Weld
|
||||
|
||||
class PaintInline(StackedPolymorphicInline.Child):
|
||||
model = Paint
|
||||
|
||||
class BaseOperationInline(StackedPolymorphicInline):
|
||||
model = BaseOperation
|
||||
child_inlines = (
|
||||
LaserCutSheetInline,
|
||||
LaserCutTubeInline,
|
||||
TurningInline,
|
||||
WeldInline,
|
||||
PaintInline,
|
||||
)
|
||||
|
||||
# --- 2. ДОЧЕРНИЕ АДМИНКИ (НУЖНЫ ДЛЯ РАБОТЫ ПОЛИМОРФИЗМА) ---
|
||||
|
||||
@admin.register(LaserCutSheet)
|
||||
class LaserCutSheetAdmin(PolymorphicChildModelAdmin):
|
||||
base_model = BaseOperation
|
||||
form = LaserCutSheetForm
|
||||
|
||||
@admin.register(LaserCutTube)
|
||||
class LaserCutTubeAdmin(PolymorphicChildModelAdmin):
|
||||
base_model = BaseOperation
|
||||
form = LaserCutTubeForm
|
||||
|
||||
@admin.register(Turning)
|
||||
class TurningAdmin(PolymorphicChildModelAdmin):
|
||||
base_model = BaseOperation
|
||||
|
||||
@admin.register(Weld)
|
||||
class WeldAdmin(PolymorphicChildModelAdmin):
|
||||
base_model = BaseOperation
|
||||
|
||||
@admin.register(Paint)
|
||||
class PaintAdmin(PolymorphicChildModelAdmin):
|
||||
base_model = BaseOperation
|
||||
|
||||
# --- 3. РОДИТЕЛЬСКАЯ АДМИНКА ОПЕРАЦИЙ ---
|
||||
|
||||
@admin.register(BaseOperation)
|
||||
class BaseOperationParentAdmin(PolymorphicParentModelAdmin):
|
||||
base_model = BaseOperation
|
||||
# Здесь указываем именно МОДЕЛИ, а не классы админок
|
||||
child_models = (LaserCutSheet, LaserCutTube, Turning, Weld, Paint)
|
||||
list_filter = (PolymorphicChildModelFilter,)
|
||||
list_display = ('item', 'order', 'polymorphic_ctype')
|
||||
|
||||
# --- 4. ИНЛАЙН ДЛЯ BOM (СОСТАВ ИЗДЕЛИЯ) ---
|
||||
|
||||
class BOMNodeInline(admin.TabularInline):
|
||||
model = BOMNode
|
||||
|
||||
# Указываем fk_key т
|
||||
fk_name = 'parent'
|
||||
|
||||
# Указываем поля для отображения в инлайне
|
||||
# fk_name = 'parent'
|
||||
fields = ('item', 'quantity')
|
||||
|
||||
# Колличество пустых строк в инлайне
|
||||
extra = 3
|
||||
|
||||
# добавим автокомплит для удобства
|
||||
extra = 1
|
||||
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)
|
||||
|
||||
# --- 5. ОСНОВНАЯ АДМИНКА КОМПОНЕНТА (ITEM) ---
|
||||
|
||||
@admin.register(Item)
|
||||
class ItemAdmin(admin.ModelAdmin):
|
||||
class ItemAdmin(PolymorphicInlineSupportMixin, 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}"
|
||||
return f"{obj.designation or ''} {obj.title}".strip()
|
||||
|
||||
inlines = [RoutingStepInline]
|
||||
# def get_queryset(self, request):
|
||||
# qs = super().get_queryset(request)
|
||||
# return qs.filter(entity_type=EntityType.ASSEMBLY)
|
||||
# Добавляем оба инлайна: и состав, и техпроцесс
|
||||
inlines = [BOMNodeInline, BaseOperationInline]
|
||||
|
||||
# --- 6. АДМИНКА ДЛЯ ДЕРЕВА BOM ---
|
||||
|
||||
@admin.register(BOMNode)
|
||||
class BOMNodeAdmin(DraggableMPTTAdmin):
|
||||
# Настройки отображения древовидной структуры
|
||||
mptt_level_indent = 40 # отступ в пикселях
|
||||
|
||||
# Поля для отображения в админке
|
||||
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')
|
||||
return super().get_queryset(request).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:
|
||||
if obj and obj.item.entity_type in [EntityType.UNIT, EntityType.ASSEMBLY, EntityType.COMPLEX]:
|
||||
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']
|
||||
# --- 7. ПРОЧИЕ СПРАВОЧНИКИ ---
|
||||
|
||||
@admin.register(WorkCenter)
|
||||
class WorkCenterAdmin(admin.ModelAdmin):
|
||||
list_display = ('name', 'rate_per_hour')
|
||||
search_fields = ['name']
|
||||
|
||||
@admin.register(Coating)
|
||||
class CoatingAdmin(admin.ModelAdmin):
|
||||
list_display = ('name', 'consumption')
|
||||
Reference in New Issue
Block a user