From 0cbe15b7413de56a758fe8b1acd043656a8339bb Mon Sep 17 00:00:00 2001 From: Karl Hobley Date: Mon, 7 Apr 2014 16:08:07 +0100 Subject: [PATCH] Make PageQuerySet inherit from treebeards MP_NodeQuerySet --- wagtail/wagtailcore/query.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/wagtail/wagtailcore/query.py b/wagtail/wagtailcore/query.py index 2918db5b35..d182ec5887 100644 --- a/wagtail/wagtailcore/query.py +++ b/wagtail/wagtailcore/query.py @@ -1,9 +1,18 @@ -from django.db.models.query import QuerySet from django.db.models import Q from django.contrib.contenttypes.models import ContentType -class PageQuerySet(QuerySet): +# hack to import our patched copy of treebeard at wagtail/vendor/django-treebeard - +# based on http://stackoverflow.com/questions/17211078/how-to-temporarily-modify-sys-path-in-python +import sys +import os +treebeard_path = os.path.join(os.path.dirname(__file__), '..', 'vendor', 'django-treebeard') +sys.path.insert(0, treebeard_path) +from treebeard.mp_tree import MP_NodeQuerySet +sys.path.pop(0) + + +class PageQuerySet(MP_NodeQuerySet): """ Defines some extra query set methods that are useful for pages. """