fix problem put method

Author: zdszero

problem/views/admin.py

@@ -1,3 +1,5 @@
+import os
+import shutil
 import json
 import zipfile
 
@@ -10,21 +12,16 @@
 from utils.api import APIView, validate_serializer
 from utils.shortcuts import rand_str
 from utils.tasks import delete_files
-from judge.testing import ZipFileUploader, create_new_problem_from_template
+from judge.testing import ZipFileUploader, create_new_problem_from_template, PathManager
 
 from ..models import Problem, ProblemTag
 from ..serializers import *
 
 class ProblemBase(APIView):
     def common_checks(self, request):
         data = request.data
-        
-
-class ProblemFormBase(APIView):
-    # use View's dispatch function
-    request_parsers = None
 
-class ProblemAPI(ProblemFormBase):
+class ProblemAPI(ProblemBase):
     # @validate_serializer(CreateProblemSerializer)
     @problem_permission_required
     def post(self, request):
@@ -128,9 +125,10 @@ def delete(self, request):
         except Problem.DoesNotExist:
             return self.error("Problem does not exists")
         ensure_created_by(problem, request.user)
-        # d = os.path.join(settings.TEST_CASE_DIR, problem.test_case_id)
-        # if os.path.isdir(d):
-        #     shutil.rmtree(d, ignore_errors=True)
+        d = PathManager.problem_dir(problem._id)
+        if os.path.isdir(d):
+            print('remove dir', d)
+            shutil.rmtree(d, ignore_errors=True)
         problem.delete()
         return self.success()
 

utils/api/api.py

@@ -32,7 +32,6 @@ class JSONParser(object):
     def parse(body):
         return json.loads(body.decode("utf-8"))
 
-
 class URLEncodedParser(object):
     content_type = ContentType.url_encoded_request
 
@@ -141,11 +140,12 @@ def paginate_data(self, request, query_set, object_serializer=None):
         return data
 
     def dispatch(self, request, *args, **kwargs):
-        if self.request_parsers:
-            try:
-                request.data = self._get_request_data(self.request)
-            except ValueError as e:
-                return self.error(err="invalid-request", msg=str(e))
+        if not (request.content_type and request.content_type.startswith(ContentType.form_data_request)):
+            if self.request_parsers:
+                try:
+                    request.data = self._get_request_data(self.request)
+                except ValueError as e:
+                    return self.error(err="invalid-request", msg=str(e))
         try:
             return super(APIView, self).dispatch(request, *args, **kwargs)
         except APIError as e: