From d09844a67bea182b3afafd07899dec1bee70d30e Mon Sep 17 00:00:00 2001 From: Ramon Wenger Date: Tue, 5 Apr 2022 13:16:09 +0200 Subject: [PATCH] Remove content blocks belonging to snapshots from CMS explorer --- server/books/wagtail_hooks.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/server/books/wagtail_hooks.py b/server/books/wagtail_hooks.py index 3e8dde70..feb691fd 100644 --- a/server/books/wagtail_hooks.py +++ b/server/books/wagtail_hooks.py @@ -1,11 +1,10 @@ from wagtail.core import hooks -from .models import ContentBlock +from .models import ContentBlockSnapshot @hooks.register('construct_explorer_page_queryset') def remove_user_created_content_blocks_from_menu(parent_page, pages, request): if parent_page.content_type.model == 'chapter': - content_block_ids = ContentBlock.get_by_parent(parent_page.specific).exclude(user_created=True).values_list('id', flat=True) - return pages.filter(id__in=content_block_ids) + return pages.not_type(ContentBlockSnapshot).exclude(contentblock__user_created=True) return pages