Edit on GitHub

sqlglot.dialects.tableau

View Source

 1from __future__ import annotations
 2
 3from sqlglot import exp, generator, parser, transforms
 4from sqlglot.dialects.dialect import Dialect, rename_func
 5
 6
 7class Tableau(Dialect):
 8    class Generator(generator.Generator):
 9        JOIN_HINTS = False
10        TABLE_HINTS = False
11
12        TRANSFORMS = {
13            **generator.Generator.TRANSFORMS,
14            exp.Coalesce: rename_func("IFNULL"),
15            exp.Select: transforms.preprocess([transforms.eliminate_distinct_on]),
16        }
17
18        PROPERTIES_LOCATION = {
19            **generator.Generator.PROPERTIES_LOCATION,
20            exp.VolatileProperty: exp.Properties.Location.UNSUPPORTED,
21        }
22
23        def if_sql(self, expression: exp.If) -> str:
24            this = self.sql(expression, "this")
25            true = self.sql(expression, "true")
26            false = self.sql(expression, "false")
27            return f"IF {this} THEN {true} ELSE {false} END"
28
29        def count_sql(self, expression: exp.Count) -> str:
30            this = expression.this
31            if isinstance(this, exp.Distinct):
32                return f"COUNTD({self.expressions(this, flat=True)})"
33            return f"COUNT({self.sql(expression, 'this')})"
34
35    class Parser(parser.Parser):
36        FUNCTIONS = {
37            **parser.Parser.FUNCTIONS,
38            "COUNTD": lambda args: exp.Count(this=exp.Distinct(expressions=args)),
39        }

class Tableau(sqlglot.dialects.dialect.Dialect): View Source

 8class Tableau(Dialect):
 9    class Generator(generator.Generator):
10        JOIN_HINTS = False
11        TABLE_HINTS = False
12
13        TRANSFORMS = {
14            **generator.Generator.TRANSFORMS,
15            exp.Coalesce: rename_func("IFNULL"),
16            exp.Select: transforms.preprocess([transforms.eliminate_distinct_on]),
17        }
18
19        PROPERTIES_LOCATION = {
20            **generator.Generator.PROPERTIES_LOCATION,
21            exp.VolatileProperty: exp.Properties.Location.UNSUPPORTED,
22        }
23
24        def if_sql(self, expression: exp.If) -> str:
25            this = self.sql(expression, "this")
26            true = self.sql(expression, "true")
27            false = self.sql(expression, "false")
28            return f"IF {this} THEN {true} ELSE {false} END"
29
30        def count_sql(self, expression: exp.Count) -> str:
31            this = expression.this
32            if isinstance(this, exp.Distinct):
33                return f"COUNTD({self.expressions(this, flat=True)})"
34            return f"COUNT({self.sql(expression, 'this')})"
35
36    class Parser(parser.Parser):
37        FUNCTIONS = {
38            **parser.Parser.FUNCTIONS,
39            "COUNTD": lambda args: exp.Count(this=exp.Distinct(expressions=args)),
40        }

Inherited Members

sqlglot.dialects.dialect.Dialect: get_or_raise; format_time; normalize_identifier; case_sensitive; can_identify; quote_identifier; parse; parse_into; generate; transpile; tokenize; parser; generator

class Tableau.Generator(sqlglot.generator.Generator): View Source

 9    class Generator(generator.Generator):
10        JOIN_HINTS = False
11        TABLE_HINTS = False
12
13        TRANSFORMS = {
14            **generator.Generator.TRANSFORMS,
15            exp.Coalesce: rename_func("IFNULL"),
16            exp.Select: transforms.preprocess([transforms.eliminate_distinct_on]),
17        }
18
19        PROPERTIES_LOCATION = {
20            **generator.Generator.PROPERTIES_LOCATION,
21            exp.VolatileProperty: exp.Properties.Location.UNSUPPORTED,
22        }
23
24        def if_sql(self, expression: exp.If) -> str:
25            this = self.sql(expression, "this")
26            true = self.sql(expression, "true")
27            false = self.sql(expression, "false")
28            return f"IF {this} THEN {true} ELSE {false} END"
29
30        def count_sql(self, expression: exp.Count) -> str:
31            this = expression.this
32            if isinstance(this, exp.Distinct):
33                return f"COUNTD({self.expressions(this, flat=True)})"
34            return f"COUNT({self.sql(expression, 'this')})"

Generator converts a given syntax tree to the corresponding SQL string.

Arguments:

pretty: Whether or not to format the produced SQL string. Default: False.
identify: Determines when an identifier should be quoted. Possible values are: False (default): Never quote, except in cases where it's mandatory by the dialect. True or 'always': Always quote. 'safe': Only quote identifiers that are case insensitive.
normalize: Whether or not to normalize identifiers to lowercase. Default: False.
pad: Determines the pad size in a formatted string. Default: 2.
indent: Determines the indentation size in a formatted string. Default: 2.
normalize_functions: Whether or not to normalize all function names. Possible values are: "upper" or True (default): Convert names to uppercase. "lower": Convert names to lowercase. False: Disables function name normalization.
unsupported_level: Determines the generator's behavior when it encounters unsupported expressions. Default ErrorLevel.WARN.
max_unsupported: Maximum number of unsupported messages to include in a raised UnsupportedError. This is only relevant if unsupported_level is ErrorLevel.RAISE. Default: 3
leading_comma: Determines whether or not the comma is leading or trailing in select expressions. This is only relevant when generating in pretty mode. Default: False
max_text_width: The max number of characters in a segment before creating new lines in pretty mode. The default is on the smaller end because the length only represents a segment and not the true line length. Default: 80
comments: Whether or not to preserve comments in the output SQL code. Default: True

def if_sql(self, expression: sqlglot.expressions.If) -> str: View Source

24        def if_sql(self, expression: exp.If) -> str:
25            this = self.sql(expression, "this")
26            true = self.sql(expression, "true")
27            false = self.sql(expression, "false")
28            return f"IF {this} THEN {true} ELSE {false} END"

def count_sql(self, expression: sqlglot.expressions.Count) -> str: View Source

30        def count_sql(self, expression: exp.Count) -> str:
31            this = expression.this
32            if isinstance(this, exp.Distinct):
33                return f"COUNTD({self.expressions(this, flat=True)})"
34            return f"COUNT({self.sql(expression, 'this')})"

@classmethod

def can_identify(text: str, identify: str | bool = 'safe') -> bool: View Source

247    @classmethod
248    def can_identify(cls, text: str, identify: str | bool = "safe") -> bool:
249        """Checks if text can be identified given an identify option.
250
251        Args:
252            text: The text to check.
253            identify:
254                "always" or `True`: Always returns true.
255                "safe": True if the identifier is case-insensitive.
256
257        Returns:
258            Whether or not the given text can be identified.
259        """
260        if identify is True or identify == "always":
261            return True
262
263        if identify == "safe":
264            return not cls.case_sensitive(text)
265
266        return False

Checks if text can be identified given an identify option.

Arguments:

text: The text to check.
identify: "always" or True: Always returns true. "safe": True if the identifier is case-insensitive.

Returns:

Whether or not the given text can be identified.

Inherited Members

sqlglot.generator.Generator: Generator; generate; unsupported; sep; seg; pad_comment; maybe_comment; wrap; no_identify; normalize_func; indent; sql; uncache_sql; cache_sql; characterset_sql; column_sql; columnposition_sql; columndef_sql; columnconstraint_sql; autoincrementcolumnconstraint_sql; compresscolumnconstraint_sql; generatedasidentitycolumnconstraint_sql; notnullcolumnconstraint_sql; primarykeycolumnconstraint_sql; uniquecolumnconstraint_sql; createable_sql; create_sql; clone_sql; describe_sql; prepend_ctes; with_sql; cte_sql; tablealias_sql; bitstring_sql; hexstring_sql; bytestring_sql; rawstring_sql; datatypesize_sql; datatype_sql; directory_sql; delete_sql; drop_sql; except_sql; except_op; fetch_sql; filter_sql; hint_sql; index_sql; identifier_sql; inputoutputformat_sql; national_sql; partition_sql; properties_sql; root_properties; properties; with_properties; locate_properties; property_sql; likeproperty_sql; fallbackproperty_sql; journalproperty_sql; freespaceproperty_sql; checksumproperty_sql; mergeblockratioproperty_sql; datablocksizeproperty_sql; blockcompressionproperty_sql; isolatedloadingproperty_sql; lockingproperty_sql; withdataproperty_sql; insert_sql; intersect_sql; intersect_op; introducer_sql; pseudotype_sql; onconflict_sql; returning_sql; rowformatdelimitedproperty_sql; table_sql; tablesample_sql; pivot_sql; tuple_sql; update_sql; values_sql; var_sql; into_sql; from_sql; group_sql; having_sql; join_sql; lambda_sql; lateral_sql; limit_sql; offset_sql; setitem_sql; set_sql; pragma_sql; lock_sql; literal_sql; escape_str; loaddata_sql; null_sql; boolean_sql; order_sql; cluster_sql; distribute_sql; sort_sql; ordered_sql; matchrecognize_sql; query_modifiers; offset_limit_modifiers; after_having_modifiers; after_limit_modifiers; select_sql; schema_sql; schema_columns_sql; star_sql; parameter_sql; sessionparameter_sql; placeholder_sql; subquery_sql; qualify_sql; union_sql; union_op; unnest_sql; where_sql; window_sql; partition_by_sql; windowspec_sql; withingroup_sql; between_sql; bracket_sql; all_sql; any_sql; exists_sql; case_sql; constraint_sql; nextvaluefor_sql; extract_sql; trim_sql; safeconcat_sql; check_sql; foreignkey_sql; primarykey_sql; matchagainst_sql; jsonkeyvalue_sql; jsonobject_sql; openjsoncolumndef_sql; openjson_sql; in_sql; in_unnest_op; interval_sql; return_sql; reference_sql; anonymous_sql; paren_sql; neg_sql; not_sql; alias_sql; aliases_sql; attimezone_sql; add_sql; and_sql; connector_sql; bitwiseand_sql; bitwiseleftshift_sql; bitwisenot_sql; bitwiseor_sql; bitwiserightshift_sql; bitwisexor_sql; cast_sql; currentdate_sql; collate_sql; command_sql; comment_sql; mergetreettlaction_sql; mergetreettl_sql; transaction_sql; commit_sql; rollback_sql; altercolumn_sql; renametable_sql; altertable_sql; droppartition_sql; addconstraint_sql; distinct_sql; ignorenulls_sql; respectnulls_sql; intdiv_sql; dpipe_sql; safedpipe_sql; div_sql; overlaps_sql; distance_sql; dot_sql; eq_sql; escape_sql; glob_sql; gt_sql; gte_sql; ilike_sql; ilikeany_sql; is_sql; like_sql; likeany_sql; similarto_sql; lt_sql; lte_sql; mod_sql; mul_sql; neq_sql; nullsafeeq_sql; nullsafeneq_sql; or_sql; slice_sql; sub_sql; trycast_sql; use_sql; binary; function_fallback_sql; func; format_args; text_width; format_time; expressions; op_expressions; naked_property; set_operation; tag_sql; token_sql; userdefinedfunction_sql; joinhint_sql; kwarg_sql; when_sql; merge_sql; tochar_sql; dictproperty_sql; dictrange_sql; dictsubproperty_sql; oncluster_sql

class Tableau.Parser(sqlglot.parser.Parser): View Source

36    class Parser(parser.Parser):
37        FUNCTIONS = {
38            **parser.Parser.FUNCTIONS,
39            "COUNTD": lambda args: exp.Count(this=exp.Distinct(expressions=args)),
40        }

Parser consumes a list of tokens produced by the Tokenizer and produces a parsed syntax tree.

Arguments:

error_level: The desired error level. Default: ErrorLevel.IMMEDIATE
error_message_context: Determines the amount of context to capture from a query string when displaying the error message (in number of characters). Default: 100
max_errors: Maximum number of error messages to include in a raised ParseError. This is only relevant if error_level is ErrorLevel.RAISE. Default: 3

Inherited Members

sqlglot.parser.Parser: Parser; reset; parse; parse_into; check_errors; raise_error; expression; validate_expression