From e15286aa1a9097c0512a0fe2d610598aca065141 Mon Sep 17 00:00:00 2001 From: virusdefender Date: Fri, 3 Feb 2017 17:15:53 +0800 Subject: [PATCH] add get problem by id --- problem/views/admin.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/problem/views/admin.py b/problem/views/admin.py index c44d4ff..77946d2 100644 --- a/problem/views/admin.py +++ b/problem/views/admin.py @@ -145,6 +145,16 @@ class ProblemAPI(APIView): return self.success() def get(self, request): + problem_id = request.GET.get("id") + if problem_id: + try: + problem = Problem.objects.get(id=problem_id) + if request.user.is_admin_role(): + problem = problem.get(created_by=request.user) + return self.success(ProblemSerializer(problem).data) + except Problem.DoesNotExist: + return self.error("Problem does not exist") + problems = Problem.objects.all().order_by("-create_time") if request.user.is_admin_role(): problems = problems.filter(created_by=request.user)