修复用户提交题目的时候,problems_status 中没有对应的 key 导致的错误

This commit is contained in:
virusdefender 2015-09-24 16:01:08 +08:00
parent 82e8ce3ea8
commit bd121c0adc
2 changed files with 4 additions and 0 deletions

View File

@ -51,6 +51,8 @@ class ContestSubmissionAPIView(APIView):
return error_response(u"提交判题任务失败") return error_response(u"提交判题任务失败")
# 修改用户解题状态 # 修改用户解题状态
problems_status = request.user.problems_status 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 problems_status["contest_problems"][str(data["problem_id"])] = 2
request.user.problems_status = problems_status request.user.problems_status = problems_status
request.user.save() request.user.save()

View File

@ -54,6 +54,8 @@ class SubmissionAPIView(APIView):
return error_response(u"提交判题任务失败") return error_response(u"提交判题任务失败")
# 修改用户解题状态 # 修改用户解题状态
problems_status = request.user.problems_status problems_status = request.user.problems_status
if "problems" not in problems_status:
problems_status["problems"] = {}
problems_status["problems"][str(data["problem_id"])] = 2 problems_status["problems"][str(data["problem_id"])] = 2
request.user.problems_status = problems_status request.user.problems_status = problems_status
request.user.save() request.user.save()