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

add get problem by id

Browse Source
This commit is contained in:
virusdefender 2017-02-03 17:15:53 +08:00
parent 8ce6040a5b
commit e15286aa1a
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
problem/views/admin.py
View File

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