Add problem volunteer

judge/views/volunteer.py

@@ -0,0 +1,33 @@
+from django.http import HttpResponseBadRequest, JsonResponse
+from django.db import transaction
+
+from judge.models import VolunteerProblemVote, Problem, ProblemType
+
+
+def vote_problem(request):
+    if not request.user or not request.user.has_perm('judge.suggest_problem_changes'):
+        return HttpResponseBadRequest()
+    if not request.method == 'POST':
+        return HttpResponseBadRequest()
+    try:
+        types_id = request.POST.getlist('types[]')
+        types = ProblemType.objects.filter(id__in=types_id)
+        problem = Problem.objects.get(code=request.POST['problem'])
+        knowledge_points = request.POST['knowledge_points']
+        thinking_points = request.POST['thinking_points']
+        feedback = request.POST['feedback']
+    except Exception as e:
+        return HttpResponseBadRequest()
+
+    with transaction.atomic():
+        vote, _ = VolunteerProblemVote.objects.get_or_create(
+            voter=request.profile,
+            problem=problem,
+            defaults={'knowledge_points': 0, 'thinking_points': 0},
+        )
+        vote.knowledge_points = knowledge_points
+        vote.thinking_points = thinking_points
+        vote.feedback = feedback
+        vote.types.set(types)
+        vote.save()
+    return JsonResponse({})