avoid duplication

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 enabled_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/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 enabled_required
 
 
 class ProblemView(APIView):
     @method_decorator(parameter_required("pid"))
+    @method_decorator(enabled_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(enabled_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)