Fixed #864 -- Ensuring we don't use the MP_NodeQuerySet for the Produ…

…cts (which don't make use of mp_tree)

src/open_inwoner/pdc/managers.py

@@ -1,7 +1,17 @@
+from django.db import models
+
 from treebeard.mp_tree import MP_NodeQuerySet
 
 
-class PublishedQueryset(MP_NodeQuerySet):
+class PublishedQueryset(models.QuerySet):
+    def published(self):
+        return self.filter(published=True)
+
+    def draft(self):
+        return self.filter(published=False)
+
+
+class CategoryPublishedQueryset(MP_NodeQuerySet):
     def published(self):
         return self.filter(published=True)
 

src/open_inwoner/pdc/models/category.py

@@ -6,7 +6,7 @@
 from treebeard.exceptions import InvalidMoveToDescendant
 from treebeard.mp_tree import MP_MoveHandler, MP_Node
 
-from ..managers import PublishedQueryset
+from ..managers import CategoryPublishedQueryset
 
 
 class PublishedMoveHandler(MP_MoveHandler):
@@ -62,7 +62,7 @@ class Category(MP_Node):
     )
 
     node_order_by = ["slug"]
-    objects = PublishedQueryset.as_manager()
+    objects = CategoryPublishedQueryset.as_manager()
 
     class Meta:
         verbose_name = _("Category")