feat(bigquery): add EXPORT DATA statement support #4688
Conversation
|
Hey @ArnoldHueteG, thank you for your contribution! Could you please run |
ArnoldHueteG
force-pushed
the
bigquery-add-support-export
branch
from
5e34015 to
05fc102
Compare
| ID_VAR_TOKENS = { | ||
| *parser.Parser.ID_VAR_TOKENS, | ||
| TokenType.EXPORT, | ||
| } |
There was a problem hiding this comment.
Add this token in the base class' set instead of copying it here.
| parts = [] | ||
| while True: | ||
| part = self._parse_var() | ||
| if not part: | ||
| break | ||
| parts.append(part.name) | ||
| if not self._match(TokenType.DOT): | ||
| break | ||
|
|
||
| if not parts: | ||
| self.raise_error("Expected connection name after WITH CONNECTION") | ||
|
|
||
| with_connection = exp.Identifier(this=".".join(parts)) |
There was a problem hiding this comment.
Let's not parse the connection name manually here, consider using _parse_table_parts instead.
| if self._match_text_seq("OPTIONS"): | ||
| self._match(TokenType.L_PAREN) | ||
| options = self._parse_properties() | ||
| self._match(TokenType.R_PAREN) | ||
| else: | ||
| self.raise_error("Expected 'OPTIONS' after 'EXPORT DATA'") |
There was a problem hiding this comment.
We should be able to clean this up using the OPTIONS property parser.
| else: | ||
| self.raise_error("Expected 'OPTIONS' after 'EXPORT DATA'") | ||
|
|
||
| self._match_text_seq("AS") |
There was a problem hiding this comment.
| self._match_text_seq("AS") | |
| self._match(TokenType.ALIAS) |
| # Parse the full SELECT statement | ||
| query = self._parse_statement() | ||
| if not isinstance(query, exp.Select): | ||
| self.raise_error("Expected SELECT statement in EXPORT DATA") | ||
|
|
There was a problem hiding this comment.
You don't have to do error handling like this, SQLGlot is intentionally lenient in these cases.
|
Hey @ArnoldHueteG, thank you for the contribution & iterations, I have some bandwidth available so I'll drive this to the finish line. |
Add support for BigQuery's EXPORT DATA statement including: