Fix: parse query modifiers for ddl selects (#1703)

tobymao · May 30, 2023 · 611c234 · 611c234
1 parent 910166c
commit 611c234
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/sqlglot/parser.py b/sqlglot/parser.py
@@ -4135,8 +4135,8 @@ def _parse_select_or_expression(self, alias: bool = False) -> t.Optional[exp.Exp
         )
 
     def _parse_ddl_select(self) -> t.Optional[exp.Expression]:
-        return self._parse_set_operations(
-            self._parse_select(nested=True, parse_subquery_alias=False)
+        return self._parse_query_modifiers(
+            self._parse_set_operations(self._parse_select(nested=True, parse_subquery_alias=False))
         )
 
     def _parse_transaction(self) -> exp.Expression:

diff --git a/tests/dialects/test_bigquery.py b/tests/dialects/test_bigquery.py
@@ -31,6 +31,9 @@ def test_bigquery(self):
         self.validate_identity(
             "SELECT * FROM (SELECT * FROM `t`) AS a UNPIVOT((c) FOR c_name IN (v1, v2))"
         )
+        self.validate_identity(
+            "CREATE TABLE IF NOT EXISTS foo AS SELECT * FROM bla EXCEPT DISTINCT (SELECT * FROM bar) LIMIT 0"
+        )
 
         self.validate_all('x <> ""', write={"bigquery": "x <> ''"})
         self.validate_all('x <> """"""', write={"bigquery": "x <> ''"})