exceptions.py 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. import http
  2. import typing
  3. import warnings
  4. __all__ = ("HTTPException", "WebSocketException")
  5. class HTTPException(Exception):
  6. def __init__(
  7. self,
  8. status_code: int,
  9. detail: typing.Optional[str] = None,
  10. headers: typing.Optional[dict] = None,
  11. ) -> None:
  12. if detail is None:
  13. detail = http.HTTPStatus(status_code).phrase
  14. self.status_code = status_code
  15. self.detail = detail
  16. self.headers = headers
  17. def __repr__(self) -> str:
  18. class_name = self.__class__.__name__
  19. return f"{class_name}(status_code={self.status_code!r}, detail={self.detail!r})"
  20. class WebSocketException(Exception):
  21. def __init__(self, code: int, reason: typing.Optional[str] = None) -> None:
  22. self.code = code
  23. self.reason = reason or ""
  24. def __repr__(self) -> str:
  25. class_name = self.__class__.__name__
  26. return f"{class_name}(code={self.code!r}, reason={self.reason!r})"
  27. __deprecated__ = "ExceptionMiddleware"
  28. def __getattr__(name: str) -> typing.Any: # pragma: no cover
  29. if name == __deprecated__:
  30. from starlette.middleware.exceptions import ExceptionMiddleware
  31. warnings.warn(
  32. f"{__deprecated__} is deprecated on `starlette.exceptions`. "
  33. f"Import it from `starlette.middleware.exceptions` instead.",
  34. category=DeprecationWarning,
  35. stacklevel=3,
  36. )
  37. return ExceptionMiddleware
  38. raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
  39. def __dir__() -> typing.List[str]:
  40. return sorted(list(__all__) + [__deprecated__]) # pragma: no cover