diff --git a/profiler/msprof_analyze/compare_tools/compare_backend/view/excel_view.py b/profiler/msprof_analyze/compare_tools/compare_backend/view/excel_view.py index 8e35bc84e6f853ea4df0a92f5b19f8be3e3400c9..57f2803e96ee0c7559c7f8aed6f2270f939a3896 100644 --- a/profiler/msprof_analyze/compare_tools/compare_backend/view/excel_view.py +++ b/profiler/msprof_analyze/compare_tools/compare_backend/view/excel_view.py @@ -29,9 +29,8 @@ class ExcelView(BaseView): self._args = args def generate_view(self): - workbook = Workbook(self._file_path) - for sheet_name, data in self._data_dict.items(): - WorkSheetCreator(workbook, sheet_name, data, self._args).create_sheet() - workbook.close() + with Workbook(self._file_path) as workbook: + for sheet_name, data in self._data_dict.items(): + WorkSheetCreator(workbook, sheet_name, data, self._args).create_sheet() os.chmod(self._file_path, Constant.FILE_AUTHORITY)