_brave.py 文件源码

def __parse_relations(self):
        relation_args = {}
        for rel in self.doc_data['relations']:
            key, name, role_ents = rel
            for role, ent_key in role_ents:
                curr_roles = relation_args.get(name, {})
                curr_types = curr_roles.get(role, set())
                curr_types.add(self.ent_dict[ent_key])
                curr_roles[role] = curr_types
                relation_args[name] = curr_roles
        range_ = range(0, len(relations_palette), (len(relations_palette) // len(relation_args.keys())))
        colors = [relations_palette[i] for i in range_]
        rel_colors = dict(zip(relation_args.keys(), colors))
        relation_types = []
        for name, args in iteritems(relation_args):
            rel_dict = {
                'args': [{'role': role, 'targets': list(targets)} for role, targets in iteritems(args)],
                'color': rel_colors[name],
                'dashArray': '3,3',
                'labels': [name, name[0:3]],
                'type': name

            }
            relation_types.append(rel_dict)
        self.coll_data['relation_types'] = relation_types