Tree exclusion for PartCategory and StockLocation

2025-08-07 20:32:12 +00:00 · 2021-07-21 21:23:30 +10:00
parent bee0a519ef
commit 85a40ec418
2 changed files with 28 additions and 0 deletions
--- a/InvenTree/part/api.py
+++ b/InvenTree/part/api.py
@@ -105,6 +105,20 @@ class CategoryList(generics.ListCreateAPIView):
            except (ValueError, PartCategory.DoesNotExist):
                pass

+        # Exclude PartCategory tree
+        exclude_tree = params.get('exclude_tree', None)
+
+        if exclude_tree is not None:
+            try:
+                cat = PartCategory.objects.get(pk=exclude_tree)
+
+                queryset = queryset.exclude(
+                    pk__in=[c.pk for c in cat.get_descendants(include_self=True)]
+                )
+
+            except (ValueError, PartCategory.DoesNotExist):
+                pass
+
        return queryset

    filter_backends = [