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}')"