| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758 |
- # Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- import os
- import sys
- import traceback
- from .file_interface import write_json_file
- def try_except_decorator(func):
- """try-except"""
- def wrap(self, *args, **kwargs):
- try:
- result = func(self, *args, **kwargs)
- if result:
- save_result(True, self._mode, self._output, result_dict=result)
- except Exception:
- exc_type, exc_value, exc_tb = sys.exc_info()
- save_result(
- False,
- self._mode,
- self._output,
- err_type=str(exc_type),
- err_msg=str(exc_value),
- )
- traceback.print_exception(exc_type, exc_value, exc_tb)
- sys.exit(1)
- return wrap
- def save_result(run_pass, mode, output, result_dict=None, err_type=None, err_msg=None):
- """format, build and save result"""
- json_data = {"done_flag": run_pass}
- if not run_pass:
- assert result_dict is None and err_type is not None and err_msg is not None
- json_data.update({"err_type": err_type, "err_msg": err_msg})
- else:
- assert result_dict is not None and err_type is None and err_msg is None
- json_data.update(result_dict)
- if not os.path.exists(output):
- os.makedirs(output, exist_ok=True)
- write_json_file(json_data, os.path.join(output, f"{mode}_result.json"), indent=2)
|
