update to make use of renamed NetworkTemplate fields

[plstackapi.git] / planetstack / dependency_walker.py

diff --git a/planetstack/dependency_walker.py b/planetstack/dependency_walker.py
index ca1959e..bc75b0c 100755 (executable)
--- a/planetstack/dependency_walker.py
+++ b/planetstack/dependency_walker.py
@@ -55,7 +55,6 @@ def walk_inv_deps(fn, object):
 
 def __walk_deps(fn, object, deps):
        model = object.__class__.__name__
-
        for dep in deps:
                #print "Checking dep %s"%dep
                peer=None
@@ -74,11 +73,14 @@ def __walk_deps(fn, object, deps):
                if (peer):
                        try:
                                peer_objects = peer.all()
-                       except:
+                       except AttributeError:
                                peer_objects = [peer]
+                       except:
+                               peer_objects = []
 
                        for o in peer_objects:
                                fn(o, object)
+                               # Uncomment the following line to enable recursion
                                # walk_inv_deps(fn, o)
 
 def p(x):