Skip to content

inject_params

fastagency.api.dependency_injection.inject_params #

inject_params(
    f: Callable[..., Any], ctx: dict[str, Any]
) -> Callable[..., Any]
Source code in fastagency/api/dependency_injection.py 
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
def inject_params(f: Callable[..., Any], ctx: dict[str, Any]) -> Callable[..., Any]:
    keys_used = set(signature(f).parameters.keys()) & set(ctx.keys())

    @wraps(f)
    def wrapper(*args: Any, **kwargs: dict[str, Any]) -> Any:
        # check if all required parameters are present
        if not keys_used.issubset(ctx.keys()):
            raise ValueError(f"Missing required parameters: {keys_used - ctx.keys()}")

        params = {k: ctx[k] for k in keys_used}
        return f(**params, **kwargs)

    # Update the signature of wrapper to remove parameters passed in kwargs
    sig = signature(f)
    new_params = [
        param for name, param in sig.parameters.items() if name not in keys_used
    ]
    wrapper.__signature__ = sig.replace(parameters=new_params)  # type: ignore[attr-defined]

    return wrapper