Spaces:
Running
Running
File size: 855 Bytes
6c58cf4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | 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}')"
|