Spaces:
Build error
Build error
| # src/exception.py | |
| import sys | |
| from src.logger import logging | |
| class CustomException(Exception): | |
| def __init__(self, error_message: Exception, error_detail: sys): | |
| super().__init__(error_message) | |
| self.error_message = CustomException.get_detailed_error_message(error_message, error_detail) | |
| def get_detailed_error_message(error: Exception, error_detail: sys): | |
| _, _, exc_tb = error_detail.exc_info() | |
| file_name = exc_tb.tb_frame.f_code.co_filename | |
| line_number = exc_tb.tb_lineno | |
| error_message = f"Error occurred in script: {file_name} at line {line_number}: {str(error)}" | |
| return error_message | |
| def __str__(self): | |
| return self.error_message |