AlainDeLong/End-To-End-Machine-Learning-Project

1import sys2from src.logger import logging3 4 5def error_message_detail(error, error_detail: sys):6    _, _, exc_tb = error_detail.exc_info()7    file_name = exc_tb.tb_frame.f_code.co_filename8    error_message = "Error occured in python script name [{0}] line number [{1}] error message [{2}]".format(9        file_name, exc_tb.tb_lineno, str(error)10    )11    return error_message12 13 14class CustomException(Exception):15    def __init__(self, error_message, error_detail: sys):16        super().__init__(error_message)17        self.error_message = error_message_detail(18            error_message, error_detail=error_detail19        )20 21    def __str__(self):22        return self.error_message23 24 25# if __name__ == "__main__":26#     try:27#         a = 1 / 028#     except Exception as e:29#         logging.info("Divide by Zero")30#         raise CustomException(e, sys)31