From b7e823037f9be4a13c46d684d38f73b417cc44c0 Mon Sep 17 00:00:00 2001
From: Karl Hobley <karl@torchbox.com>
Date: Tue, 22 Apr 2014 09:50:18 +0100
Subject: [PATCH] Updated wagtail to use treebeard 2.0

---
 setup.py                      |  1 +
 wagtail/wagtailcore/models.py | 10 ++--------
 wagtail/wagtailcore/query.py  |  9 ---------
 3 files changed, 3 insertions(+), 17 deletions(-)

diff --git a/setup.py b/setup.py
index eee178ed2f..181672e069 100644
--- a/setup.py
+++ b/setup.py
@@ -45,6 +45,7 @@ setup(
         "django-libsass>=0.1",
         "django-modelcluster>=0.1",
         "django-taggit>=0.11.2",
+        "django-treebeard==2.0",
         "Pillow>=2.3.0",
         "beautifulsoup4>=4.3.2",
         "lxml>=3.3.0",
diff --git a/wagtail/wagtailcore/models.py b/wagtail/wagtailcore/models.py
index d322ab69b5..bde5a0449c 100644
--- a/wagtail/wagtailcore/models.py
+++ b/wagtail/wagtailcore/models.py
@@ -13,20 +13,14 @@ from django.conf import settings
 from django.template.response import TemplateResponse
 from django.utils.translation import ugettext_lazy as _
 
+from treebeard.mp_tree import MP_Node
+
 from wagtail.wagtailcore.util import camelcase_to_underscore
 from wagtail.wagtailcore.query import PageQuerySet
 
 from wagtail.wagtailsearch import Indexed, get_search_backend
 
 
-# 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
-treebeard_path = os.path.join(os.path.dirname(__file__), '..', 'vendor', 'django-treebeard')
-sys.path.insert(0, treebeard_path)
-from treebeard.mp_tree import MP_Node
-sys.path.pop(0)
-
-
 class SiteManager(models.Manager):
     def get_by_natural_key(self, hostname):
         return self.get(hostname=hostname)
diff --git a/wagtail/wagtailcore/query.py b/wagtail/wagtailcore/query.py
index 048ccda063..9404e25b4d 100644
--- a/wagtail/wagtailcore/query.py
+++ b/wagtail/wagtailcore/query.py
@@ -1,15 +1,6 @@
 from django.db.models import Q
 from django.contrib.contenttypes.models import ContentType
-
-
-# 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):