作者 wanweibin

update

@@ -4,33 +4,25 @@ @@ -4,33 +4,25 @@
4 from rest_framework import serializers 4 from rest_framework import serializers
5 5
6 from .models import Auditor, Project, Result 6 from .models import Auditor, Project, Result
7 -from usercenter.serializers import UserSerializer  
8 7
9 8
10 class AuditorSerializer(serializers.ModelSerializer): 9 class AuditorSerializer(serializers.ModelSerializer):
11 10
12 class Meta: 11 class Meta:
13 model = Auditor 12 model = Auditor
14 - # fields = ("user", "leader")  
15 fields = '__all__' 13 fields = '__all__'
16 depth = 1 14 depth = 1
17 15
18 16
19 class ProjectSerializer(serializers.ModelSerializer): 17 class ProjectSerializer(serializers.ModelSerializer):
20 - # creator = serializers.ReadOnlyField(source='project_creator')  
21 - # teacher = serializers.ReadOnlyField(source='teacher.username') # 外键字段 只读  
22 - # result = serializers.PrimaryKeyRelatedField()  
23 result = serializers.ReadOnlyField() 18 result = serializers.ReadOnlyField()
24 -  
25 - # @staticmethod  
26 - # def get_result(obj):  
27 - # result = item.result_project.filter(auditor__user_id=self.request.user.id).values('is_accept').first() 19 + cur_auditor = serializers.ReadOnlyField()
28 20
29 class Meta: 21 class Meta:
30 model = Project 22 model = Project
31 fields = ( 23 fields = (
32 'id', 'result', 'creator', 'category', 'model_type', 'market_share_analysis', 'context_analysis', 24 'id', 'result', 'creator', 'category', 'model_type', 'market_share_analysis', 'context_analysis',
33 - 'attachments', 'create_time' 25 + 'attachments', 'create_time', "cur_auditor"
34 ) 26 )
35 depth = 1 27 depth = 1
36 28
@@ -54,6 +54,13 @@ class ProjectDetail(RetrieveAPIView): @@ -54,6 +54,13 @@ class ProjectDetail(RetrieveAPIView):
54 54
55 def get(self, request, *args, **kwargs): 55 def get(self, request, *args, **kwargs):
56 obj = self.get_object() 56 obj = self.get_object()
  57 + auditor_len = Result.objects.filter(project=obj).count()
  58 + if auditor_len == 2:
  59 + obj.cur_auditor = "已完结"
  60 + elif auditor_len == 1:
  61 + obj.cur_auditor = AUDITORS[1][1]
  62 + else:
  63 + obj.cur_auditor = AUDITORS[0][1]
57 return response(ProjectSerializer(obj).data) 64 return response(ProjectSerializer(obj).data)
58 65
59 66