Enable and test clearing files for a dataset

Author: summerhenson

sasdata/fair_database/data/test/test_dataset.py

@@ -554,6 +554,12 @@ def test_update_dataset_replace_files(self):
         self.public_dataset.files.add(self.file)
         self.public_dataset.files.remove(file)
 
+    def test_update_dataset_clear_files(self):
+        request = self.auth_client1.put("/v1/data/set/1/", data={"files": [""]})
+        self.assertEqual(request.status_code, status.HTTP_200_OK)
+        self.assertEqual(len(DataSet.objects.get(id=1).files.all()), 0)
+        self.public_dataset.files.add(self.file)
+
     # Test that a dataset cannot be updated to be private and unowned
     def test_update_dataset_no_private_unowned(self):
         request1 = self.auth_client1.put("/v1/data/set/2/", data={"current_user": ""})

sasdata/fair_database/data/views.py

@@ -300,8 +300,18 @@ def put(self, request, data_id, version=None):
         serializer = DataSetSerializer(
             db, request.data, context={"request": request}, partial=True
         )
+        clear_files = "files" in request.data and not request.data["files"]
+        if clear_files:
+            data_copy = request.data.copy()
+            data_copy.pop("files")
+            serializer = DataSetSerializer(
+                db, data_copy, context={"request": request}, partial=True
+            )
         if serializer.is_valid(raise_exception=True):
             serializer.save()
+        if clear_files:
+            db.files.clear()
+            db.save()
         data = {"data_id": db.id, "name": db.name, "is_public": db.is_public}
         return Response(data)