Add set_params function (#1029)

Author: uriyyo

fastapi_pagination/__init__.py

@@ -5,6 +5,7 @@
     "resolve_params",
     "response",
     "set_page",
+    "set_params",
     "Page",
     "Params",
     "LimitOffsetPage",
@@ -21,6 +22,7 @@
     resolve_params,
     response,
     set_page,
+    set_params,
 )
 from .default import Page, Params
 from .limit_offset import LimitOffsetPage, LimitOffsetParams

fastapi_pagination/api.py

@@ -6,6 +6,7 @@
     "response",
     "request",
     "set_page",
+    "set_params",
     "set_items_transformer",
     "apply_items_transformer",
     "pagination_ctx",
@@ -207,6 +208,10 @@ def _reset_ctx() -> Iterator[None]:
     return _reset_ctx()
 
 
+def set_params(params: AbstractParams) -> ContextManager[None]:
+    return _ctx_var_with_reset(_params_val, params)
+
+
 def set_page(page: Type[AbstractPage[Any]]) -> ContextManager[None]:
     return _ctx_var_with_reset(_page_val, page)
 
@@ -270,7 +275,7 @@ def _create_params_dependency(
 ) -> Callable[[TAbstractParams], AsyncIterator[TAbstractParams]]:
     async def _pagination_params(*args: Any, **kwargs: Any) -> AsyncIterator[TAbstractParams]:
         val = params(*args, **kwargs)
-        with _ctx_var_with_reset(_params_val, cast(AbstractParams, val)):
+        with set_params(val):
             yield val
 
     sign = inspect.signature(params)