X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/e10a6144b092d89e1410b240d7380d48a646ffc3..26c548c5fe8c7cc7227b5f38cd27e0278442dd76:/contrib/shipherd/admin.py

diff --git a/contrib/shipherd/admin.py b/contrib/shipherd/admin.py
index 606971d..a54d0f1 100644
--- a/contrib/shipherd/admin.py
+++ b/contrib/shipherd/admin.py
@@ -1,5 +1,6 @@
 from django.contrib import admin
 from philo.admin import TreeEntityAdmin, COLLAPSE_CLASSES, NodeAdmin
+from philo.models import Node
 from philo.contrib.shipherd.models import Navigation
 
 
@@ -34,8 +35,14 @@ class NavigationNavigationInline(NavigationInline):
 	verbose_name_plural = "children"
 
 
-class NodeNavigationInline(NavigationInline):
-	verbose_name_plural = 'navigation'
+class NodeHostedNavigationInline(NavigationInline):
+	verbose_name_plural = 'hosted navigation'
+	fk_name = 'hosting_node'
+
+
+class NodeTargetingNavigationInline(NavigationInline):
+	verbose_name_plural = 'targeting navigation'
+	fk_name = 'target_node'
 
 
 class NavigationAdmin(TreeEntityAdmin):
@@ -60,7 +67,9 @@ class NavigationAdmin(TreeEntityAdmin):
 	inlines = [NavigationNavigationInline] + TreeEntityAdmin.inlines
 
 
-NodeAdmin.inlines = [NodeNavigationInline] + NodeAdmin.inlines
+NodeAdmin.inlines = [NodeHostedNavigationInline, NodeTargetingNavigationInline] + NodeAdmin.inlines
 
 
+admin.site.unregister(Node)
+admin.site.register(Node, NodeAdmin)
 admin.site.register(Navigation, NavigationAdmin)
\ No newline at end of file