App/schemas.py

from pydantic import BaseModel
from typing import Optional, Any
from fastapi import HTTPException


class ResponseModel(BaseModel):
    success: bool
    message: str
    data: Optional[Any] = None


class AppException(HTTPException):
    def __init__(self, status_code: int = 400, detail: str | ResponseModel = None):
        if isinstance(detail, ResponseModel):
            super().__init__(status_code=status_code, detail=detail.message)
            self.data = detail.data
            self.response_model = detail
        else:
            super().__init__(status_code=status_code, detail=str(detail) if detail else "An error occurred")
            self.data = None
            self.response_model = ResponseModel(
                success=False,
                message=str(detail) if detail else "An error occurred",
                data=None
            )