from typing import Optional


class FastAPIChameleonException(Exception):
    pass


class FastAPIChameleonNotFoundException(FastAPIChameleonException):
    def __init__(self, message: Optional[str] = None, four04template_file: str = 'errors/404.pt'):
        super().__init__(message)

        self.template_file: str = four04template_file
        self.message: Optional[str] = message


class FastAPIChameleonGenericException(FastAPIChameleonException):
    def __init__(self, template_file: str, status_code: int,
                 message: Optional[str] = None, template_data: Optional[dict] = None):
        super().__init__(message)

        self.template_file: str = template_file
        self.status_code: int = status_code
        self.message: Optional[str] = message
        self.template_data: Optional[dict] = template_data