QingdaoU/OnlineJudge
1
0
Fork 0
mirror of https://github.com/QingdaoU/OnlineJudge.git synced 2024-09-21 00:13:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix error in refresh displayID

Browse Source
This commit is contained in:
zema1 2017-11-09 11:21:41 +08:00
parent 343eff1c51
commit 48f65d1a14
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
account/views/oj.py
View File

@ -385,13 +385,19 @@ class UserRankAPI(APIView):
class ProfileProblemDisplayIDRefreshAPI(APIView):
@login_required
def get(self, request):
user_id = request.GET.get("user_id")
if not user_id:
return self.error("Invalid parameter, user_id is required")
if request.user.id != user_id:
return self.error("Only user self can require a refresh")
profile = request.user.userprofile
acm_problems = profile.acm_problems_status["problems"]
oi_problems = profile.oi_problems_status["problems"]
acm_problems = profile.acm_problems_status.get("problems", {})
oi_problems = profile.oi_problems_status.get("problems", {})
ids = list(acm_problems.keys()) + list(oi_problems.keys())
if not ids:
return self.success()
display_ids = Problem.objects.filter(id__in=ids).values_list("_id", flat=True)
id_map = dict(zip(ids, display_ids))
print(id_map)
for k, v in acm_problems.items():
v["_id"] = id_map[k]
for k, v in oi_problems.items():