Source code for jsonpolars.chain

# -*- coding: utf-8 -*-

"""
Allow to use the polars liked chain syntax to create the expression.
"""

import typing as T
from .arg import PRE


if T.TYPE_CHECKING:  # pragma: no cover
    from .expr.api import T_EXPR


_ = PRE


def chain(*args: "T_EXPR") -> "T_EXPR":
    """
    Allow to use the polars liked chain syntax to create the expression.

    The next expression automatically use the previous one as the ``expr`` attribute.
    """
    ex = None
    for arg in args:
        # print(ex, arg)
        if ex is not None:
            setattr(arg, "expr", ex)
        ex = arg
    # print(ex)
    return ex