From bd121c0adcfac2d1210bbd365e08647c6e8174a9 Mon Sep 17 00:00:00 2001
From: virusdefender <1670873886@qq.com>
Date: Thu, 24 Sep 2015 16:01:08 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E7=94=A8=E6=88=B7=E6=8F=90?=
 =?UTF-8?q?=E4=BA=A4=E9=A2=98=E7=9B=AE=E7=9A=84=E6=97=B6=E5=80=99=EF=BC=8C?=
 =?UTF-8?q?problems=5Fstatus=20=E4=B8=AD=E6=B2=A1=E6=9C=89=E5=AF=B9?=
 =?UTF-8?q?=E5=BA=94=E7=9A=84=20key=20=E5=AF=BC=E8=87=B4=E7=9A=84=E9=94=99?=
 =?UTF-8?q?=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 contest_submission/views.py | 2 ++
 submission/views.py         | 2 ++
 2 files changed, 4 insertions(+)

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()