Failing to parse some CREATE TABLE statements with generated columns

[bholzer]

I am trying to parse table create statements used for Databricks, and they include generated columns. The example query from the Databricks documentation fails:

import sqlglot

q = """
CREATE TABLE default.people10m (
  id INT,
  firstName STRING,
  middleName STRING,
  lastName STRING,
  gender STRING,
  birthDate TIMESTAMP,
  dateOfBirth DATE GENERATED ALWAYS AS (CAST(birthDate AS DATE)),
  ssn STRING,
  salary INT
)
"""

sqlglot.parse_one(q, read="databricks")

sqlglot.errors.ParseError: Expecting ) on the AS of GENERATED ALWAYS AS

[georgesittas]

Thanks for the report, will take a look soon

[bholzer]

Big thanks, impressive turnaround here 👍