extract problem description

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

Author: ZTL-UwU

problem/serializers.py

@@ -6,6 +6,7 @@
 from segmentoj import tools
 
 class ProblemSerializer(serializers.ModelSerializer):
+
     class Meta:
         model = Problem
         fields = [
@@ -22,6 +23,21 @@ class Meta:
 
         depth = 0
         read_only_fields = ["id", "date_added"]
+        extra_kwargs = {
+            "description": {"write_only": True},
+        }
+
+class ProblemDescriptionSerializer(serializers.ModelSerializer):
+
+    class Meta:
+        model = Problem
+        fields = [
+            "pid", 
+            "description", 
+        ]
+
+        depth = 0
+        read_only_fields = ["id"]
 
 class TagSerializer(serializers.ModelSerializer):
 

problem/tests.py

@@ -4,7 +4,7 @@
 from rest_framework.test import APIRequestFactory, force_authenticate
 
 from .models import Problem
-from .views import ProblemView, TagView, TagListView
+from .views import ProblemView, ProblemDescriptionView, TagView, TagListView
 from account.models import User
 
 # Create your tests here.
@@ -21,7 +21,6 @@ def setUp(self):
     def testZ_get_problem(self):
         ac_data = {
             "title": "Simple Problem",
-            "description": "This is description",
             "pid": 5,
             "allow_html": False,
             "enabled": True,
@@ -38,7 +37,6 @@ def testZ_get_problem(self):
         self.assertIsNotNone(data)
 
         self.assertEqual(data.get("title"), ac_data["title"])
-        self.assertEqual(data.get("description"), ac_data["description"])
         self.assertEqual(data.get("pid"), ac_data["pid"])
         self.assertEqual(data.get("allow_html"), ac_data["allow_html"])
         self.assertEqual(data.get("enabled"), ac_data["enabled"])
@@ -87,6 +85,28 @@ def testW_change_problem(self):
         self.assertEqual(target.enabled, ac_data["enabled"])
 
 
+class ProblemDescriptionViewTest(TestCase):
+    fixtures = ["testdatabase.yaml"]
+
+    def setUp(self):
+        self.base_url = "/api/problem/{pid}/description"
+        self.factory = APIRequestFactory()
+        self.view = ProblemDescriptionView.as_view()
+
+    def testZ_get_problem_description(self):
+        ac_data = {
+            "description": "This is description"
+        }
+
+        request = self.factory.get(self.base_url.format(pid=5))
+        response = self.view(request, pid=5)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        data = response.data.get("res")
+        self.assertIsNotNone(data)
+        self.assertEqual(data.get("description"), ac_data["description"])
+
+
 class TagViewTest(TestCase):
     fixtures = ["testdatabase.yaml"]
 

problem/views.py

@@ -10,7 +10,12 @@
 
 from segmentoj import tools
 from problem.models import Problem, Tag
-from .serializers import ProblemSerializer, ProblemListSerializer, TagSerializer
+from .serializers import (
+    ProblemSerializer,
+    ProblemDescriptionSerializer,
+    ProblemListSerializer,
+    TagSerializer,
+)
 from segmentoj.decorator import syllable_required, parameter_required
 from status.models import Status
 
@@ -60,6 +65,20 @@ def delete(self, request, pid):
         return Response(status=status.HTTP_204_NO_CONTENT)
 
 
+class ProblemDescriptionView(APIView):
+    @method_decorator(parameter_required("pid"))
+    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)
+
+
 class TagView(APIView):
     @method_decorator(parameter_required("tid"))
     def get(self, request, tid):
@@ -126,7 +145,6 @@ def process_data(x):
             status=status.HTTP_200_OK,
         )
 
-
 class ProblemListCountView(APIView):
     def get(self, request):
 

segmentoj/urls.py

@@ -52,6 +52,7 @@
     # Problem
     path("api/problem", problem.views.ProblemView.as_view()),
     path("api/problem/<int:pid>", problem.views.ProblemView.as_view()),
+    path("api/problem/<int:pid>/description", problem.views.ProblemDescriptionView.as_view()),
     path("api/problem/tag", problem.views.TagView.as_view()),
     path("api/problem/tag/<int:tid>", problem.views.TagView.as_view()),
     path("api/problem/list", problem.views.ProblemListView.as_view()),