Class ColumnExpressionExt

test/sql/test_syntax_extensions.py:103–118

class ColumnExpressionExt(SyntaxExtension, ClauseElement):
    _traverse_internals: _TraverseInternalsType = [
        ("_exprs", InternalTraversal.dp_clauseelement_tuple),
    ]

    def __init__(self, *exprs):
        self._exprs = tuple(
            coercions.expect(roles.ByOfRole, e, apply_propagate_attrs=self)
            for e in exprs
        )

    def apply_to_select(self, select_stmt):
        select_stmt.apply_syntax_extension_point(
            lambda existing: [*existing, self],
            "post_select",
        )


@compiles(PostSelectClause)