Spaces:
Running
Running
| import sys | |
| import traceback | |
| from typing import Optional | |
| class CustomException(Exception): | |
| def __init__(self, error_message: str, error_detail: Optional[sys.exc_info] = None): | |
| super().__init__(error_message) | |
| self.error_message = error_message | |
| if error_detail: | |
| _, _, exc_tb = error_detail | |
| self.error_message = self._get_detailed_error_message(error_message, exc_tb) | |
| def _get_detailed_error_message(self, error_message: str, exc_tb) -> str: | |
| file_name = exc_tb.tb_frame.f_code.co_filename | |
| line_number = exc_tb.tb_lineno | |
| return f"Error in [{file_name}] at line [{line_number}]: {error_message}" | |
| def __str__(self): | |
| return self.error_message | |
| def __repr__(self): | |
| return f"{self.__class__.__name__}('{self.error_message}')" | |