Override StockItem create to accept list of results

Author: 1337joe

inventree/stock.py

@@ -11,6 +11,9 @@
 import inventree.report
 
 
+logger = logging.getLogger('inventree')
+
+
 class StockLocation(
     inventree.base.BarcodeMixin,
     inventree.base.MetadataMixin,
@@ -58,6 +61,31 @@ class StockItem(
 
     MODEL_TYPE = 'stockitem'
 
+    @classmethod
+    def create(cls, api, data, **kwargs):
+        """ Override default create method to support multiple object return. """
+
+        cls.checkApiVersion(api)
+
+        # Ensure the pk value is None so an existing object is not updated
+        if cls.getPkField() in data.keys():
+            data.pop(cls.getPkField())
+
+        response = api.post(cls.URL, data, **kwargs)
+
+        if response is None:
+            logger.error("Error creating new object")
+            return None
+
+        if isinstance(response, list):
+            allResponses = []
+            for element in response:
+                allResponses.append(cls(api, data=element))
+            return allResponses
+
+        else:
+            return [cls(api, data=response)]
+
     @classmethod
     def adjustStockItems(cls, api: inventree.api.InvenTreeAPI, method: str, items: list, **kwargs):
         """Perform a generic stock 'adjustment' action.

test/test_api.py

@@ -209,8 +209,7 @@ def test_create_stuff(self):
             'part': p.pk,
             'quantity': 45,
             'notes': 'This is a note',
-
-        })
+        })[0]
 
         self.assertIsNotNone(s)
         self.assertEqual(s.part, p.pk)

test/test_stock.py

@@ -426,6 +426,10 @@ def test_assign_stock(self):
             }
         )
 
+        # Verify a single result was returned
+        self.assertEqual(1, len(assignitem))
+        assignitem = assignitem[0]
+
         # Assign the item
         assignitem.assignStock(customer=customer, notes='Sell on the side')