Reflect single submission in serializer.

angstromctf · Jul 26, 2017 · ce04f8c · ce04f8c
1 parent a84be9a
commit ce04f8c
Showing 1 changed file with 3 additions and 11 deletions.
diff --git a/api/serializers.py b/api/serializers.py
@@ -55,14 +55,6 @@ class Meta:
         fields = ('code',)
 
 
-class SubmissionSerializer(serializers.ModelSerializer):
-    problem = ProblemSerializer()
-
-    class Meta:
-        model = models.Submission
-        fields = ('problem', 'time')
-
-
 class UserSerializer(serializers.ModelSerializer):
     class Meta:
         model = models.User
@@ -89,17 +81,17 @@ class Meta:
         fields = ('id', 'name', 'school', 'score', 'score_lastupdate', 'eligible')
 
 
-class CorrectSubmissionSerializer(serializers.ModelSerializer):
+class SubmissionSerializer(serializers.ModelSerializer):
     problem = ProblemSerializer()
     team = TeamSerializer()
 
     class Meta:
         model = models.Submission
-        fields = ('problem', 'team', 'new_score', 'time', 'problem', 'team')
+        fields = ('problem', 'team', 'new_score', 'time', 'problem', 'team', 'correct')
 
 
 class TeamProfileSerializer(serializers.ModelSerializer):
-    solves = CorrectSubmissionSerializer(many=True, read_only=True)
+    solves = SubmissionSerializer(many=True, read_only=True)
     members = ProfileSerializer(many=True, read_only=True)
     place = serializers.SerializerMethodField()