X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/a05c936dabd92bf4756f4ca2c23ef0320b0543d9..d0dbd3635a4c1d1c47040fd17f5a8d2bec08a7e9:/philo/contrib/shipherd/models.py

diff --git a/philo/contrib/shipherd/models.py b/philo/contrib/shipherd/models.py
index c299a14..95be501 100644
--- a/philo/contrib/shipherd/models.py
+++ b/philo/contrib/shipherd/models.py
@@ -27,13 +27,12 @@ class NavigationMapper(object, DictMixin):
 		self._cache = {}
 	
 	def __getitem__(self, key):
-		return Navigation.objects.get_for_node(self.node, key)
-		#if key not in self._cache:
-		#	try:
-		#		self._cache[key] = Navigation.objects.get_for_node(self.node, key)
-		#	except Navigation.DoesNotExist:
-		#		self._cache[key] = None
-		#return self._cache[key]
+		if key not in self._cache:
+			try:
+				self._cache[key] = Navigation.objects.get_for_node(self.node, key)
+			except Navigation.DoesNotExist:
+				self._cache[key] = None
+		return self._cache[key]
 
 
 def navigation(self):
@@ -62,7 +61,7 @@ class NavigationManager(models.Manager):
 				"node__%s__gte" % opts.right_attr: right,
 				"node__%s" % opts.tree_id_attr: tree_id
 			}
-			navs = self.filter(key=key, **kwargs).order_by('-node__%s' % opts.level_attr)
+			navs = self.filter(key=key, **kwargs).select_related('node').order_by('-node__%s' % opts.level_attr)
 			nav = navs[0]
 			roots = nav.roots.all().select_related('target_node').order_by('order')
 			item_opts = NavigationItem._mptt_meta
@@ -75,7 +74,8 @@ class NavigationManager(models.Manager):
 				by_pk[root.pk] = root
 				tree_ids.append(getattr(root, item_opts.tree_id_attr))
 				root._cached_children = []
-				root.target_node.get_path(root=site_root_node)
+				if root.target_node:
+					root.target_node.get_path(root=site_root_node)
 				root.navigation = nav
 			
 			kwargs = {
@@ -90,7 +90,8 @@ class NavigationManager(models.Manager):
 				parent_pk = getattr(item, '%s_id' % item_opts.parent_attr)
 				item.parent = by_pk[parent_pk]
 				item.parent._cached_children.append(item)
-				item.target_node.get_path(root=site_root_node)
+				if item.target_node:
+					item.target_node.get_path(root=site_root_node)
 			
 			cached = roots
 			cache.set(cache_key, cached)