def main(usename, password, raw_graph_class_node, raw_graph_class_edge,
query_size, graph_size, socket, uri, graph_path):
print(
"[INFO] Start. (" +
str(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))) +
")")
# Neo4j database configuration
# uri = "bolt://127.0.0.1:7687"
# usename = "neo4j"
# password = "******"
# raw_graph_class_node = "item_1"
# raw_graph_class_edge = "rel_1"
# 使用datetime.now()
now = datetime.datetime.now()
timestr = str(now.year) + str(now.month) + str(now.day) + str(
now.hour) + str(now.minute) + str(now.second)
# 需要根据项目所在目录进行改变的
#获取项目的路径
base_path = os.getcwd()
base_path_data = os.getcwd() + "/data/data_" + timestr
base_path_result = os.getcwd() + "/result/result_" + timestr
print("base_path" + base_path)
if not os.path.exists(base_path_data):
os.makedirs(base_path_data)
if not os.path.exists(base_path_result):
os.makedirs(base_path_result)
#获取服务器上neo4j（图数据库）的路径
# graph_path = "/Users/mark/Documents/MasterLife/GraphDB/neo4j-community-3.4.6"
# New graph configuration: graph_conf[size, level, query_num]
# query_size = 20
# graph_size = [ [2000, 5, 1] ]
# query_size = 1
# graph_size = [ [3000, 7, 1], [6000, 10, 1], [12000, 14, 1], [24000, 20, 1] ]
# 2000，5，1&2000, 6, 1
# graph_size = "2000,5,1"
_graph_size = graph_size.split("&")
graph_conf = []
for i in _graph_size:
_i = i.split(",")
for j in range(0, int(_i[2])):
graph_conf.append([int(_i[0]), int(_i[1]), query_size])
# print(graph_conf)
# graph_conf = []
# for i in graph_size:
# for j in range(0, i[2]):
# graph_conf.append([i[0], i[1], query_size])
# Export raw graph from Neo4j DB
socket.wfile.write(("Start to Analyze ！<br>").encode())
ge = GraphExporter()
raw_graph, g1_level, g2_level = ge.export(uri, usename, password,
raw_graph_class_node,
raw_graph_class_edge)
#
# # Create new graph (including dumping to file)
# # gg = GraphGenerator(raw_graph, g1_level, g2_level, graph_size, graph_level, graph_number)
print("Analyze End!")
socket.wfile.write(("Analyze End!！<br>").encode())
gg = GraphGenerator(raw_graph, (g1_level + 1), (g2_level + 1), graph_conf,
base_path, base_path_data, base_path_result)
new_graphs, node_id, edge_id = gg.create()
print("BOM Generate End!")
socket.wfile.write(("BOM Generate End!<br>").encode())
qg = QueryGenerator(new_graphs, graph_conf, node_id, edge_id, base_path,
graph_path, base_path_data, base_path_result)
qg.generate()
print("Query Generate End!")
socket.wfile.write(("Query Generate End!<br>").encode())
print(
"[INFO] End. (" +
str(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))) +
")")
return
# print(graph_conf)
# graph_conf = []
# for i in graph_size:
# for j in range(0, i[2]):
# graph_conf.append([i[0], i[1], query_size])
# Export raw graph from Neo4j DB
ge = GraphExporter()
raw_graph, g1_level, g2_level = ge.export(uri, usename, password,
raw_graph_class_node,
raw_graph_class_edge)
#
# # Create new graph (including dumping to file)
# # gg = GraphGenerator(raw_graph, g1_level, g2_level, graph_size, graph_level, graph_number)
print("Analyze End!")
gg = GraphGenerator(raw_graph, (g1_level + 1), (g2_level + 1), graph_conf,
base_path, base_path_data, base_path_result)
new_graphs, node_id, edge_id = gg.create()
print("BOM Generate End!")
qg = QueryGenerator(new_graphs, graph_conf, node_id, edge_id, base_path,
graph_path, base_path_data, base_path_result)
qg.generate()
print("Query Generate End!")
print(
"[INFO] End. (" +
str(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time()))) +
")")
# return
