openstack/vitrage
Code Issues Proposed changes

Merge "bugfix - dict item removed during iteration"

Browse Source
This commit is contained in:
Zuul 2018-01-17 14:15:36 +00:00 committed by Gerrit Code Review
commit 15e956848a
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
vitrage/graph/algo_driver/networkx_algorithm.py
View File

@ -230,6 +230,8 @@ class NXAlgorithm(GraphAlgorithm):
@staticmethod
def _apply_edge_attr_filter(graph, edge_attr_filter):
for source, target, edge_data in graph._g.edges_iter(data=True):
if not check_filter(edge_data, edge_attr_filter):
graph._g.remove_edge(u=source, v=target)
edges_iter = graph._g.edges_iter(data=True, keys=True)
edges_to_remove = [(u, v, k) for (u, v, k, d) in edges_iter
if not check_filter(d, edge_attr_filter)]
for source, target, key in edges_to_remove:
graph._g.remove_edge(u=source, v=target, key=key)