Support boolean casting and other trino operation for Trino SQL execution

[wmaiouiru]

I am trying to use sqlglot for unit testing complex queries with test case data.
However, Casting to 'Type.BOOLEAN' not implemented.

from sqlglot.executor import execute

tables = {
    "dates": [
        {"duration": "P4Y11M17D"},
        {"duration": "P6M1D"},
        {"duration": "P8M"},
    ],
}
sql_query = """
    SELECT sac.duration,
    CASE WHEN CAST(POSITION('D' IN duration) AS BOOLEAN) THEN 
        CAST(REGEXP_REPLACE(SUBSTR(duration, POSITION('D' IN duration) -2 , 2), '[^0-9]', '') AS INTEGER)
    ELSE
        0
    END as duration_day
    FROM dates sac
"""

schema = {
    "dates": {
        "duration": "VARCHAR",
    }
}


result = execute(sql_query, schema=schema, tables=tables, read='trino')
print(result)

Curious if this could be easily supported... any guides on how to support this?

[tobymao]

i wouldn't use sqlglot for unit testing complex queries, it's in a "toy" state. you should probably use duckdb or pyspark