Merge pull request #231 from Tatsh/route-info-schema-out

Allow customisation of schema_out in route_info dicts

Author: eadwinCode

docs/api_controller/model_controller/02_model_configuration.md

@@ -148,6 +148,7 @@ class EventModelController(ModelControllerBase):
             "summary": "List all events",
             "description": "Retrieves a paginated list of all events",
             "tags": ["events"],
+            "schema_out": CustomListSchema,
         },
         find_one_route_info={
             "summary": "Get event details",

ninja_extra/controllers/model/builder.py

@@ -60,7 +60,7 @@ def _register_create_endpoint(self) -> None:
         kw.update(self._config.create_route_info)
         create_item = self._route_factory.create(
             schema_in=self._create_schema,  # type:ignore[arg-type]
-            schema_out=self._retrieve_schema,  # type:ignore[arg-type]
+            schema_out=kw.pop('schema_out', self._retrieve_schema),  # type:ignore[arg-type]
             **kw,  # type:ignore[arg-type]
         )
 
@@ -82,7 +82,7 @@ def _register_update_endpoint(self) -> None:
             path=_path,
             lookup_param=self._model_pk_name,
             schema_in=self._update_schema,  # type:ignore[arg-type]
-            schema_out=self._retrieve_schema,  # type:ignore[arg-type]
+            schema_out=kw.pop('schema_out', self._retrieve_schema),  # type:ignore[arg-type]
             **kw,  # type:ignore[arg-type]
         )
 
@@ -105,7 +105,7 @@ def _register_patch_endpoint(self) -> None:
         patch_item = self._route_factory.patch(
             path=_path,
             lookup_param=self._model_pk_name,
-            schema_out=self._retrieve_schema,  # type:ignore[arg-type]
+            schema_out=kw.pop('schema_out', self._retrieve_schema),  # type:ignore[arg-type]
             schema_in=self._patch_schema,  # type:ignore[arg-type]
             **kw,  # type:ignore[arg-type]
         )
@@ -154,7 +154,7 @@ def _register_list_endpoint(self) -> None:
 
         list_items = self._route_factory.list(
             path="/",
-            schema_out=self._retrieve_schema,  # type:ignore[arg-type]
+            schema_out=kw.pop('schema_out', self._retrieve_schema),  # type:ignore[arg-type]
             **kw,  # type:ignore[arg-type]
             **paginate_kwargs,
         )