Function _get_idx_expr_recurse

tables/conditions.py:194–329

Actual implementation of the `get_idx_expr()` wrapper. 'idxexprs' is a list of expressions in the form ``(var, (ops), (limits))``. 'strexpr' is the indexable expression in string format. These parameters will be received empty (i.e. [], ['']) for the first time and populated during

(
    exprnode: ne.expressions.ExpressionNode,
    indexedcols: frozenset[str],
    idxexprs: list,
    strexpr: list[str],
)

def _get_idx_expr_recurse(
    exprnode: ne.expressions.ExpressionNode,
    indexedcols: frozenset[str],
    idxexprs: list,
    strexpr: list[str],
) -> (
    list[tuple[Any, tuple[str], tuple[Any]]]
    | list[
        tuple[ne.expressions.ExpressionNode, tuple[str, str], tuple[Any, Any]]
    ]
    | tuple[list, list[str]]
):
    """Actual implementation of the `get_idx_expr()` wrapper.

    'idxexprs' is a list of expressions in the form ``(var, (ops),
    (limits))``. 'strexpr' is the indexable expression in string format.
    These parameters will be received empty (i.e. [], ['']) for the
    first time and populated during the different recursive calls.
    Finally, they are returned in the last level to the original
    wrapper.  If 'exprnode' is not indexable, it will return the tuple
    ([], ['']) so as to signal this.

    """
    not_indexable = ([], [""])
    op_conv = {
        "and": "&",
        "or": "|",
        "not": "~",
    }
    negcmp = {
        "lt": "ge",
        "le": "gt",
        "ge": "lt",
        "gt": "le",
    }

    def fix_invert(
        idxcmp: tuple[Any, str, Any] | tuple[None, None, None],
        exprnode: ne.expressions.ExpressionNode,
        indexedcols: frozenset[str],
    ) -> tuple[
        tuple[Any, str, Any] | tuple[None, None, None],
        ne.expressions.ExpressionNode,
        bool,
    ]:
        invert = False
        # Loop until all leading negations have been dealt with
        while idxcmp[1] == "invert":
            invert ^= True
            # The information about the negated node is in first position
            exprnode = idxcmp[0]
            idxcmp = _get_indexable_cmp(exprnode, indexedcols)
        return idxcmp, exprnode, invert

    # Indexable variable-constant comparison.
    idxcmp = _get_indexable_cmp(exprnode, indexedcols)
    idxcmp, exprnode, invert = fix_invert(idxcmp, exprnode, indexedcols)
    if idxcmp[0]: