From b98429886e4cdbc1c14972d45d73478527fb70b7 Mon Sep 17 00:00:00 2001 From: Sapan Bhatia Date: Wed, 5 Nov 2014 00:14:23 -0500 Subject: [PATCH] Support walking both inverse and forward dependency grahps --- planetstack/dependency_walker.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/planetstack/dependency_walker.py b/planetstack/dependency_walker.py index a4e9dc8..ca1959e 100755 --- a/planetstack/dependency_walker.py +++ b/planetstack/dependency_walker.py @@ -37,6 +37,13 @@ def plural(name): return name+'s' +def walk_deps(fn, object): + model = object.__class__.__name__ + try: + deps = dependencies[model] + except: + deps = [] + __walk_deps(fn, object, deps) def walk_inv_deps(fn, object): model = object.__class__.__name__ @@ -44,6 +51,10 @@ def walk_inv_deps(fn, object): deps = inv_dependencies[model] except: deps = [] + __walk_deps(fn, object, deps) + +def __walk_deps(fn, object, deps): + model = object.__class__.__name__ for dep in deps: #print "Checking dep %s"%dep -- 2.43.0