Merge 5cbfdf0 into 33bfe74

inventree/base.py

@@ -4,6 +4,8 @@
 import logging
 import json
 
+from . import api as inventree_api
+
 
 INVENTREE_PYTHON_VERSION = "0.8.1"
 
@@ -273,11 +275,58 @@ def __setitem__(self, name, value):
             raise KeyError(f"Key '{name}' does not exist in dataset")
 
 
-class Attachment(InventreeObject):
+class BulkDeleteMixin:
+    """Mixin class for models which support 'bulk deletion'
+
+    - Perform a DELETE operation against the LIST endpoint for the model
+    - Provide a list of items to be deleted, or filters to apply
+
+    Requires API version 58
+    """
+
+    @classmethod
+    def bulkDelete(cls, api: inventree_api.InvenTreeAPI, items=None, filters=None):
+        """Perform bulk delete operation
+
+        Arguments:
+            api: InventreeAPI instance
+            items: Optional list of items (pk values) to be deleted
+            filters: Optional query filters to delete
+
+        Returns:
+            API response object
+        
+        Throws:
+            NotImplementError: The server API version is too old (requires v58)
+            ValueError: Neither items or filters are supplied
+
+        """
+
+        if api.api_version < 58:
+            raise NotImplementedError("bulkDelete requires API version 58 or newer")
+
+        if not items and not filters:
+            raise ValueError("Must supply either 'items' or 'filters' argument")
+
+        data = {}
+
+        if items:
+            data['items'] = items
+
+        if filters:
+            data['filters'] = filters
+
+        return api.delete(
+            cls.URL,
+            json=data,
+        )
+
+
+class Attachment(BulkDeleteMixin, InventreeObject):
     """
     Class representing a file attachment object
     
-    Multiple sub-classes exist, representing various types of attachment models in the database
+    Multiple sub-classes exist, representing various types of attachment models in the database.
     """
 
     # List of required kwargs required for the particular subclass

inventree/company.py

@@ -63,8 +63,11 @@ def createSalesOrder(self, **kwargs):
         )
 
 
-class SupplierPart(inventree.base.InventreeObject):
-    """ Class representing the SupplierPart database model """
+class SupplierPart(inventree.base.BulkDeleteMixin, inventree.base.InventreeObject):
+    """Class representing the SupplierPart database model
+
+    - Implements the BulkDeleteMixin
+    """
 
     URL = 'company/part'
 
@@ -74,9 +77,10 @@ def getPriceBreaks(self):
         return SupplierPriceBreak.list(self._api, part=self.pk)
 
 
-class ManufacturerPart(inventree.base.InventreeObject):
-    """
-    Class representing the ManufacturerPart database model
+class ManufacturerPart(inventree.base.BulkDeleteMixin, inventree.base.InventreeObject):
+    """Class representing the ManufacturerPart database model
+
+    - Implements the BulkDeleteMixin
     """
 
     URL = 'company/part/manufacturer'
@@ -102,9 +106,10 @@ def uploadAttachment(self, attachment, comment=''):
         )
 
 
-class ManufacturerPartParameter(inventree.base.InventreeObject):
-    """
-    Class representing the ManufacturerPartParameter database model.
+class ManufacturerPartParameter(inventree.base.BulkDeleteMixin, inventree.base.InventreeObject):
+    """Class representing the ManufacturerPartParameter database model.
+
+    - Implements the BulkDeleteMixin
     """
 
     URL = 'company/part/manufacturer/parameter'

test/test_build.py

@@ -72,3 +72,47 @@ def test_build_attachment(self):
         self.assertEqual(response['comment'], 'A self referencing upload!')
 
         self.assertEqual(len(build.getAttachments()), n + 1)
+
+    def test_build_attachment_bulk_delete(self):
+        """Test 'bulk delete' operation for the BuildAttachment class"""
+
+        build = self.get_build()
+
+        n = len(BuildAttachment.list(self.api, build=build.pk))
+
+        fn = os.path.join(os.path.dirname(__file__), 'test_build.py')
+
+        pk_values = []
+
+        # Create a number of new attachments
+        for i in range(10):
+            response = build.uploadAttachment(fn, comment=f"Build attachment {i}")
+            pk_values.append(response['pk'])
+
+        self.assertEqual(len(BuildAttachment.list(self.api, build=build.pk)), n + 10)
+
+        # Call without providing required arguments
+        with self.assertRaises(ValueError):
+            BuildAttachment.bulkDelete(self.api)
+
+        BuildAttachment.bulkDelete(self.api, items=pk_values)
+
+        # The number of attachments has been reduced to the original value
+        self.assertEqual(len(BuildAttachment.list(self.api, build=build.pk)), n)
+
+        # Now, delete using the 'filters' argument
+        for i in range(99, 109):
+            response = build.uploadAttachment(fn, comment=f"Build attachment {i}")
+            pk_values.append(response['pk'])
+
+        self.assertEqual(len(BuildAttachment.list(self.api, build=build.pk)), n + 10)
+
+        response = BuildAttachment.bulkDelete(
+            self.api,
+            filters={
+                "build": build.pk,
+            }
+        )
+
+        # All attachments for this Build should have been deleted
+        self.assertEqual(len(BuildAttachment.list(self.api, build=build.pk)), 0)