优化前台提交页面的显示

submission/serializers.py

@@ -1,6 +1,8 @@
 # coding=utf-8
 from rest_framework import serializers
 
+from account.models import User
+from .models import Submission
 
 
 class CreateSubmissionSerializer(serializers.Serializer):
@@ -8,3 +10,13 @@ class CreateSubmissionSerializer(serializers.Serializer):
     language = serializers.IntegerField()
     code = serializers.CharField(max_length=3000)
 
+
+class SubmissionSerializer(serializers.ModelSerializer):
+    user = serializers.SerializerMethodField("_get_submission_user")
+
+    class Meta:
+        model = Submission
+        fields = ["id", "result", "create_time", "language", "user"]
+
+    def _get_submission_user(self, obj):
+        return User.objects.get(id=obj.user_id).username