def create_edge_like(line):
assert len(line) == 3
edge = ttypes.Edge()
edge.src = utf8b(line[0])
edge.dst = utf8b(line[1])
edge.type = 1
edge.ranking = 0
edge.name = utf8b('like')
props = dict()
likeness = ttypes.Value()
likeness.set_iVal(int(line[2]))
props[utf8b('likeness')] = likeness
edge.props = props
return edge
def create_vertex_team(line):
assert len(line) == 2
vertex = ttypes.Vertex()
vertex.vid = utf8b(line[0])
tags = []
tag = ttypes.Tag()
tag.name = utf8b('team')
props = dict()
name = ttypes.Value()
name.set_sVal(utf8b(line[1]))
props[utf8b('name')] = name
tag.props = props
tags.append(tag)
vertex.tags = tags
return vertex
def create_edge_teammate(line):
assert len(line) == 4
edge = ttypes.Edge()
edge.src = utf8b(line[0])
edge.dst = utf8b(line[1])
edge.type = 1
edge.ranking = 0
edge.name = utf8b('teammate')
props = dict()
start_year = ttypes.Value()
start_year.set_iVal(int(line[2]))
end_year = ttypes.Value()
end_year.set_iVal(int(line[3]))
props[utf8b('start_year')] = start_year
props[utf8b('end_year')] = end_year
edge.props = props
return edge
def create_vertex_player(line):
assert len(line) == 3
vertex = ttypes.Vertex()
vertex.vid = utf8b(line[0])
tags = []
tag = ttypes.Tag()
tag.name = utf8b('player')
props = dict()
name = ttypes.Value()
name.set_sVal(utf8b(line[1]))
props[utf8b('name')] = name
age = ttypes.Value()
age.set_iVal(int(line[2]))
props[utf8b('age')] = age
tag.props = props
tags.append(tag)
vertex.tags = tags
return vertex
def create_edge_serve(line):
assert len(line) == 4
edge = ttypes.Edge()
src = ttypes.Value()
src.set_sVal(utf8b(line[0]))
edge.src = src
dst = ttypes.Value()
if '@' in line[1]:
temp = list(map(lambda i: i.strip('"'), re.split('@', line[1])))
dst.set_sVal(utf8b(temp[0]))
edge.dst = dst
edge.ranking = int(temp[1])
else:
dst.set_sVal(utf8b(line[1]))
edge.dst = dst
edge.ranking = 0
edge.type = 1
edge.name = utf8b('serve')
props = dict()
start_year = ttypes.Value()
start_year.set_iVal(int(line[2]))
end_year = ttypes.Value()
end_year.set_iVal(int(line[3]))
props[utf8b('start_year')] = start_year
props[utf8b('end_year')] = end_year
edge.props = props
return edge
def create_vertex_bachelor(line):
assert len(line) == 3
vertex = ttypes.Vertex()
vertex.vid = utf8b(line[0])
tags = []
tag = ttypes.Tag()
tag.name = utf8b('bachelor')
props = dict()
name = ttypes.Value()
name.set_sVal(utf8b(line[1]))
props[utf8b('name')] = name
speciality = ttypes.Value()
speciality.set_sVal(utf8b(line[2]))
props[utf8b('speciality')] = speciality
tag.props = props
tags.append(tag)
vertex.tags = tags
return vertex