Fix #591 (#601)

Author: RobertoPrevato

CHANGELOG.md

@@ -5,7 +5,7 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [2.4.1] - 2025-09-??
+## [2.4.1] - 2025-09-28
 
 - Correct bug in controller inheritance that would prevent argument types and
   return type hints from working as expected ([#594](https://github.com/Neoteroi/BlackSheep/pull/594)).
@@ -46,6 +46,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
   Replace the Application `exception_handlers` dictionary with a user defined
   dictionary that validates values, and change a piece of code that causes
   a recursive error when an exception handler itself is buggy.
+- Add support for specifying the status code in view functions
+  ([#591](https://github.com/Neoteroi/BlackSheep/issues/591)).
 - Fix `license` field in `pyproject.toml`.
 
 ## [2.4.0] - 2025-06-22

blacksheep/server/controllers.py

@@ -434,36 +434,46 @@ def full_view_name(self, name: str) -> str:
         return f"{self.class_name()}/{name}"
 
     def view(
-        self, name: Optional[str] = None, model: Optional[Any] = None, **kwargs
+        self,
+        name: Optional[str] = None,
+        model: Optional[Any] = None,
+        status: int = 200,
+        **kwargs,
     ) -> Response:
         """
         Returns a view rendered synchronously.
 
         :param name: name of the template (path to the template file,
             optionally without '.html' extension
         :param model: optional model, required to render the template.
+        :param status: optional status code for the response, default 200.
         :return: a Response object
         """
         if name is None:
             name = self.get_default_view_name()
 
-        return view(self.full_view_name(name), model, **kwargs)
+        return view(self.full_view_name(name), model, status, **kwargs)
 
     async def view_async(
-        self, name: Optional[str] = None, model: Optional[Any] = None, **kwargs
+        self,
+        name: Optional[str] = None,
+        model: Any = None,
+        status: int = 200,
+        **kwargs,
     ) -> Response:
         """
         Returns a view rendered asynchronously.
 
         :param name: name of the template (path to the template file,
             optionally without '.html' extension
         :param model: optional model, required to render the template.
+        :param status: optional status code for the response, default 200.
         :return: a Response object
         """
         if name is None:
             name = self.get_default_view_name()
 
-        return await view_async(self.full_view_name(name), model, **kwargs)
+        return await view_async(self.full_view_name(name), model, status, **kwargs)
 
 
 class APIController(Controller):

blacksheep/server/responses.py

@@ -320,14 +320,14 @@ def file(
     return _file(value, content_type, content_disposition, file_name)
 
 
-def _create_html_response(html: str):
+def _create_html_response(html: str, status: int = 200):
     """Creates a Response to serve dynamic HTML. Caching is disabled."""
-    return Response(200, [(b"Cache-Control", b"no-cache")]).with_content(
+    return Response(status, [(b"Cache-Control", b"no-cache")]).with_content(
         Content(b"text/html; charset=utf-8", html.encode("utf8"))
     )
 
 
-def view(name: str, model: Any = None, **kwargs) -> Response:
+def view(name: str, model: Any = None, status: int = 200, **kwargs) -> Response:
     """
     Returns a Response object with HTML obtained using synchronous rendering.
 
@@ -338,12 +338,15 @@ def view(name: str, model: Any = None, **kwargs) -> Response:
     renderer = html_settings.renderer
     if model:
         return _create_html_response(
-            renderer.render(name, html_settings.model_to_params(model), **kwargs)
+            renderer.render(name, html_settings.model_to_params(model), **kwargs),
+            status,
         )
-    return _create_html_response(renderer.render(name, None, **kwargs))
+    return _create_html_response(renderer.render(name, None, **kwargs), status)
 
 
-async def view_async(name: str, model: Any = None, **kwargs) -> Response:
+async def view_async(
+    name: str, model: Any = None, status: int = 200, **kwargs
+) -> Response:
     """
     Returns a Response object with HTML obtained using asynchronous rendering.
 
@@ -356,6 +359,9 @@ async def view_async(name: str, model: Any = None, **kwargs) -> Response:
         return _create_html_response(
             await renderer.render_async(
                 name, html_settings.model_to_params(model), **kwargs
-            )
+            ),
+            status,
         )
-    return _create_html_response(await renderer.render_async(name, None, **kwargs))
+    return _create_html_response(
+        await renderer.render_async(name, None, **kwargs), status
+    )

tests/test_templating.py

@@ -125,8 +125,11 @@ def specific_text():
 </html>"""
 
 
-async def _home_scenario(app: FakeApplication, url="/", expected_text=None):
+async def _home_scenario(
+    app: FakeApplication, url="/", expected_text=None, status: int = 200
+):
     await app(get_example_scope("GET", url), MockReceive(), MockSend())
+    assert app.response is not None
     text = await app.response.text()
 
     if expected_text is None:
@@ -143,7 +146,7 @@ async def _home_scenario(app: FakeApplication, url="/", expected_text=None):
 </html>"""
 
     assert text == expected_text
-    assert app.response.status == 200
+    assert app.response.status == status
 
 
 async def _view_scenario(app: FakeApplication, expected_text, url="/"):
@@ -163,6 +166,16 @@ async def home():
     await _home_scenario(app)
 
 
+async def test_jinja_async_mode_with_status(home_model, async_jinja_env):
+    app, render = get_app(True)
+
+    @app.router.get("/")
+    async def home():
+        return await render("home", home_model, status=500)
+
+    await _home_scenario(app, status=500)
+
+
 async def test_jinja_sync_mode(home_model):
     app, render = get_app(False)
 
@@ -173,6 +186,16 @@ async def home():
     await _home_scenario(app)
 
 
+async def test_jinja_sync_mode_with_status(home_model):
+    app, render = get_app(False)
+
+    @app.router.get("/")
+    async def home():
+        return render("home", home_model, 500)
+
+    await _home_scenario(app, status=500)
+
+
 async def test_controller_conventional_view_name(home_model):
     app, _ = get_app(False)
     app.controllers_router = RoutesRegistry()
@@ -212,6 +235,34 @@ def index(self):
     await _home_scenario(app, expected_text=specific_text)
 
 
+async def test_controller_conventional_view_name_async_with_status(
+    home_model, async_jinja_env
+):
+    app, _ = get_app(True)
+    app.controllers_router = RoutesRegistry()
+    get = app.controllers_router.get
+
+    class Lorem(Controller):
+        @get()
+        async def index(self):
+            return await self.view_async(model=home_model, status=404)
+
+    await _home_scenario(app, status=404)
+
+
+async def test_controller_specific_view_name_with_status(home_model, specific_text):
+    app, _ = get_app(False)
+    app.controllers_router = RoutesRegistry()
+    get = app.controllers_router.get
+
+    class Lorem(Controller):
+        @get()
+        def index(self):
+            return self.view("specific", home_model, status=404)
+
+    await _home_scenario(app, expected_text=specific_text, status=404)
+
+
 async def test_controller_specific_view_name_async(
     home_model, specific_text, async_jinja_env
 ):