sqlglot.parsers.exasol

  1from __future__ import annotations
  2
  3
  4from sqlglot import exp, parser
  5from sqlglot.dialects.dialect import (
  6    binary_from_function,
  7    build_date_delta,
  8    build_formatted_time,
  9    build_timetostr_or_tochar,
 10    build_trunc,
 11)
 12from sqlglot.helper import seq_get
 13from sqlglot.tokens import TokenType
 14
 15DATE_UNITS = {"DAY", "WEEK", "MONTH", "YEAR", "HOUR", "MINUTE", "SECOND"}
 16
 17
 18# https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/zeroifnull.htm
 19def _build_zeroifnull(args: list) -> exp.If:
 20    cond = exp.Is(this=seq_get(args, 0), expression=exp.Null())
 21    return exp.If(this=cond, true=exp.Literal.number(0), false=seq_get(args, 0))
 22
 23
 24# https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/nullifzero.htm
 25def _build_nullifzero(args: list) -> exp.If:
 26    cond = exp.EQ(this=seq_get(args, 0), expression=exp.Literal.number(0))
 27    return exp.If(this=cond, true=exp.Null(), false=seq_get(args, 0))
 28
 29
 30class ExasolParser(parser.Parser):
 31    TABLE_ALIAS_TOKENS = parser.Parser.TABLE_ALIAS_TOKENS | {
 32        TokenType.ANTI,
 33        TokenType.SEMI,
 34    }
 35
 36    FUNCTIONS = {
 37        **parser.Parser.FUNCTIONS,
 38        **{f"ADD_{unit}S": build_date_delta(exp.DateAdd, default_unit=unit) for unit in DATE_UNITS},
 39        **{
 40            f"{unit}S_BETWEEN": build_date_delta(exp.DateDiff, default_unit=unit)
 41            for unit in DATE_UNITS
 42        },
 43        "APPROXIMATE_COUNT_DISTINCT": exp.ApproxDistinct.from_arg_list,
 44        "BIT_AND": binary_from_function(exp.BitwiseAnd),
 45        "BIT_OR": binary_from_function(exp.BitwiseOr),
 46        "BIT_XOR": binary_from_function(exp.BitwiseXor),
 47        "BIT_NOT": lambda args: exp.BitwiseNot(this=seq_get(args, 0)),
 48        "BIT_LSHIFT": binary_from_function(exp.BitwiseLeftShift),
 49        "BIT_RSHIFT": binary_from_function(exp.BitwiseRightShift),
 50        # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/convert_tz.htm
 51        "CONVERT_TZ": lambda args: exp.ConvertTimezone(
 52            source_tz=seq_get(args, 1),
 53            target_tz=seq_get(args, 2),
 54            timestamp=seq_get(args, 0),
 55            options=seq_get(args, 3),
 56        ),
 57        "CURDATE": exp.CurrentDate.from_arg_list,
 58        # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/date_trunc.htm#DATE_TRUNC
 59        "DATE_TRUNC": lambda args: exp.TimestampTrunc(this=seq_get(args, 1), unit=seq_get(args, 0)),
 60        "DIV": binary_from_function(exp.IntDiv),
 61        "EVERY": lambda args: exp.All(this=seq_get(args, 0)),
 62        "EDIT_DISTANCE": exp.Levenshtein.from_arg_list,
 63        "FROM_POSIX_TIME": exp.UnixToTime.from_arg_list,
 64        "HASH_SHA": exp.SHA.from_arg_list,
 65        "HASH_SHA1": exp.SHA.from_arg_list,
 66        "HASH_MD5": exp.MD5.from_arg_list,
 67        "HASHTYPE_MD5": exp.MD5Digest.from_arg_list,
 68        "HASH_SHA256": lambda args: exp.SHA2(this=seq_get(args, 0), length=exp.Literal.number(256)),
 69        "HASH_SHA512": lambda args: exp.SHA2(this=seq_get(args, 0), length=exp.Literal.number(512)),
 70        "NOW": exp.CurrentTimestamp.from_arg_list,
 71        "NULLIFZERO": _build_nullifzero,
 72        "REGEXP_LIKE": lambda args: exp.RegexpLike(
 73            this=seq_get(args, 0),
 74            expression=seq_get(args, 1),
 75            flag=seq_get(args, 2),
 76            full_match=True,
 77        ),
 78        "REGEXP_SUBSTR": exp.RegexpExtract.from_arg_list,
 79        "REGEXP_REPLACE": lambda args: exp.RegexpReplace(
 80            this=seq_get(args, 0),
 81            expression=seq_get(args, 1),
 82            replacement=seq_get(args, 2),
 83            position=seq_get(args, 3),
 84            occurrence=seq_get(args, 4),
 85        ),
 86        "TRUNC": build_trunc,
 87        "TRUNCATE": build_trunc,
 88        "TO_CHAR": build_timetostr_or_tochar,
 89        "TO_DATE": build_formatted_time(exp.TsOrDsToDate, "exasol"),
 90        "USER": exp.CurrentUser.from_arg_list,
 91        "VAR_POP": exp.VariancePop.from_arg_list,
 92        "ZEROIFNULL": _build_zeroifnull,
 93    }
 94    CONSTRAINT_PARSERS = {
 95        **parser.Parser.CONSTRAINT_PARSERS,
 96        "COMMENT": lambda self: self.expression(
 97            exp.CommentColumnConstraint(this=self._match(TokenType.IS) and self._parse_string())
 98        ),
 99    }
100
101    RANGE_PARSERS = {
102        **parser.Parser.RANGE_PARSERS,
103        TokenType.RLIKE: lambda self, this: self.expression(
104            exp.RegexpLike(this=this, expression=self._parse_bitwise(), full_match=True)
105        ),
106    }
107
108    FUNC_TOKENS = {
109        *parser.Parser.FUNC_TOKENS,
110        TokenType.SYSTIMESTAMP,
111    }
112
113    NO_PAREN_FUNCTIONS = {
114        **parser.Parser.NO_PAREN_FUNCTIONS,
115        TokenType.SYSTIMESTAMP: exp.Systimestamp,
116        TokenType.CURRENT_SCHEMA: exp.CurrentSchema,
117    }
118
119    FUNCTION_PARSERS = {
120        **parser.Parser.FUNCTION_PARSERS,
121        # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/listagg.htm
122        # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/group_concat.htm
123        **dict.fromkeys(("GROUP_CONCAT", "LISTAGG"), lambda self: self._parse_group_concat()),
124        # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/json_value.htm
125        "JSON_VALUE": lambda self: self._parse_json_value(),
126        # https://docs.exasol.com/db/latest/sql_references/functions/alphabeticallistfunctions/json_extract.htm
127        "JSON_EXTRACT": lambda self: self._parse_json_extract(),
128    }
129
130    def _parse_column(self) -> exp.Expr | None:
131        column = super()._parse_column()
132        if not isinstance(column, exp.Column):
133            return column
134        table_ident = column.args.get("table")
135        if (
136            isinstance(table_ident, exp.Identifier)
137            and table_ident.name.upper() == "LOCAL"
138            and not bool(table_ident.args.get("quoted"))
139        ):
140            column.set("table", None)
141        return column
142
143    def _parse_json_extract(self) -> exp.JSONExtract:
144        args = self._parse_expressions()
145        self._match_r_paren()
146
147        expression = exp.JSONExtract(expressions=args)
148
149        if self._match_texts("EMITS"):
150            expression.set("emits", self._parse_schema())
151
152        return expression
153
154    ODBC_DATETIME_LITERALS = {
155        "d": exp.Date,
156        "ts": exp.Timestamp,
157    }