diff --git a/contest_submission/views.py b/contest_submission/views.py index 8611ff48..2d35e79b 100644 --- a/contest_submission/views.py +++ b/contest_submission/views.py @@ -51,6 +51,8 @@ class ContestSubmissionAPIView(APIView): return error_response(u"提交判题任务失败") # 修改用户解题状态 problems_status = request.user.problems_status + if "contest_problems" not in problems_status: + problems_status["contest_problems"] = {} problems_status["contest_problems"][str(data["problem_id"])] = 2 request.user.problems_status = problems_status request.user.save() diff --git a/submission/views.py b/submission/views.py index 7a4d7ea1..06f14039 100644 --- a/submission/views.py +++ b/submission/views.py @@ -54,6 +54,8 @@ class SubmissionAPIView(APIView): return error_response(u"提交判题任务失败") # 修改用户解题状态 problems_status = request.user.problems_status + if "problems" not in problems_status: + problems_status["problems"] = {} problems_status["problems"][str(data["problem_id"])] = 2 request.user.problems_status = problems_status request.user.save()