inspecttask增加template_name字段

This commit is contained in:
caoqianming 2021-04-28 11:19:34 +08:00
parent b9204670aa
commit b944e38335
1 changed files with 7 additions and 0 deletions

View File

@ -1,3 +1,4 @@
from re import template
from django.db.models.base import Model from django.db.models.base import Model
from rest_framework import serializers from rest_framework import serializers
from .models import * from .models import *
@ -30,6 +31,7 @@ class InspectTaskSerializer(serializers.ModelSerializer):
create_by_ = UserSimpleSerializer(source='create_by', read_only=True) create_by_ = UserSimpleSerializer(source='create_by', read_only=True)
dept_count = serializers.SerializerMethodField() dept_count = serializers.SerializerMethodField()
subtask_count = serializers.SerializerMethodField() subtask_count = serializers.SerializerMethodField()
template_name = serializers.CharField(source='template.name', read_only=True)
class Meta: class Meta:
model = InspectTask model = InspectTask
@ -41,6 +43,11 @@ class InspectTaskSerializer(serializers.ModelSerializer):
def get_subtask_count(self, obj): def get_subtask_count(self, obj):
return SubInspectTask.objects.filter(inspecttask=obj, is_deleted=False).count() return SubInspectTask.objects.filter(inspecttask=obj, is_deleted=False).count()
@staticmethod
def setup_eager_loading(queryset):
queryset = queryset.select_related('template',)
return queryset
class InspectTaskSimpleSerializer(serializers.ModelSerializer): class InspectTaskSimpleSerializer(serializers.ModelSerializer):
class Meta: class Meta:
model = InspectTask model = InspectTask