Adding headers to HTTPError

bocadillo/error_handlers.py

@@ -23,6 +23,8 @@ async def error_to_html(req: Request, res: Response, exc: HTTPError):
     html = f"<h1>{exc.title}</h1>"
     if exc.detail:
         html += f"\n<p>{exc.detail}</p>"
+    if exc.headers:
+        html += f"\n<p>{exc.headers}</p>"
     res.html = html
 
 
@@ -46,6 +48,7 @@ async def error_to_json(req: Request, res: Response, exc: HTTPError):
     ```
     """
     res.status_code = exc.status_code
+    res.headers = exc.headers
     res.json = exc.as_json()
 
 
@@ -65,4 +68,6 @@ async def error_to_text(req: Request, res: Response, exc: HTTPError):
     text = exc.title
     if exc.detail:
         text += f"\n{exc.detail}"
+    if exc.headers:
+        text += f"\n{exc.headers}"
     res.text = text

bocadillo/errors.py

@@ -22,7 +22,10 @@ class HTTPError(Exception):
     __slots__ = ("_status", "detail")
 
     def __init__(
-        self, status: typing.Union[int, HTTPStatus], detail: typing.Any = ""
+        self,
+        status: typing.Union[int, HTTPStatus],
+        detail: typing.Any = "",
+        headers: typing.Dict = None,
     ):
         if isinstance(status, int):
             status = HTTPStatus(  # pylint: disable=no-value-for-parameter
@@ -34,6 +37,7 @@ def __init__(
             ), f"Expected int or HTTPStatus, got {type(status)}"
         self._status = status
         self.detail = detail
+        self.headers = headers
 
     @property
     def status_code(self) -> int:
@@ -54,6 +58,8 @@ def as_json(self) -> dict:
         data = {"error": self.title, "status": self.status_code}
         if self.detail:
             data["detail"] = self.detail
+        if self.headers:
+            data["headers"] = self.headers
         return data
 
     def __str__(self):