Vikas2425/Safety-Helmet-Detection

1import sys
2from src.logger import logging
3
4def error_message_detail(error,error_detail:sys):
5    _,_,exe_tb = error_detail.exc_info()
6
7    file_name = exe_tb.tb_frame.f_code.co_filename
8
9    line_number = exe_tb.tb_lineno
10
11    error_message = f"Error occurred in python script: [{file_name}] line number: [{line_number}] error message: [{str(error)}]"
12
13    return error_message
14
15class CustomException(Exception):
16    def __init__(self,error_message,error_detail:sys):
17        super().__init__(error_message)
18
19        self.error_message = error_message_detail(error_message,error_detail = error_detail)
20
21    def __str__(self):
22        return self.error_message
23