Merge pull request #150 from inventree/count

Adds "count" functionality for each API endpoint

inventree/base.py

@@ -149,9 +149,22 @@ def create(cls, api, data, **kwargs):
 
         return cls(api, data=response)
 
+    @classmethod
+    def count(cls, api, **kwargs):
+        """Return a count of all items of this class in the database"""
+
+        params = kwargs
+
+        # By limiting to a single result, we perform a fast query, but get a total number of results
+        params['limit'] = 1
+
+        response = api.get(url=cls.URL, params=params, **kwargs)
+
+        return response['count']
+
     @classmethod
     def list(cls, api, **kwargs):
-        """ Return a list of all items in this class on the database.
+        """Return a list of all items in this class on the database.
 
         Requires:
 

test/test_part.py

@@ -32,6 +32,9 @@ def test_part_cats(self):
         n = len(cats)
         self.assertTrue(len(cats) >= 9)
 
+        # Check that the 'count' method returns the same result
+        self.assertEqual(n, PartCategory.count(self.api))
+
         # Filtered categories must be fewer than *all* categories
         cats = PartCategory.list(self.api, parent=1)