🔨 Add http_exception handler to Admin class

sqladmin/application.py

@@ -5,6 +5,7 @@
 from typing import (
     TYPE_CHECKING,
     Any,
+    Awaitable,
     Callable,
     List,
     Optional,
@@ -371,15 +372,17 @@ def __init__(
 
         statics = StaticFiles(packages=["sqladmin"])
 
-        # def http_exception(request: Request, exc: Exception) -> Response:
-        #     assert isinstance(exc, HTTPException)
-        #     context = {
-        #         "status_code": exc.status_code,
-        #         "message": exc.detail,
-        #     }
-        #     return self.templates.TemplateResponse(
-        #         request, "error.html", context, status_code=exc.status_code
-        #     )
+        async def http_exception(
+            request: Request, exc: Exception
+        ) -> Union[Response, Awaitable[Response]]:
+            assert isinstance(exc, HTTPException)
+            context = {
+                "status_code": exc.status_code,
+                "message": exc.detail,
+            }
+            return await self.templates.TemplateResponse(
+                request, "error.html", context, status_code=exc.status_code
+            )
 
         routes = [
             Mount("/statics", app=statics, name="statics"),
@@ -417,7 +420,7 @@ def __init__(
         ]
 
         self.admin.router.routes = routes
-        # self.admin.exception_handlers = {HTTPException: http_exception}
+        self.admin.exception_handlers = {HTTPException: http_exception}
         self.admin.debug = debug
         self.app.mount(base_url, app=self.admin, name="admin")
 

sqladmin/helpers.py

@@ -159,7 +159,7 @@ def write(self, value: T) -> T:
 
 
 def stream_to_csv(
-    callback: Callable[[Writer], AsyncGenerator[T, None]]
+    callback: Callable[[Writer], AsyncGenerator[T, None]],
 ) -> Generator[T, None, None]:
     """Function that takes a callable (that yields from a CSV Writer), and
     provides it a writer that streams the output directly instead of