Merging upstream version 17.9.1.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
2bf6699c56
commit
9777880e00
GPG key ID:
FBB4F0E80A80222F
87 changed files with 45907 additions and 42511 deletions
|
|
@ -67,8 +67,9 @@ class Expression(metaclass=_Expression):
|
|||
uses to refer to it.
|
||||
comments: a list of comments that are associated with a given expression. This is used in
|
||||
order to preserve comments when transpiling SQL code.
|
||||
_type: the `sqlglot.expressions.DataType` type of an expression. This is inferred by the
|
||||
type: the `sqlglot.expressions.DataType` type of an expression. This is inferred by the
|
||||
optimizer, in order to enable some transformations that require type information.
|
||||
meta: a dictionary that can be used to store useful metadata for a given expression.
|
||||
|
||||
Example:
|
||||
>>> class Foo(Expression):
|
||||
|
|
@ -767,7 +768,7 @@ class Condition(Expression):
|
|||
**opts,
|
||||
) -> In:
|
||||
return In(
|
||||
this=_maybe_copy(self, copy),
|
||||
this=maybe_copy(self, copy),
|
||||
expressions=[convert(e, copy=copy) for e in expressions],
|
||||
query=maybe_parse(query, copy=copy, **opts) if query else None,
|
||||
unnest=Unnest(
|
||||
|
|
@ -781,7 +782,7 @@ class Condition(Expression):
|
|||
|
||||
def between(self, low: t.Any, high: t.Any, copy: bool = True, **opts) -> Between:
|
||||
return Between(
|
||||
this=_maybe_copy(self, copy),
|
||||
this=maybe_copy(self, copy),
|
||||
low=convert(low, copy=copy, **opts),
|
||||
high=convert(high, copy=copy, **opts),
|
||||
)
|
||||
|
|
@ -990,7 +991,28 @@ class Uncache(Expression):
|
|||
arg_types = {"this": True, "exists": False}
|
||||
|
||||
|
||||
class Create(Expression):
|
||||
class DDL(Expression):
|
||||
@property
|
||||
def ctes(self):
|
||||
with_ = self.args.get("with")
|
||||
if not with_:
|
||||
return []
|
||||
return with_.expressions
|
||||
|
||||
@property
|
||||
def named_selects(self) -> t.List[str]:
|
||||
if isinstance(self.expression, Subqueryable):
|
||||
return self.expression.named_selects
|
||||
return []
|
||||
|
||||
@property
|
||||
def selects(self) -> t.List[Expression]:
|
||||
if isinstance(self.expression, Subqueryable):
|
||||
return self.expression.selects
|
||||
return []
|
||||
|
||||
|
||||
class Create(DDL):
|
||||
arg_types = {
|
||||
"with": False,
|
||||
"this": True,
|
||||
|
|
@ -1206,6 +1228,19 @@ class MergeTreeTTL(Expression):
|
|||
}
|
||||
|
||||
|
||||
# https://dev.mysql.com/doc/refman/8.0/en/create-table.html
|
||||
class IndexConstraintOption(Expression):
|
||||
arg_types = {
|
||||
"key_block_size": False,
|
||||
"using": False,
|
||||
"parser": False,
|
||||
"comment": False,
|
||||
"visible": False,
|
||||
"engine_attr": False,
|
||||
"secondary_engine_attr": False,
|
||||
}
|
||||
|
||||
|
||||
class ColumnConstraint(Expression):
|
||||
arg_types = {"this": False, "kind": True}
|
||||
|
||||
|
|
@ -1272,6 +1307,11 @@ class GeneratedAsIdentityColumnConstraint(ColumnConstraintKind):
|
|||
}
|
||||
|
||||
|
||||
# https://dev.mysql.com/doc/refman/8.0/en/create-table.html
|
||||
class IndexColumnConstraint(ColumnConstraintKind):
|
||||
arg_types = {"this": False, "schema": True, "kind": False, "type": False, "options": False}
|
||||
|
||||
|
||||
class InlineLengthColumnConstraint(ColumnConstraintKind):
|
||||
pass
|
||||
|
||||
|
|
@ -1496,7 +1536,7 @@ class JoinHint(Expression):
|
|||
|
||||
|
||||
class Identifier(Expression):
|
||||
arg_types = {"this": True, "quoted": False}
|
||||
arg_types = {"this": True, "quoted": False, "global": False, "temporary": False}
|
||||
|
||||
@property
|
||||
def quoted(self) -> bool:
|
||||
|
|
@ -1525,7 +1565,7 @@ class Index(Expression):
|
|||
}
|
||||
|
||||
|
||||
class Insert(Expression):
|
||||
class Insert(DDL):
|
||||
arg_types = {
|
||||
"with": False,
|
||||
"this": True,
|
||||
|
|
@ -1892,6 +1932,10 @@ class EngineProperty(Property):
|
|||
arg_types = {"this": True}
|
||||
|
||||
|
||||
class HeapProperty(Property):
|
||||
arg_types = {}
|
||||
|
||||
|
||||
class ToTableProperty(Property):
|
||||
arg_types = {"this": True}
|
||||
|
||||
|
|
@ -2182,7 +2226,7 @@ class Tuple(Expression):
|
|||
**opts,
|
||||
) -> In:
|
||||
return In(
|
||||
this=_maybe_copy(self, copy),
|
||||
this=maybe_copy(self, copy),
|
||||
expressions=[convert(e, copy=copy) for e in expressions],
|
||||
query=maybe_parse(query, copy=copy, **opts) if query else None,
|
||||
unnest=Unnest(
|
||||
|
|
@ -2212,7 +2256,7 @@ class Subqueryable(Unionable):
|
|||
Returns:
|
||||
Alias: the subquery
|
||||
"""
|
||||
instance = _maybe_copy(self, copy)
|
||||
instance = maybe_copy(self, copy)
|
||||
if not isinstance(alias, Expression):
|
||||
alias = TableAlias(this=to_identifier(alias)) if alias else None
|
||||
|
||||
|
|
@ -2865,7 +2909,7 @@ class Select(Subqueryable):
|
|||
self,
|
||||
expression: ExpOrStr,
|
||||
on: t.Optional[ExpOrStr] = None,
|
||||
using: t.Optional[ExpOrStr | t.List[ExpOrStr]] = None,
|
||||
using: t.Optional[ExpOrStr | t.Collection[ExpOrStr]] = None,
|
||||
append: bool = True,
|
||||
join_type: t.Optional[str] = None,
|
||||
join_alias: t.Optional[Identifier | str] = None,
|
||||
|
|
@ -2943,6 +2987,7 @@ class Select(Subqueryable):
|
|||
arg="using",
|
||||
append=append,
|
||||
copy=copy,
|
||||
into=Identifier,
|
||||
**opts,
|
||||
)
|
||||
|
||||
|
|
@ -3092,7 +3137,7 @@ class Select(Subqueryable):
|
|||
Returns:
|
||||
Select: the modified expression.
|
||||
"""
|
||||
instance = _maybe_copy(self, copy)
|
||||
instance = maybe_copy(self, copy)
|
||||
on = Tuple(expressions=[maybe_parse(on, copy=copy) for on in ons if on]) if ons else None
|
||||
instance.set("distinct", Distinct(on=on) if distinct else None)
|
||||
return instance
|
||||
|
|
@ -3123,7 +3168,7 @@ class Select(Subqueryable):
|
|||
Returns:
|
||||
The new Create expression.
|
||||
"""
|
||||
instance = _maybe_copy(self, copy)
|
||||
instance = maybe_copy(self, copy)
|
||||
table_expression = maybe_parse(
|
||||
table,
|
||||
into=Table,
|
||||
|
|
@ -3159,7 +3204,7 @@ class Select(Subqueryable):
|
|||
Returns:
|
||||
The modified expression.
|
||||
"""
|
||||
inst = _maybe_copy(self, copy)
|
||||
inst = maybe_copy(self, copy)
|
||||
inst.set("locks", [Lock(update=update)])
|
||||
|
||||
return inst
|
||||
|
|
@ -3181,7 +3226,7 @@ class Select(Subqueryable):
|
|||
Returns:
|
||||
The modified expression.
|
||||
"""
|
||||
inst = _maybe_copy(self, copy)
|
||||
inst = maybe_copy(self, copy)
|
||||
inst.set(
|
||||
"hint", Hint(expressions=[maybe_parse(h, copy=copy, dialect=dialect) for h in hints])
|
||||
)
|
||||
|
|
@ -3376,6 +3421,8 @@ class DataType(Expression):
|
|||
HSTORE = auto()
|
||||
IMAGE = auto()
|
||||
INET = auto()
|
||||
IPADDRESS = auto()
|
||||
IPPREFIX = auto()
|
||||
INT = auto()
|
||||
INT128 = auto()
|
||||
INT256 = auto()
|
||||
|
|
@ -3987,7 +4034,7 @@ class Case(Func):
|
|||
arg_types = {"this": False, "ifs": True, "default": False}
|
||||
|
||||
def when(self, condition: ExpOrStr, then: ExpOrStr, copy: bool = True, **opts) -> Case:
|
||||
instance = _maybe_copy(self, copy)
|
||||
instance = maybe_copy(self, copy)
|
||||
instance.append(
|
||||
"ifs",
|
||||
If(
|
||||
|
|
@ -3998,7 +4045,7 @@ class Case(Func):
|
|||
return instance
|
||||
|
||||
def else_(self, condition: ExpOrStr, copy: bool = True, **opts) -> Case:
|
||||
instance = _maybe_copy(self, copy)
|
||||
instance = maybe_copy(self, copy)
|
||||
instance.set("default", maybe_parse(condition, copy=copy, **opts))
|
||||
return instance
|
||||
|
||||
|
|
@ -4263,6 +4310,10 @@ class Initcap(Func):
|
|||
arg_types = {"this": True, "expression": False}
|
||||
|
||||
|
||||
class IsNan(Func):
|
||||
_sql_names = ["IS_NAN", "ISNAN"]
|
||||
|
||||
|
||||
class JSONKeyValue(Expression):
|
||||
arg_types = {"this": True, "expression": True}
|
||||
|
||||
|
|
@ -4549,6 +4600,11 @@ class StandardHash(Func):
|
|||
arg_types = {"this": True, "expression": False}
|
||||
|
||||
|
||||
class StartsWith(Func):
|
||||
_sql_names = ["STARTS_WITH", "STARTSWITH"]
|
||||
arg_types = {"this": True, "expression": True}
|
||||
|
||||
|
||||
class StrPosition(Func):
|
||||
arg_types = {
|
||||
"this": True,
|
||||
|
|
@ -4804,7 +4860,7 @@ def maybe_parse(
|
|||
return sqlglot.parse_one(sql, read=dialect, into=into, **opts)
|
||||
|
||||
|
||||
def _maybe_copy(instance: E, copy: bool = True) -> E:
|
||||
def maybe_copy(instance: E, copy: bool = True) -> E:
|
||||
return instance.copy() if copy else instance
|
||||
|
||||
|
||||
|
|
@ -4824,7 +4880,7 @@ def _apply_builder(
|
|||
):
|
||||
if _is_wrong_expression(expression, into):
|
||||
expression = into(this=expression)
|
||||
instance = _maybe_copy(instance, copy)
|
||||
instance = maybe_copy(instance, copy)
|
||||
expression = maybe_parse(
|
||||
sql_or_expression=expression,
|
||||
prefix=prefix,
|
||||
|
|
@ -4848,7 +4904,7 @@ def _apply_child_list_builder(
|
|||
properties=None,
|
||||
**opts,
|
||||
):
|
||||
instance = _maybe_copy(instance, copy)
|
||||
instance = maybe_copy(instance, copy)
|
||||
parsed = []
|
||||
for expression in expressions:
|
||||
if expression is not None:
|
||||
|
|
@ -4887,7 +4943,7 @@ def _apply_list_builder(
|
|||
dialect=None,
|
||||
**opts,
|
||||
):
|
||||
inst = _maybe_copy(instance, copy)
|
||||
inst = maybe_copy(instance, copy)
|
||||
|
||||
expressions = [
|
||||
maybe_parse(
|
||||
|
|
@ -4923,7 +4979,7 @@ def _apply_conjunction_builder(
|
|||
if not expressions:
|
||||
return instance
|
||||
|
||||
inst = _maybe_copy(instance, copy)
|
||||
inst = maybe_copy(instance, copy)
|
||||
|
||||
existing = inst.args.get(arg)
|
||||
if append and existing is not None:
|
||||
|
|
@ -5398,7 +5454,7 @@ def to_identifier(name, quoted=None, copy=True):
|
|||
return None
|
||||
|
||||
if isinstance(name, Identifier):
|
||||
identifier = _maybe_copy(name, copy)
|
||||
identifier = maybe_copy(name, copy)
|
||||
elif isinstance(name, str):
|
||||
identifier = Identifier(
|
||||
this=name,
|
||||
|
|
@ -5735,7 +5791,7 @@ def convert(value: t.Any, copy: bool = False) -> Expression:
|
|||
Expression: the equivalent expression object.
|
||||
"""
|
||||
if isinstance(value, Expression):
|
||||
return _maybe_copy(value, copy)
|
||||
return maybe_copy(value, copy)
|
||||
if isinstance(value, str):
|
||||
return Literal.string(value)
|
||||
if isinstance(value, bool):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue