22 lines
1.1 KiB
Python
22 lines
1.1 KiB
Python
from rest_framework import serializers
|
|
from apps.pm.models import ProductionPlan, SubProductionPlan
|
|
from apps.mtm.serializers import ProcessSimpleSerializer
|
|
|
|
class SubplanGanttSerializer(serializers.ModelSerializer):
|
|
count = serializers.IntegerField(source='main_count')
|
|
count_real = serializers.IntegerField(source='main_count_real')
|
|
count_ok = serializers.IntegerField(source='main_count_ok')
|
|
process_ = ProcessSimpleSerializer(source='process', read_only=True)
|
|
class Meta:
|
|
model = SubProductionPlan
|
|
fields = ['id', 'number', 'start_date', 'end_date', 'count', 'count_real', 'count_ok', 'start_date_real', 'end_date_real', 'process_']
|
|
|
|
class PlanGanttSerializer(serializers.ModelSerializer):
|
|
children = serializers.SerializerMethodField()
|
|
class Meta:
|
|
model = ProductionPlan
|
|
fields = ['id', 'number', 'start_date', 'end_date', 'children', 'count', 'count_real', 'count_ok']
|
|
|
|
def get_children(self, obj):
|
|
subplans = SubProductionPlan.objects.filter(production_plan=obj).order_by('process__number')
|
|
return SubplanGanttSerializer(instance=subplans, many=True).data |