Skip to content

mirascope.core.base.merge_decorators

merge_decorators 

merge_decorators(
    decorator: Callable[
        [Callable[_P, _R]], Callable[_WP, _WR]
    ],
    *additional_decorators: Callable[[Callable], Callable]
) -> Callable[
    [], Callable[[Callable[_P, _R]], Callable[_WP, _WR]]
]

Combines multiple decorators into a single decorator factory.

This function allows you to merge multiple decorators into a single decorator factory. The decorators are applied in the order they are passed to the function. All function metadata (e.g. docstrings, function name) is preserved through the decoration chain.

Parameters:

Name Type Description Default
decorator Callable[[Callable[_P, _R]], Callable[_WP, _WR]]

The base decorator that determines the type signature of the decorated function.

 required
*additional_decorators Callable[[Callable], Callable]

Additional decorators to be merged with the base decorator.

 ()

Returns:

Type Description
Callable[[], Callable[[Callable[_P, _R]], Callable[_WP, _WR]]]

A decorator factory function that applies all decorators in the specified order.
Example 
# Combine multiple decorators
merged = merge_decorators(
    retry(),
    with_logging(),
    handle_errors()
)

# Apply the merged decorators
@merged()
def my_function():
    pass
Source code in mirascope/core/base/merge_decorators.py 
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
def merge_decorators(
    decorator: Callable[[Callable[_P, _R]], Callable[_WP, _WR]],
    *additional_decorators: Callable[[Callable], Callable],
) -> Callable[[], Callable[[Callable[_P, _R]], Callable[_WP, _WR]]]:
    """Combines multiple decorators into a single decorator factory.

    This function allows you to merge multiple decorators into a single decorator factory.
    The decorators are applied in the order they are passed to the function.
    All function metadata (e.g. docstrings, function name) is preserved through the decoration chain.

    Args:
        decorator: The base decorator that determines the type signature of the decorated function.
        *additional_decorators: Additional decorators to be merged with the base decorator.

    Returns:
        A decorator factory function that applies all decorators in the specified order.

    Example:
        ```python
        # Combine multiple decorators
        merged = merge_decorators(
            retry(),
            with_logging(),
            handle_errors()
        )

        # Apply the merged decorators
        @merged()
        def my_function():
            pass
        ```
    """
    decorators = [decorator] + list(additional_decorators)

    def decorator_factory() -> Callable[[Callable[_P, _R]], Callable[_WP, _WR]]:
        def inner(func: Callable[_P, _R]) -> Callable[_WP, _WR]:
            def compose(f: Callable, d: Callable) -> Callable:
                @wraps(f)
                def wrapped(*args: Any, **kwargs: Any) -> Any:  # noqa: ANN401
                    return d(f)(*args, **kwargs)

                return wrapped

            return reduce(compose, reversed(decorators), func)  # pyright: ignore [reportReturnType]

        return inner

    return decorator_factory