change user profile on submit

Author: zdszero

account/models.py

@@ -70,33 +70,15 @@ class Meta:
 
 class UserProfile(models.Model):
     user = models.OneToOneField(User, on_delete=models.CASCADE)
-    # Task_problems_status examples:
-    # {
-    #     "problems": {
-    #         "1": {
-    #             "status": JudgeStatus.ACCEPTED,
-    #             "Score:" 80,
-    #             "passed test case": [1, 2],
-    #             "_id": "1000"
-    #         }
-    #     },
-    #     "contest_problems": {
-    #         "1": {
-    #             "status": JudgeStatus.Error,
-    #             "Score:" 80,
-    #             "passed test case": [1, 2],
-    #             "_id": "1000"
-    #         }
-    #     }
-    # }
+    # dict of problem.id:max_grade
     problems_status = JSONField(default=dict)
     real_name = models.TextField(null=True)
     blog = models.URLField(null=True)
     github = models.TextField(null=True)
     school = models.TextField(null=True)
     major = models.TextField(null=True)
     language = models.TextField(null=True)
-    #for Contest
+    # for Contest
     accepted_number = models.IntegerField(default=0)
     total_score = models.IntegerField(default=0)
 
@@ -109,4 +91,4 @@ def add_score(self, this_time_score, last_time_score=None):
         self.total_score = models.F("total_score") - last_time_score + this_time_score
         self.save()
     class Meta:
-        db_table = "user_profile"
+        db_table = "user_profile"

submission/views/user.py

@@ -73,7 +73,6 @@ def check_contest_permission(self, request):
     def post(self, request):
         # print(request.data)
         data = request.data
-        hide_id = False
 
         error = self.throttling(request)
         if error:
@@ -133,10 +132,26 @@ def post(self, request):
             self.error(str(e))
         problem.save()
 
-        if hide_id:
-            return self.success()
+        score = submission.grade
+        user = User.objects.get(id=request.user.id)
+        assert user
+        profile = UserProfile.objects.get(user=user)
+        assert profile
+        if problem._id not in profile.problems_status:
+            profile.problems_status[problem._id] = score
+            profile.total_score += score
+            if score == 100:
+                profile.accepted_number += 1
         else:
-            return self.success(SubmissionModelSerializer(submission).data)
+            prev_score = profile.problems_status[problem._id]
+            if score > prev_score:
+                profile.problems_status[problem._id] = score
+                profile.total_score += (score - prev_score)
+                if score == 100:
+                    profile.accepted_number += 1
+        profile.save()
+
+        return self.success(SubmissionModelSerializer(submission).data)
 
     @login_required
     def get(self, request):