|
|
|
@ -473,19 +473,24 @@ class FrameworkParser:
|
|
|
|
|
with open(self._save_path, 'w') as save_file:
|
|
|
|
|
csv_writer = csv.writer(save_file)
|
|
|
|
|
csv_writer.writerow(self._col_names)
|
|
|
|
|
pre_graph_info = None
|
|
|
|
|
for path in self._framework_path['graph']:
|
|
|
|
|
first_row = True
|
|
|
|
|
with open(path, 'r') as graph_file:
|
|
|
|
|
for graph_info in graph_file:
|
|
|
|
|
result = self._parse_one_row_graph_info(graph_info)
|
|
|
|
|
task_info = task_cache.get(result[0])
|
|
|
|
|
if task_info:
|
|
|
|
|
task_info.extend(result)
|
|
|
|
|
csv_writer.writerow(task_info)
|
|
|
|
|
del task_cache[result[0]]
|
|
|
|
|
else:
|
|
|
|
|
save_info = [None, None, None]
|
|
|
|
|
save_info.extend(result)
|
|
|
|
|
csv_writer.writerow(save_info)
|
|
|
|
|
if first_row is True:
|
|
|
|
|
first_row = False
|
|
|
|
|
# The last row of the previous file and the first row of the current file may need
|
|
|
|
|
# to be combined to one row
|
|
|
|
|
if graph_info.startswith("op_name:") is False:
|
|
|
|
|
pre_graph_info = pre_graph_info + graph_info
|
|
|
|
|
continue
|
|
|
|
|
if pre_graph_info is not None:
|
|
|
|
|
self._parse_graph_row_and_save(task_cache, csv_writer, pre_graph_info)
|
|
|
|
|
pre_graph_info = graph_info
|
|
|
|
|
|
|
|
|
|
if pre_graph_info is not None:
|
|
|
|
|
self._parse_graph_row_and_save(task_cache, csv_writer, pre_graph_info)
|
|
|
|
|
|
|
|
|
|
none_list = [None, None, None, None]
|
|
|
|
|
for key, value in task_cache.items():
|
|
|
|
@ -494,6 +499,26 @@ class FrameworkParser:
|
|
|
|
|
csv_writer.writerow(value)
|
|
|
|
|
os.chmod(self._save_path, stat.S_IREAD | stat.S_IWRITE)
|
|
|
|
|
|
|
|
|
|
def _parse_graph_row_and_save(self, task_cache, csv_writer, graph_info):
|
|
|
|
|
"""
|
|
|
|
|
Parse the framework graph row and save the framework information.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
task_cache (dict): The task information cache.
|
|
|
|
|
csv_writer (csv): Csv writer.
|
|
|
|
|
graph_info (str): Row info of graph.
|
|
|
|
|
"""
|
|
|
|
|
result = self._parse_one_row_graph_info(graph_info)
|
|
|
|
|
task_info = task_cache.get(result[0])
|
|
|
|
|
if task_info:
|
|
|
|
|
task_info.extend(result)
|
|
|
|
|
csv_writer.writerow(task_info)
|
|
|
|
|
del task_cache[result[0]]
|
|
|
|
|
else:
|
|
|
|
|
save_info = [None, None, None]
|
|
|
|
|
save_info.extend(result)
|
|
|
|
|
csv_writer.writerow(save_info)
|
|
|
|
|
|
|
|
|
|
def _parse_one_row_graph_info(self, row_info):
|
|
|
|
|
"""
|
|
|
|
|
Parse the graph information in one row.
|
|
|
|
|