exceptions.py 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  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[typing.Dict[str, str]] = 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 __str__(self) -> str:
  18. return f"{self.status_code}: {self.detail}"
  19. def __repr__(self) -> str:
  20. class_name = self.__class__.__name__
  21. return f"{class_name}(status_code={self.status_code!r}, detail={self.detail!r})"
  22. class WebSocketException(Exception):
  23. def __init__(self, code: int, reason: typing.Optional[str] = None) -> None:
  24. self.code = code
  25. self.reason = reason or ""
  26. def __str__(self) -> str:
  27. return f"{self.code}: {self.reason}"
  28. def __repr__(self) -> str:
  29. class_name = self.__class__.__name__
  30. return f"{class_name}(code={self.code!r}, reason={self.reason!r})"
  31. __deprecated__ = "ExceptionMiddleware"
  32. def __getattr__(name: str) -> typing.Any: # pragma: no cover
  33. if name == __deprecated__:
  34. from starlette.middleware.exceptions import ExceptionMiddleware
  35. warnings.warn(
  36. f"{__deprecated__} is deprecated on `starlette.exceptions`. "
  37. f"Import it from `starlette.middleware.exceptions` instead.",
  38. category=DeprecationWarning,
  39. stacklevel=3,
  40. )
  41. return ExceptionMiddleware
  42. raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
  43. def __dir__() -> typing.List[str]:
  44. return sorted(list(__all__) + [__deprecated__]) # pragma: no cover