analyze_query crashes the process on PIVOT over a CTE

[kvlonge]

Summary

analyze_query() crashes the Python process (segmentation fault, apparently unbounded recursion) when a PIVOT reads from a CTE. The same PIVOT over a physical table works.

Reproduction

pip install polyglot-sql==0.12.1

import polyglot_sql

sql = """
WITH pivot_input AS (
    SELECT customer_id, category, amount FROM staged_orders
)
SELECT * FROM pivot_input
PIVOT(MAX(amount) FOR category IN ('books', 'games'))
"""
polyglot_sql.analyze_query(sql, {"dialect": "snowflake"})

Actual

The interpreter exits with Segmentation fault (core dumped). No exception is raised. The dynamic form IN (ANY ORDER BY category) crashes the same way.

Replacing FROM pivot_input with FROM staged_orders returns normally.

Expected

analyze_query() returns an analysis, or raises a catchable error, for a PIVOT whose input is a CTE. It must not terminate the host process.

[tobilg]

Thanks for reporting, this will be fixed in the 0.13.0 release that will be done today. Please check and close if this fixed your issue.