Skip to content

Commit d05f70a

Browse files
committed
Update neo4j ingestor to work with 0.3.0
1 parent 832ab72 commit d05f70a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

maltoolbox/ingestors/neo4j.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def ingest_attack_graph(graph,
3535

3636
nodes = {}
3737
rels = []
38-
for node in graph.nodes:
38+
for node in graph.nodes.values():
3939
node_dict = node.to_dict()
4040
nodes[node.id] = Node(
4141
node_dict['asset'] if 'asset' in node_dict else node_dict['id'],
@@ -50,7 +50,7 @@ def ingest_attack_graph(graph,
5050
in node_dict else 'N/A')
5151

5252

53-
for node in graph.nodes:
53+
for node in graph.nodes.values():
5454
for child in node.children:
5555
rels.append(Relationship(nodes[node.id], nodes[child.id]))
5656

0 commit comments

Comments
 (0)