def idToNode(g: GraphTraversalSource, id: str):
return g.V().has('user', 'userId', id) \
.fold() \
.coalesce(__.unfold(),
__.add_v('user').property('userId', id)) \
.next()
def loadVotes():
with open("/tmp/wiki-Vote.txt", "r") as file:
for _ in range(4):
next(file)
for line in file:
ids = line.split('\t')
from_node = idToNode(g, ids[0])
to_node = idToNode(g, ids[1])
g.add_e('votesFor').from_(from_node).to(to_node).iterate()
def idToNode(g: GraphTraversalSource, id: str):
return g.V().has('user', 'userId', id) \
.fold() \
.coalesce(__.unfold(),
__.add_v('user').property('userId', id)) \
.next()
def loadVotes():
with open("/tmp/wiki-Vote.txt", "r") as file:
for _ in range(4):
next(file)
for line in file:
ids = line.split('\t')
from_node = idToNode(g, ids[0])
to_node = idToNode(g, ids[1])
g.add_e('votesFor').from_(from_node).to(to_node).iterate()