Snowflake Table Generator during Execute

[brentonmallen1]

It seems like the execution module doesn't support the Snowflake Generator function.

minimal reproducible sample:

import sqlglot
from sqlglot.executor import execute

a = '''select * from table(generator(rowcount => CAST(10 AS DOUBLE)))'''
a_ast = sqlglot.parse_one(a, dialect=dialect)
execute(a_ast, dialect=dialect)

Yields: ExecuteError: Step 'Scan: _q_0 (#########)' failed: 'NoneType' object has no attribute 'range_reader'

It seems like it doesn't get recognized as a Table expression.

Am I approaching this incorrectly?

Thanks for any insight!

[georgesittas]

Am I approaching this incorrectly?

This is not supported by the executor today, so this seems expected.

[brentonmallen1]

I appreciate the response. Do you have any suggestions on how I might possibly create a workaround and/or point me to a couple of places in the codebase so that I might be able to contribute supporting this? Presumably, it would be some combination of the context/python/table files in the executor module, and/or maybe the snowflake dialect file?

[georgesittas]

You would have to start by adding parsing/generation support for the GENERATOR clause. Then, you would have to implement its semantics in the executor, but I'm not sure what exactly that'd entail. I'm also not sure how the planner deals with table functions at the moment, perhaps you may also have to augment it in order to take them into account?