skillbox/server/books/wagtail_hooks.py

11 lines
358 B
Python

from wagtail.core import hooks
from .models import ContentBlock
@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':
return ContentBlock.get_by_parent(parent_page.specific).exclude(user_created=True)
return pages