flow/hooks/DFGHook: add iterator on hooks

This commit is contained in:
Sebastien Bourdeauducq 2012-08-03 18:48:35 +02:00
parent 37fe6d64c3
commit adacdadd58
1 changed files with 5 additions and 4 deletions

View File

@ -37,9 +37,10 @@ class DFGHook:
ep = data["source"]
ep_to_hook[ep] = create(u, ep, v)
def get_fragment(self):
frag = Fragment()
def hooks_iter(self):
for v1 in self.nodepair_to_ep.values():
for v2 in v1.values():
frag += v2.get_fragment()
return frag
yield v2
def get_fragment(self):
return sum([h.get_fragment() for h in self.hooks_iter()], Fragment())