def test_write_node_dict(file_r_name, path_w): # build default node_dict with open(file_r_name, 'r') as file_json: node_dict = lem.build_node_dict(file_json) # default with open(path_w + 'nodes.csv', 'w') as file_w: lem.write_node_dict(file_w, node_dict) # different options path_w += 'options/' if not os.path.exists(path_w): os.makedirs(path_w) # with cut with open(path_w + 'nodes (cut=10).csv', 'w') as file_w: lem.write_node_dict(file_w, node_dict, cut=10) # with '\t' with open(path_w + 'nodes.tsv', 'w') as file_w: lem.write_node_dict(file_w, node_dict, sep='\t') print('Done test_write_node_dict')
file_nodes_name = path_w + 'nodes.csv' file_edges_name = path_w + 'edges.csv' file_nodes_json_name = path_w + 'nodes.json' file_edges_json_name = path_w + 'edges.json' # Создаём узлы with open(file_json_name, 'r') as file_json, \ open(file_weight_name, 'r') as file_weight, \ open(file_nodes_name, 'w') as file_nodes, \ open(file_nodes_json_name, 'w') as file_nodes_json: # создаём словарь узлов node_dict = lem.build_node_dict(file_json, weights=file_weight, include_bastard=bastard, include_non_cyrillic=non_cyrillic) # записываем узлы в файл для Gephi lem.write_node_dict(file_nodes, node_dict) # отдельно сохраняем json-версию словаря узлов (для отладки) pretty_json(node_dict, file_nodes_json) # Создаём рёбра with open(file_json_name, 'r') as file_json, \ open(file_weight_name, 'r') as file_weight, \ open(file_edges_name, 'w') as file_edges, \ open(file_edges_json_name, 'w') as file_edges_json: # создаём словарь рёбер edge_dict = lem.build_edge_dict(file_json, weights=file_weight, include_bastard=bastard, include_non_cyrillic=non_cyrillic) # записываем рёбра в файл для Gephi lem.write_edge_dict(file_edges, edge_dict) # отдельно сохраняем json-версию словаря рёбер (для отладки)