diff --git a/src/open_inwoner/pdc/managers.py b/src/open_inwoner/pdc/managers.py index 6fe65247dc..61b0986924 100644 --- a/src/open_inwoner/pdc/managers.py +++ b/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) diff --git a/src/open_inwoner/pdc/models/category.py b/src/open_inwoner/pdc/models/category.py index 3b982aa077..8908761606 100644 --- a/src/open_inwoner/pdc/models/category.py +++ b/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")