def build_graph(code): graph = Graph() blocks = [b["name"] for b in code["blocks"]] edges = [(b["name"], e) for b in code["blocks"] for e in b["next_block"]] graph.add_nodes(*blocks) graph.add_edges(*edges) return graph