sqlglot.dialects.tableau
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 }
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 }
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
@classmethod
def
can_identify(text: str, identify: str | bool = 'safe') -> bool:
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
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