fix comma

Signed-off-by: zhangtianli2006 <zhangtianli2006@163.com>

Author: ZTL-UwU

problem/decorator.py

@@ -0,0 +1,21 @@
+from rest_framework.response import Response
+from rest_framework import status
+from django.shortcuts import get_object_or_404
+from problem.models import Problem
+
+
+def view_hidden_problem_permission_required():
+    def decorator(func):
+        def _wrapped_view(request, pid, *args, **kwargs):
+            problem = get_object_or_404(Problem, pid=pid)
+
+            if not problem.enabled and not request.user.has_perm("problem.view_hidden"):
+                return Response(
+                    {"detail": "Problem is hidden."}, status=status.HTTP_403_FORBIDDEN
+                )
+
+            return func(request, pid=pid, *args, **kwargs)
+
+        return _wrapped_view
+
+    return decorator

problem/serializers.py

@@ -10,14 +10,14 @@ class ProblemSerializer(serializers.ModelSerializer):
     class Meta:
         model = Problem
         fields = [
-            "pid", 
-            "date_added", 
-            "title", 
+            "pid",
+            "date_added",
+            "title",
             "description",
-            "allow_html", 
-            "tags", 
-            "enabled", 
-            "memory_limit", 
+            "allow_html",
+            "tags",
+            "enabled",
+            "memory_limit",
             "time_limit",
         ]
 
@@ -32,8 +32,8 @@ class ProblemDescriptionSerializer(serializers.ModelSerializer):
     class Meta:
         model = Problem
         fields = [
-            "pid", 
-            "description", 
+            "pid",
+            "description",
         ]
 
         depth = 0

problem/views.py

@@ -16,21 +16,21 @@
     ProblemListSerializer,
     TagSerializer,
 )
-from segmentoj.decorator import syllable_required, parameter_required
+from segmentoj.decorator import (
+    syllable_required,
+    parameter_required,
+)
 from status.models import Status
+from .decorator import view_hidden_problem_permission_required
 
 
 class ProblemView(APIView):
     @method_decorator(parameter_required("pid"))
+    @method_decorator(view_hidden_problem_permission_required())
     def get(self, request, pid):
         # Get the content of a problem
         problem = get_object_or_404(Problem, pid=pid)
 
-        if not problem.enabled and not request.user.has_perm("problem.view_hidden"):
-            return Response(
-                {"detail": "Problem is hidden."}, status=status.HTTP_403_FORBIDDEN
-            )
-
         ps = ProblemSerializer(problem)
         return Response({"res": ps.data}, status=status.HTTP_200_OK)
 
@@ -67,14 +67,10 @@ def delete(self, request, pid):
 
 class ProblemDescriptionView(APIView):
     @method_decorator(parameter_required("pid"))
+    @method_decorator(view_hidden_problem_permission_required())
     def get(self, request, pid):
         problem = get_object_or_404(Problem, pid=pid)
 
-        if not problem.enabled and not request.user.has_perm("problem.view_hidden"):
-            return Response(
-                {"detail": "Problem is hidden."}, status=status.HTTP_403_FORBIDDEN
-            )
-
         pds = ProblemDescriptionSerializer(problem)
         return Response({"res": pds.data}, status=status.HTTP_200_OK)