class CursorFetchSegment(BaseSegment):
"""This is a FETCH statement.
https://dev.mysql.com/doc/refman/8.0/en/fetch.html
"""
type = "cursor_fetch_segment"
match_grammar = Sequence(
"FETCH",
Sequence(Ref.keyword("NEXT", optional=True), "FROM", optional=True),
Ref("NakedIdentifierSegment"),
"INTO",
Delimited(
Ref("SessionVariableNameSegment"),
Ref("LocalVariableNameSegment"),
),
)
class FromUnpivotExpressionSegment(BaseSegment):
"""An UNPIVOT expression."""
type = "from_unpivot_expression"
match_grammar = Sequence("UNPIVOT", Bracketed(Anything()))
parse_grammar = Sequence(
"UNPIVOT",
Bracketed(
Ref("SingleIdentifierGrammar"),
"FOR",
Ref("SingleIdentifierGrammar"),
"IN",
Bracketed(
Delimited(Ref("SingleIdentifierGrammar"),
delimiter=Ref("CommaSegment"))),
),
)
class TypelessStructSegment(BaseSegment):
"""Expression to construct a STRUCT with implicit types.
https://cloud.google.com/bigquery/docs/reference/standard-sql/data-types#typeless_struct_syntax
"""
type = "typeless_struct"
match_grammar = Sequence(
"STRUCT",
Bracketed(
Delimited(
Sequence(
Ref("BaseExpressionElementGrammar"),
Ref("AliasExpressionSegment", optional=True),
),
),
),
)
class DropProcedureStatementSegment(BaseSegment):
"""An `DROP PROCEDURE` statement.
https://docs.aws.amazon.com/redshift/latest/dg/r_DROP_PROCEDURE.html
"""
type = "drop_procedure_statement"
match_grammar = Sequence(
"DROP",
"PROCEDURE",
Ref("IfExistsGrammar", optional=True),
Delimited(
Sequence(
Ref("FunctionNameSegment"),
Ref("ProcedureParameterListSegment", optional=True),
), ),
)
class CreateExternalTableAsStatementSegment(BaseSegment):
"""A `CREATE EXTERNAL TABLE AS` statement.
https://docs.aws.amazon.com/redshift/latest/dg/r_CREATE_EXTERNAL_TABLE.html
"""
type = "create_external_table_statement"
match_grammar = Sequence(
"CREATE",
"EXTERNAL",
"TABLE",
Ref("TableReferenceSegment"),
Ref("PartitionedBySegment", optional=True),
Sequence(
"ROW",
"FORMAT",
"DELIMITED",
Ref("RowFormatDelimitedSegment"),
optional=True,
),
"STORED",
"AS",
OneOf(
"PARQUET",
"TEXTFILE",
),
"LOCATION",
Ref("QuotedLiteralSegment"),
Sequence(
"TABLE",
"PROPERTIES",
Bracketed(
Delimited(
Sequence(
Ref("QuotedLiteralSegment"),
Ref("EqualsSegment"),
Ref("QuotedLiteralSegment"),
), ), ),
optional=True,
),
"AS",
OptionallyBracketed(Ref("SelectableGrammar")),
)
class ReplaceClauseSegment(BaseSegment):
"""SELECT REPLACE clause."""
type = "select_replace_clause"
match_grammar = Sequence(
"REPLACE",
OneOf(
# Multiple replace in brackets
Bracketed(
Delimited(
# Not *really* a select target element. It behaves exactly
# the same way however.
Ref("SelectClauseElementSegment"),
delimiter=Ref("CommaSegment"),
)),
# Single replace not in brackets.
Ref("SelectClauseElementSegment"),
),
)
class PartitionedBySegment(BaseSegment):
"""Partitioned By Segment.
As specified in
https://docs.aws.amazon.com/redshift/latest/dg/r_CREATE_EXTERNAL_TABLE.html
"""
type = "partitioned_by_segment"
match_grammar = Sequence(
Ref.keyword("PARTITIONED"),
"BY",
Bracketed(
Delimited(
Sequence(
Ref("ColumnReferenceSegment"),
Ref("DatatypeSegment"),
), ), ),
)
class CreateGroupStatementSegment(BaseSegment):
"""`CREATE GROUP` statement.
https://docs.aws.amazon.com/redshift/latest/dg/r_CREATE_GROUP.html
"""
type = "create_group"
match_grammar = Sequence(
"CREATE",
"GROUP",
Ref("ObjectReferenceSegment"),
Sequence(
Ref.keyword("WITH", optional=True),
"USER",
Delimited(Ref("ObjectReferenceSegment"), ),
optional=True,
),
)
class ForClauseSegment(BaseSegment):
"""This is the body of a `FOR` clause."""
type = "for_clause"
match_grammar = OneOf(
Sequence(
Sequence(
"FOR",
OneOf("UPDATE", "SHARE"),
),
Sequence("OF",
Delimited(Ref("NakedIdentifierSegment")),
optional=True),
OneOf("NOWAIT", Sequence("SKIP", "LOCKED"), optional=True),
),
Sequence("LOCK", "IN", "SHARE", "MODE"),
optional=True,
)
class FunctionDefinitionGrammar(BaseSegment):
"""This is the body of a `CREATE FUNCTION AS` statement."""
type = "function_definition"
match_grammar = Sequence(
AnyNumberOf(
Sequence(
OneOf("DETERMINISTIC", Sequence("NOT", "DETERMINISTIC")),
optional=True,
),
Sequence(
"LANGUAGE",
# Not really a parameter, but best fit for now.
Ref("ParameterNameSegment"),
Sequence(
"OPTIONS",
Bracketed(
Delimited(
Sequence(
Ref("ParameterNameSegment"),
Ref("EqualsSegment"),
Anything(),
),
delimiter=Ref("CommaSegment"),
)
),
optional=True,
),
),
# There is some syntax not implemented here,
Sequence(
"AS",
OneOf(
Ref("DoubleQuotedUDFBody"),
Ref("SingleQuotedUDFBody"),
Bracketed(
OneOf(Ref("ExpressionSegment"), Ref("SelectStatementSegment"))
),
),
),
)
)
class DeclareStatementSegment(BaseSegment):
"""Declaration of a variable.
https://cloud.google.com/bigquery/docs/reference/standard-sql/scripting#declare
"""
type = "declare_segment"
match_grammar = StartsWith("DECLARE")
parse_grammar = Sequence(
"DECLARE",
Delimited(Ref("NakedIdentifierSegment")),
OneOf(
Ref("DatatypeSegment"),
Ref("DefaultDeclareOptionsGrammar"),
Sequence(
Ref("DatatypeSegment"),
Ref("DefaultDeclareOptionsGrammar"),
),
),
)
class DatatypeSegment(ansi.DatatypeSegment):
"""A data type segment.
DATE FORMAT 'YYYY-MM-DD'
"""
match_grammar = Sequence(
Ref("DatatypeIdentifierSegment"),
Bracketed(
OneOf(
Delimited(Ref("ExpressionSegment")),
# The brackets might be empty for some cases...
optional=True,
),
# There may be no brackets for some data types
optional=True,
),
Sequence( # FORMAT 'YYYY-MM-DD',
"FORMAT", Ref("QuotedLiteralSegment"), optional=True
),
)
class SelectClauseSegment(AnsiSelectClauseSegment):
"""In BigQuery, select * as struct is valid."""
parse_grammar = Sequence(
"SELECT",
Ref("SelectClauseModifierSegment", optional=True),
Indent,
OneOf(
Sequence(
"AS",
"STRUCT",
Ref("StarSegment"),
Ref("StarModifierSegment", optional=True),
),
Delimited(
Ref("SelectTargetElementSegment"),
delimiter=Ref("CommaSegment"),
allow_trailing=True,
),
),
)
class CreateScriptingLuaScriptStatementSegment(BaseSegment):
"""`CREATE SCRIPT` statement to create a Lua scripting script.
https://docs.exasol.com/sql/create_script.htm
"""
type = "create_scripting_lua_script"
is_ddl = True
is_dml = False
is_dql = False
is_dcl = False
match_grammar = StartsWith(
Sequence(
"CREATE",
Ref("OrReplaceGrammar", optional=True),
Ref.keyword("LUA", optional=True),
"SCRIPT",
))
parse_grammar = Sequence(
"CREATE",
Ref("OrReplaceGrammar", optional=True),
Ref.keyword("LUA", optional=True),
"SCRIPT",
Ref("ScriptReferenceSegment"),
Bracketed(
Delimited(
Sequence(Ref.keyword("ARRAY", optional=True),
Ref("SingleIdentifierGrammar")),
optional=True,
),
optional=True,
),
Sequence(Ref.keyword("RETURNS"),
OneOf("TABLE", "ROWCOUNT"),
optional=True),
"AS",
Ref("ScriptContentSegment"),
)
class FileSegment(BaseFileSegment):
"""A segment representing a whole file or script.
This is also the default "root" segment of the dialect,
and so is usually instantiated directly. It therefore
has no match_grammar.
Override ANSI to allow addition of ExecuteFileSegment without
ending in DelimiterGrammar
"""
# NB: We don't need a match_grammar here because we're
# going straight into instantiating it directly usually.
parse_grammar = AnyNumberOf(
Ref("ExecuteFileSegment"),
Delimited(
Ref("StatementSegment"),
delimiter=AnyNumberOf(Ref("DelimiterGrammar"), min_times=1),
allow_gaps=True,
allow_trailing=True,
),
)
class CreateTableStatementSegment(BaseSegment):
"""A `CREATE [MULTISET| SET] TABLE` statement."""
type = "create_table_statement"
match_grammar = Sequence(
"CREATE",
Sequence("OR", "REPLACE", optional=True),
# Adding Teradata specific [MULTISET| SET]
OneOf("SET", "MULTISET", optional=True),
OneOf(Sequence("GLOBAL", "TEMPORARY"), "VOLATILE", optional=True),
"TABLE",
Sequence("IF", "NOT", "EXISTS", optional=True),
Ref("TableReferenceSegment"),
# , NO FALLBACK, NO BEFORE JOURNAL, NO AFTER JOURNAL
OneOf(Ref("TdCreateTableOptions"), optional=True),
OneOf(
# Columns and comment syntax:
Sequence(
Bracketed(
Delimited(
OneOf(
Ref("ColumnDefinitionSegment"),
Ref("TableConstraintSegment"),
),
delimiter=Ref("CommaSegment"),
)),
Sequence( # [COMMENT 'string'] (MySQL)
"COMMENT",
Ref("QuotedLiteralSegment"),
optional=True),
),
# Create AS syntax:
Sequence("AS", Ref("SelectableGrammar")),
# Create like syntax
Sequence("LIKE", Ref("TableReferenceSegment")),
),
# PRIMARY INDEX( COD_TARJETA, COD_EST, IND_TIPO_TARJETA, FEC_ANIO_MES )
OneOf(Ref("TdTableConstraints"), optional=True),
)
class CreateTableStatementSegment(BaseSegment):
"""A `CREATE TABLE` statement.
As specified in
https://docs.aws.amazon.com/redshift/latest/dg/r_CREATE_TABLE_NEW.html
"""
type = "create_table_statement"
match_grammar = Sequence(
"CREATE",
Ref.keyword("LOCAL", optional=True),
Ref("TemporaryGrammar", optional=True),
"TABLE",
Ref("IfNotExistsGrammar", optional=True),
Ref("TableReferenceSegment"),
Bracketed(
OneOf(
# Columns and comment syntax:
Delimited(
Sequence(
Ref("ColumnReferenceSegment"),
Ref("DatatypeSegment"),
AnyNumberOf(Ref("ColumnAttributeSegment"),
optional=True),
AnyNumberOf(Ref("ColumnConstraintSegment"),
optional=True),
),
Ref("TableConstraintSegment", optional=True),
),
Sequence(
"LIKE",
Ref("TableReferenceSegment"),
AnyNumberOf(Ref("LikeOptionSegment"), optional=True),
),
)),
Sequence("BACKUP", OneOf("YES", "NO", optional=True), optional=True),
AnyNumberOf(Ref("TableAttributeSegment"), optional=True),
)
class SelectClauseModifierSegment(BaseSegment):
"""Things that come after SELECT but before the columns."""
type = "select_clause_modifier"
match_grammar = OneOf(
Sequence("DISTINCT",
Sequence("ON", Bracketed(Anything()), optional=True)),
"ALL",
)
parse_grammar = OneOf(
Sequence(
"DISTINCT",
Sequence(
"ON",
Bracketed(
Delimited(Ref("ExpressionSegment"),
delimiter=Ref("CommaSegment"))),
optional=True,
),
),
"ALL",
)
class DeclareStatementSegment(BaseSegment):
"""Declaration of a variable.
https://docs.microsoft.com/en-us/sql/t-sql/language-elements/declare-local-variable-transact-sql?view=sql-server-ver15
"""
type = "declare_segment"
match_grammar = StartsWith("DECLARE")
parse_grammar = Sequence(
"DECLARE",
Delimited(Ref("ParameterNameSegment")),
Ref("DatatypeSegment"),
Sequence(
Ref("EqualsSegment"),
OneOf(
Ref("LiteralGrammar"),
Bracketed(Ref("SelectStatementSegment")),
Ref("BareFunctionSegment"),
Ref("FunctionSegment"),
),
optional=True,
),
)
class DatatypeSegment(BaseSegment):
"""A data type segment.
In particular here, this enabled the support for
the STRUCT datatypes.
"""
type = "data_type"
match_grammar = OneOf( # Parameter type
Ref("DatatypeIdentifierSegment"), # Simple type
Sequence("ANY", "TYPE"), # SQL UDFs can specify this "type"
Sequence("ARRAY", Bracketed(Ref("DatatypeSegment"), bracket_type="angle")),
Sequence(
"STRUCT",
Bracketed(
Delimited( # Comma-separated list of field names/types
Sequence(Ref("ParameterNameSegment"), Ref("DatatypeSegment")),
delimiter=Ref("CommaSegment"),
),
bracket_type="angle",
),
),
)
class DeclareStatementSegment(BaseSegment):
"""Declaration of a variable.
https://cloud.google.com/bigquery/docs/reference/standard-sql/scripting#declare
"""
type = "declare_segment"
match_grammar = StartsWith("DECLARE")
parse_grammar = Sequence(
"DECLARE",
Delimited(Ref("NakedIdentifierSegment")),
Ref("DatatypeIdentifierSegment"),
Sequence(
"DEFAULT",
OneOf(
Ref("LiteralGrammar"),
Bracketed(Ref("SelectStatementSegment")),
Ref("BareFunctionSegment"),
Ref("FunctionSegment"),
),
optional=True,
),
)
class ClusterByClauseSegment(ansi.OrderByClauseSegment):
"""A `CLUSTER BY` clause like in `SELECT`."""
type = "clusterby_clause"
match_grammar: Matchable = StartsWith(
Sequence("CLUSTER", "BY"),
terminator=ansi.OrderByClauseSegment.match_grammar.terminator, # type: ignore
)
parse_grammar: Optional[Matchable] = Sequence(
"CLUSTER",
"BY",
Indent,
Delimited(
Sequence(
OneOf(
Ref("ColumnReferenceSegment"),
Ref("NumericLiteralSegment"),
Ref("ExpressionSegment"),
),
),
terminator=OneOf(Ref.keyword("LIMIT"), Ref("FrameClauseUnitGrammar")),
),
Dedent,
)
class LoopStatementSegment(BaseSegment):
"""A `LOOP` statement.
https://dev.mysql.com/doc/refman/8.0/en/loop.html
"""
type = "loop_statement"
match_grammar = OneOf(
Sequence(
Sequence(
Ref("SingleIdentifierGrammar"), Ref("ColonSegment"), optional=True
),
"LOOP",
Delimited(
Ref("StatementSegment"),
),
),
Sequence(
"END",
"LOOP",
Ref("SingleIdentifierGrammar", optional=True),
),
)
class ExplainStatementSegment(
ansi_dialect.get_segment("ExplainStatementSegment")): # type: ignore
"""An `Explain` statement.
EXPLAIN [ ( option [, ...] ) ] statement
EXPLAIN [ ANALYZE ] [ VERBOSE ] statement
https://www.postgresql.org/docs/9.1/sql-explain.html
"""
parse_grammar = Sequence(
"EXPLAIN",
OneOf(
Sequence(
Ref.keyword("ANALYZE", optional=True),
Ref.keyword("VERBOSE", optional=True),
),
Bracketed(
Delimited(Ref("ExplainOptionSegment"),
delimiter=Ref("CommaSegment"))),
optional=True,
),
ansi_dialect.get_segment("ExplainStatementSegment").explainable_stmt,
)
class AnalyzeCompressionStatementSegment(BaseSegment):
"""An `ANALYZE COMPRESSION` statement.
https://docs.aws.amazon.com/redshift/latest/dg/r_ANALYZE_COMPRESSION.html
"""
type = "analyze_compression_statement"
match_grammar = Sequence(
OneOf("ANALYZE", "ANALYSE"),
"COMPRESSION",
Sequence(
Ref("TableReferenceSegment"),
Bracketed(
Delimited(Ref("ColumnReferenceSegment"), ),
optional=True,
),
Sequence(
"COMPROWS",
Ref("NumericLiteralSegment"),
optional=True,
),
optional=True,
),
)
class TdCollectStatisticsStatementSegment(BaseSegment):
"""A `COLLECT STATISTICS (Optimizer Form)` statement.
# TODO: add expression
COLLECT [SUMMARY] (STATISTICS|STAT) [[COLUMN| [UNIQUE] INDEX] (expression (, expression ...)] ON TABLENAME
"""
type = "collect_statistics_statement"
match_grammar = Sequence(
"COLLECT",
Ref.keyword("SUMMARY", optional=True),
OneOf("STATISTICS", "STAT"),
Sequence(
"USING",
Delimited(
Ref("TdCollectStatUsingOptionClauseSegment"),
delimiter="AND",
),
optional=True,
),
Delimited(
OneOf(
# UNIQUE INDEX index_name ALL (column_name, ...) ORDER BY VALUES|HASH (column_name)
Sequence(
Ref.keyword("UNIQUE", optional=True),
"INDEX",
Ref("IndexReferenceSegment", optional=True),
Ref.keyword("ALL", optional=True),
Bracketed(
Delimited(Ref("ColumnReferenceSegment"),
delimiter=Ref("CommaSegment"))),
Ref("TdOrderByStatClauseSegment", optional=True),
),
# UNIQUE INDEX index_name
Sequence(
Ref.keyword("UNIQUE", optional=True),
"INDEX",
Ref("IndexReferenceSegment"),
),
# COLUMN ...
Sequence(
"COLUMN",
Bracketed(
Delimited(
OneOf(
Ref("ColumnReferenceSegment"),
Ref.keyword("PARTITION"),
# TODO: expression
),
delimiter=Ref("CommaSegment"),
), ),
Sequence(
Ref.keyword("AS", optional=True),
Ref("ObjectReferenceSegment"), # statistics_name
optional=True,
),
),
),
delimiter=Ref("CommaSegment"),
optional=True,
),
"ON",
Ref.keyword("TEMPORARY", optional=True),
Ref("TableReferenceSegment"),
)
class CreateExternalTableStatementSegment(BaseSegment):
"""A `CREATE EXTERNAL TABLE` statement.
https://docs.aws.amazon.com/redshift/latest/dg/r_CREATE_EXTERNAL_TABLE.html
"""
type = "create_external_table_statement"
match_grammar = Sequence(
"CREATE",
"EXTERNAL",
"TABLE",
Ref("TableReferenceSegment"),
Bracketed(
# Columns and comment syntax:
Delimited(
Sequence(
Ref("ColumnReferenceSegment"),
Ref("DatatypeSegment"),
), ), ),
Ref("PartitionedBySegment", optional=True),
Sequence(
"ROW",
"FORMAT",
OneOf(
Sequence(
"DELIMITED",
Ref("RowFormatDelimitedSegment"),
),
Sequence(
"SERDE",
Ref("QuotedLiteralSegment"),
Sequence(
"WITH",
"SERDEPROPERTIES",
Bracketed(
Delimited(
Sequence(
Ref("QuotedLiteralSegment"),
Ref("EqualsSegment"),
Ref("QuotedLiteralSegment"),
), ), ),
optional=True,
),
),
),
optional=True,
),
"STORED",
"AS",
OneOf(
"PARQUET",
"RCFILE",
"SEQUENCEFILE",
"TEXTFILE",
"ORC",
"AVRO",
Sequence(
"INPUTFORMAT",
Ref("QuotedLiteralSegment"),
"OUTPUTFORMAT",
Ref("QuotedLiteralSegment"),
),
),
"LOCATION",
Ref("QuotedLiteralSegment"),
Sequence(
"TABLE",
"PROPERTIES",
Bracketed(
Delimited(
Sequence(
Ref("QuotedLiteralSegment"),
Ref("EqualsSegment"),
Ref("QuotedLiteralSegment"),
), ), ),
optional=True,
),
)
class CreateModelStatementSegment(BaseSegment):
"""A `CREATE MODEL` statement.
https://docs.aws.amazon.com/redshift/latest/dg/r_CREATE_MODEL.html
NB: order of keywords matter
"""
type = "create_model_statement"
match_grammar = Sequence(
"CREATE",
"MODEL",
Ref("ObjectReferenceSegment"),
Sequence(
"FROM",
OneOf(
Ref("QuotedLiteralSegment"),
Bracketed(Ref("SelectableGrammar")),
Ref("ObjectReferenceSegment"),
),
optional=True,
),
Sequence(
"TARGET",
Ref("ColumnReferenceSegment"),
optional=True,
),
Sequence(
"FUNCTION",
Ref("ObjectReferenceSegment"),
Bracketed(
Delimited(Ref("DatatypeSegment")),
optional=True,
),
),
Sequence(
"RETURNS",
Ref("DatatypeSegment"),
optional=True,
),
Sequence(
"SAGEMAKER",
Ref("QuotedLiteralSegment"),
optional=True,
),
Sequence(
"IAM_ROLE",
OneOf(
"DEFAULT",
Ref("QuotedLiteralSegment"),
),
),
Sequence(
"AUTO",
OneOf(
"ON",
"OFF",
),
optional=True,
),
Sequence(
"MODEL_TYPE",
OneOf(
"XGBOOST",
"MLP",
"KMEANS",
),
optional=True,
),
Sequence(
"PROBLEM_TYPE",
OneOf(
"REGRESSION",
"BINARY_CLASSIFICATION",
"MULTICLASS_CLASSIFICATION",
),
optional=True,
),
Sequence(
"OBJECTIVE",
Ref("QuotedLiteralSegment"),
optional=True,
),
Sequence(
"PREPROCESSORS",
Ref("QuotedLiteralSegment"),
optional=True,
),
Sequence(
"HYPERPARAMETERS",
"DEFAULT",
Sequence(
"EXCEPT",
Bracketed(Delimited(Anything(), ), ),
optional=True,
),
optional=True,
),
Sequence(
"SETTINGS",
Bracketed(
Sequence(
"S3_BUCKET",
Ref("QuotedLiteralSegment"),
Sequence(
"KMS_KEY_ID",
Ref("QuotedLiteralSegment"),
optional=True,
),
Sequence(
"S3_GARBAGE_COLLECT",
OneOf(
"ON",
"OFF",
),
optional=True,
),
Sequence(
"MAX_CELLS",
Ref("NumericLiteralSegment"),
optional=True,
),
Sequence(
"MAX_RUNTIME",
Ref("NumericLiteralSegment"),
optional=True,
),
), ),
optional=True,
),
)
class DatatypeSegment(BaseSegment):
"""A data type segment.
Indicates a data type.
https://docs.aws.amazon.com/redshift/latest/dg/c_Supported_data_types.html
"""
type = "data_type"
match_grammar = OneOf(
# numeric types
"SMALLINT",
"INT2",
"INTEGER",
"INT",
"INT4",
"BIGINT",
"INT8",
"REAL",
"FLOAT4",
Sequence("DOUBLE", "PRECISION"),
"FLOAT8",
"FLOAT",
# numeric types [precision ["," scale])]
Sequence(
OneOf("DECIMAL", "NUMERIC"),
Bracketed(
Delimited(Ref("NumericLiteralSegment")),
optional=True,
),
),
# character types
OneOf(
Sequence(
OneOf(
"CHAR",
"CHARACTER",
"NCHAR",
"VARCHAR",
Sequence("CHARACTER", "VARYING"),
"NVARCHAR",
),
Bracketed(
OneOf(
Ref("NumericLiteralSegment"),
"MAX",
),
optional=True,
),
),
"BPCHAR",
"TEXT",
),
Ref("DateTimeTypeIdentifier"),
# INTERVAL is a data type *only* for conversion operations
"INTERVAL",
# boolean types
OneOf("BOOLEAN", "BOOL"),
# hllsketch type
"HLLSKETCH",
# super type
"SUPER",
# spatial data
"GEOMETRY",
"GEOGRAPHY",
# binary type
Sequence(
OneOf(
"VARBYTE",
"VARBINARY",
Sequence("BINARY", "VARYING"),
),
Bracketed(
Ref("NumericLiteralSegment"),
optional=True,
),
),
)
class AltereDatashareStatementSegment(BaseSegment):
"""An `ALTER DATASHARE` statement.
https://docs.aws.amazon.com/redshift/latest/dg/r_ALTER_DATASHARE.html
"""
type = "create_datashare_statement"
match_grammar = Sequence(
"ALTER",
"DATASHARE",
Ref("ObjectReferenceSegment"),
OneOf(
# add or remove objects to the datashare
Sequence(
OneOf(
"ADD",
"REMOVE",
),
OneOf(
Sequence(
"TABLE",
Delimited(Ref("TableReferenceSegment")),
),
Sequence(
"SCHEMA",
Delimited(Ref("SchemaReferenceSegment")),
),
Sequence(
"FUNCTION",
Delimited(Ref("FunctionNameSegment")),
),
Sequence(
"ALL",
OneOf("TABLES", "FUNCTIONS"),
"IN",
"SCHEMA",
Delimited(Ref("SchemaReferenceSegment")),
),
),
),
# configure the properties of the datashare
Sequence(
"SET",
OneOf(
Sequence(
"PUBLICACCESSIBLE",
Ref("EqualsSegment", optional=True),
Ref("BooleanLiteralGrammar"),
),
Sequence(
"INCLUDENEW",
Ref("EqualsSegment", optional=True),
Ref("BooleanLiteralGrammar"),
"FOR",
"SCHEMA",
Ref("SchemaReferenceSegment"),
),
),
),
),
)