BFS and DFS for PyGraphviz

pygraphviz-search.py

INDEX_BFS = 0
INDEX_DFS = -1
NOOP = lambda x: None


def traverse(graph, nodes, callback, index):
  visited = set()
  while nodes:
    node = nodes.pop(index)
    if node not in visited:
      visited.add(node)
      callback(node)
      nodes.extend(graph.successors(node))
  return visited


def bfs(graph, nodes, callback=NOOP):
  return traverse(graph, nodes, callback, INDEX_BFS)


def dfs(graph, nodes, callback=NOOP):
  return traverse(graph, nodes, callback, INDEX_DFS)