sqlglot.dialects.sqlite
1from __future__ import annotations 2 3import typing as t 4 5from sqlglot import exp, generator, parser, tokens, transforms 6from sqlglot.dialects.dialect import ( 7 Dialect, 8 arrow_json_extract_scalar_sql, 9 arrow_json_extract_sql, 10 concat_to_dpipe_sql, 11 count_if_to_sum, 12 no_ilike_sql, 13 no_pivot_sql, 14 no_tablesample_sql, 15 no_trycast_sql, 16 rename_func, 17) 18from sqlglot.tokens import TokenType 19 20 21def _date_add_sql(self: generator.Generator, expression: exp.DateAdd) -> str: 22 modifier = expression.expression 23 modifier = modifier.name if modifier.is_string else self.sql(modifier) 24 unit = expression.args.get("unit") 25 modifier = f"'{modifier} {unit.name}'" if unit else f"'{modifier}'" 26 return self.func("DATE", expression.this, modifier) 27 28 29def _transform_create(expression: exp.Expression) -> exp.Expression: 30 """Move primary key to a column and enforce auto_increment on primary keys.""" 31 schema = expression.this 32 33 if isinstance(expression, exp.Create) and isinstance(schema, exp.Schema): 34 defs = {} 35 primary_key = None 36 37 for e in schema.expressions: 38 if isinstance(e, exp.ColumnDef): 39 defs[e.name] = e 40 elif isinstance(e, exp.PrimaryKey): 41 primary_key = e 42 43 if primary_key and len(primary_key.expressions) == 1: 44 column = defs[primary_key.expressions[0].name] 45 column.append( 46 "constraints", exp.ColumnConstraint(kind=exp.PrimaryKeyColumnConstraint()) 47 ) 48 schema.expressions.remove(primary_key) 49 else: 50 for column in defs.values(): 51 auto_increment = None 52 for constraint in column.constraints.copy(): 53 if isinstance(constraint.kind, exp.PrimaryKeyColumnConstraint): 54 break 55 if isinstance(constraint.kind, exp.AutoIncrementColumnConstraint): 56 auto_increment = constraint 57 if auto_increment: 58 column.constraints.remove(auto_increment) 59 60 return expression 61 62 63class SQLite(Dialect): 64 # https://sqlite.org/forum/forumpost/5e575586ac5c711b?raw 65 RESOLVES_IDENTIFIERS_AS_UPPERCASE = None 66 67 class Tokenizer(tokens.Tokenizer): 68 IDENTIFIERS = ['"', ("[", "]"), "`"] 69 HEX_STRINGS = [("x'", "'"), ("X'", "'"), ("0x", ""), ("0X", "")] 70 71 class Parser(parser.Parser): 72 FUNCTIONS = { 73 **parser.Parser.FUNCTIONS, 74 "EDITDIST3": exp.Levenshtein.from_arg_list, 75 } 76 77 class Generator(generator.Generator): 78 JOIN_HINTS = False 79 TABLE_HINTS = False 80 QUERY_HINTS = False 81 82 TYPE_MAPPING = { 83 **generator.Generator.TYPE_MAPPING, 84 exp.DataType.Type.BOOLEAN: "INTEGER", 85 exp.DataType.Type.TINYINT: "INTEGER", 86 exp.DataType.Type.SMALLINT: "INTEGER", 87 exp.DataType.Type.INT: "INTEGER", 88 exp.DataType.Type.BIGINT: "INTEGER", 89 exp.DataType.Type.FLOAT: "REAL", 90 exp.DataType.Type.DOUBLE: "REAL", 91 exp.DataType.Type.DECIMAL: "REAL", 92 exp.DataType.Type.CHAR: "TEXT", 93 exp.DataType.Type.NCHAR: "TEXT", 94 exp.DataType.Type.VARCHAR: "TEXT", 95 exp.DataType.Type.NVARCHAR: "TEXT", 96 exp.DataType.Type.BINARY: "BLOB", 97 exp.DataType.Type.VARBINARY: "BLOB", 98 } 99 100 TOKEN_MAPPING = { 101 TokenType.AUTO_INCREMENT: "AUTOINCREMENT", 102 } 103 104 TRANSFORMS = { 105 **generator.Generator.TRANSFORMS, 106 exp.Concat: concat_to_dpipe_sql, 107 exp.CountIf: count_if_to_sum, 108 exp.Create: transforms.preprocess([_transform_create]), 109 exp.CurrentDate: lambda *_: "CURRENT_DATE", 110 exp.CurrentTime: lambda *_: "CURRENT_TIME", 111 exp.CurrentTimestamp: lambda *_: "CURRENT_TIMESTAMP", 112 exp.DateAdd: _date_add_sql, 113 exp.DateStrToDate: lambda self, e: self.sql(e, "this"), 114 exp.ILike: no_ilike_sql, 115 exp.JSONExtract: arrow_json_extract_sql, 116 exp.JSONExtractScalar: arrow_json_extract_scalar_sql, 117 exp.JSONBExtract: arrow_json_extract_sql, 118 exp.JSONBExtractScalar: arrow_json_extract_scalar_sql, 119 exp.Levenshtein: rename_func("EDITDIST3"), 120 exp.LogicalOr: rename_func("MAX"), 121 exp.LogicalAnd: rename_func("MIN"), 122 exp.Pivot: no_pivot_sql, 123 exp.SafeConcat: concat_to_dpipe_sql, 124 exp.Select: transforms.preprocess( 125 [transforms.eliminate_distinct_on, transforms.eliminate_qualify] 126 ), 127 exp.TableSample: no_tablesample_sql, 128 exp.TimeStrToTime: lambda self, e: self.sql(e, "this"), 129 exp.TryCast: no_trycast_sql, 130 } 131 132 PROPERTIES_LOCATION = { 133 k: exp.Properties.Location.UNSUPPORTED 134 for k, v in generator.Generator.PROPERTIES_LOCATION.items() 135 } 136 137 LIMIT_FETCH = "LIMIT" 138 139 def cast_sql(self, expression: exp.Cast, safe_prefix: t.Optional[str] = None) -> str: 140 if expression.is_type("date"): 141 return self.func("DATE", expression.this) 142 143 return super().cast_sql(expression) 144 145 def datediff_sql(self, expression: exp.DateDiff) -> str: 146 unit = expression.args.get("unit") 147 unit = unit.name.upper() if unit else "DAY" 148 149 sql = f"(JULIANDAY({self.sql(expression, 'this')}) - JULIANDAY({self.sql(expression, 'expression')}))" 150 151 if unit == "MONTH": 152 sql = f"{sql} / 30.0" 153 elif unit == "YEAR": 154 sql = f"{sql} / 365.0" 155 elif unit == "HOUR": 156 sql = f"{sql} * 24.0" 157 elif unit == "MINUTE": 158 sql = f"{sql} * 1440.0" 159 elif unit == "SECOND": 160 sql = f"{sql} * 86400.0" 161 elif unit == "MILLISECOND": 162 sql = f"{sql} * 86400000.0" 163 elif unit == "MICROSECOND": 164 sql = f"{sql} * 86400000000.0" 165 elif unit == "NANOSECOND": 166 sql = f"{sql} * 8640000000000.0" 167 else: 168 self.unsupported("DATEDIFF unsupported for '{unit}'.") 169 170 return f"CAST({sql} AS INTEGER)" 171 172 # https://www.sqlite.org/lang_aggfunc.html#group_concat 173 def groupconcat_sql(self, expression: exp.GroupConcat) -> str: 174 this = expression.this 175 distinct = expression.find(exp.Distinct) 176 177 if distinct: 178 this = distinct.expressions[0] 179 distinct_sql = "DISTINCT " 180 else: 181 distinct_sql = "" 182 183 if isinstance(expression.this, exp.Order): 184 self.unsupported("SQLite GROUP_CONCAT doesn't support ORDER BY.") 185 if expression.this.this and not distinct: 186 this = expression.this.this 187 188 separator = expression.args.get("separator") 189 return f"GROUP_CONCAT({distinct_sql}{self.format_args(this, separator)})" 190 191 def least_sql(self, expression: exp.Least) -> str: 192 if len(expression.expressions) > 1: 193 return rename_func("MIN")(self, expression) 194 195 return self.sql(expression, "this") 196 197 def transaction_sql(self, expression: exp.Transaction) -> str: 198 this = expression.this 199 this = f" {this}" if this else "" 200 return f"BEGIN{this} TRANSACTION"
64class SQLite(Dialect): 65 # https://sqlite.org/forum/forumpost/5e575586ac5c711b?raw 66 RESOLVES_IDENTIFIERS_AS_UPPERCASE = None 67 68 class Tokenizer(tokens.Tokenizer): 69 IDENTIFIERS = ['"', ("[", "]"), "`"] 70 HEX_STRINGS = [("x'", "'"), ("X'", "'"), ("0x", ""), ("0X", "")] 71 72 class Parser(parser.Parser): 73 FUNCTIONS = { 74 **parser.Parser.FUNCTIONS, 75 "EDITDIST3": exp.Levenshtein.from_arg_list, 76 } 77 78 class Generator(generator.Generator): 79 JOIN_HINTS = False 80 TABLE_HINTS = False 81 QUERY_HINTS = False 82 83 TYPE_MAPPING = { 84 **generator.Generator.TYPE_MAPPING, 85 exp.DataType.Type.BOOLEAN: "INTEGER", 86 exp.DataType.Type.TINYINT: "INTEGER", 87 exp.DataType.Type.SMALLINT: "INTEGER", 88 exp.DataType.Type.INT: "INTEGER", 89 exp.DataType.Type.BIGINT: "INTEGER", 90 exp.DataType.Type.FLOAT: "REAL", 91 exp.DataType.Type.DOUBLE: "REAL", 92 exp.DataType.Type.DECIMAL: "REAL", 93 exp.DataType.Type.CHAR: "TEXT", 94 exp.DataType.Type.NCHAR: "TEXT", 95 exp.DataType.Type.VARCHAR: "TEXT", 96 exp.DataType.Type.NVARCHAR: "TEXT", 97 exp.DataType.Type.BINARY: "BLOB", 98 exp.DataType.Type.VARBINARY: "BLOB", 99 } 100 101 TOKEN_MAPPING = { 102 TokenType.AUTO_INCREMENT: "AUTOINCREMENT", 103 } 104 105 TRANSFORMS = { 106 **generator.Generator.TRANSFORMS, 107 exp.Concat: concat_to_dpipe_sql, 108 exp.CountIf: count_if_to_sum, 109 exp.Create: transforms.preprocess([_transform_create]), 110 exp.CurrentDate: lambda *_: "CURRENT_DATE", 111 exp.CurrentTime: lambda *_: "CURRENT_TIME", 112 exp.CurrentTimestamp: lambda *_: "CURRENT_TIMESTAMP", 113 exp.DateAdd: _date_add_sql, 114 exp.DateStrToDate: lambda self, e: self.sql(e, "this"), 115 exp.ILike: no_ilike_sql, 116 exp.JSONExtract: arrow_json_extract_sql, 117 exp.JSONExtractScalar: arrow_json_extract_scalar_sql, 118 exp.JSONBExtract: arrow_json_extract_sql, 119 exp.JSONBExtractScalar: arrow_json_extract_scalar_sql, 120 exp.Levenshtein: rename_func("EDITDIST3"), 121 exp.LogicalOr: rename_func("MAX"), 122 exp.LogicalAnd: rename_func("MIN"), 123 exp.Pivot: no_pivot_sql, 124 exp.SafeConcat: concat_to_dpipe_sql, 125 exp.Select: transforms.preprocess( 126 [transforms.eliminate_distinct_on, transforms.eliminate_qualify] 127 ), 128 exp.TableSample: no_tablesample_sql, 129 exp.TimeStrToTime: lambda self, e: self.sql(e, "this"), 130 exp.TryCast: no_trycast_sql, 131 } 132 133 PROPERTIES_LOCATION = { 134 k: exp.Properties.Location.UNSUPPORTED 135 for k, v in generator.Generator.PROPERTIES_LOCATION.items() 136 } 137 138 LIMIT_FETCH = "LIMIT" 139 140 def cast_sql(self, expression: exp.Cast, safe_prefix: t.Optional[str] = None) -> str: 141 if expression.is_type("date"): 142 return self.func("DATE", expression.this) 143 144 return super().cast_sql(expression) 145 146 def datediff_sql(self, expression: exp.DateDiff) -> str: 147 unit = expression.args.get("unit") 148 unit = unit.name.upper() if unit else "DAY" 149 150 sql = f"(JULIANDAY({self.sql(expression, 'this')}) - JULIANDAY({self.sql(expression, 'expression')}))" 151 152 if unit == "MONTH": 153 sql = f"{sql} / 30.0" 154 elif unit == "YEAR": 155 sql = f"{sql} / 365.0" 156 elif unit == "HOUR": 157 sql = f"{sql} * 24.0" 158 elif unit == "MINUTE": 159 sql = f"{sql} * 1440.0" 160 elif unit == "SECOND": 161 sql = f"{sql} * 86400.0" 162 elif unit == "MILLISECOND": 163 sql = f"{sql} * 86400000.0" 164 elif unit == "MICROSECOND": 165 sql = f"{sql} * 86400000000.0" 166 elif unit == "NANOSECOND": 167 sql = f"{sql} * 8640000000000.0" 168 else: 169 self.unsupported("DATEDIFF unsupported for '{unit}'.") 170 171 return f"CAST({sql} AS INTEGER)" 172 173 # https://www.sqlite.org/lang_aggfunc.html#group_concat 174 def groupconcat_sql(self, expression: exp.GroupConcat) -> str: 175 this = expression.this 176 distinct = expression.find(exp.Distinct) 177 178 if distinct: 179 this = distinct.expressions[0] 180 distinct_sql = "DISTINCT " 181 else: 182 distinct_sql = "" 183 184 if isinstance(expression.this, exp.Order): 185 self.unsupported("SQLite GROUP_CONCAT doesn't support ORDER BY.") 186 if expression.this.this and not distinct: 187 this = expression.this.this 188 189 separator = expression.args.get("separator") 190 return f"GROUP_CONCAT({distinct_sql}{self.format_args(this, separator)})" 191 192 def least_sql(self, expression: exp.Least) -> str: 193 if len(expression.expressions) > 1: 194 return rename_func("MIN")(self, expression) 195 196 return self.sql(expression, "this") 197 198 def transaction_sql(self, expression: exp.Transaction) -> str: 199 this = expression.this 200 this = f" {this}" if this else "" 201 return f"BEGIN{this} TRANSACTION"
tokenizer_class =
<class 'sqlglot.dialects.sqlite.SQLite.Tokenizer'>
parser_class =
<class 'sqlglot.dialects.sqlite.SQLite.Parser'>
generator_class =
<class 'sqlglot.dialects.sqlite.SQLite.Generator'>
Inherited Members
- sqlglot.dialects.dialect.Dialect
- INDEX_OFFSET
- UNNEST_COLUMN_ONLY
- ALIAS_POST_TABLESAMPLE
- IDENTIFIERS_CAN_START_WITH_DIGIT
- STRICT_STRING_CONCAT
- NORMALIZE_FUNCTIONS
- NULL_ORDERING
- DATE_FORMAT
- DATEINT_FORMAT
- TIME_FORMAT
- TIME_MAPPING
- FORMAT_MAPPING
- PSEUDOCOLUMNS
- get_or_raise
- format_time
- normalize_identifier
- case_sensitive
- can_identify
- quote_identifier
- parse
- parse_into
- generate
- transpile
- tokenize
- tokenizer
- parser
- generator
68 class Tokenizer(tokens.Tokenizer): 69 IDENTIFIERS = ['"', ("[", "]"), "`"] 70 HEX_STRINGS = [("x'", "'"), ("X'", "'"), ("0x", ""), ("0X", "")]
Inherited Members
72 class Parser(parser.Parser): 73 FUNCTIONS = { 74 **parser.Parser.FUNCTIONS, 75 "EDITDIST3": exp.Levenshtein.from_arg_list, 76 }
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
FUNCTIONS =
{'ABS': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Abs'>>, 'ANY_VALUE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.AnyValue'>>, 'APPROX_DISTINCT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ApproxDistinct'>>, 'APPROX_COUNT_DISTINCT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ApproxDistinct'>>, 'APPROX_QUANTILE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ApproxQuantile'>>, 'ARRAY': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Array'>>, 'ARRAY_AGG': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArrayAgg'>>, 'ARRAY_ALL': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArrayAll'>>, 'ARRAY_ANY': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArrayAny'>>, 'ARRAY_CONCAT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArrayConcat'>>, 'ARRAY_CONTAINS': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArrayContains'>>, 'FILTER': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArrayFilter'>>, 'ARRAY_FILTER': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArrayFilter'>>, 'ARRAY_JOIN': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArrayJoin'>>, 'ARRAY_SIZE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArraySize'>>, 'ARRAY_SORT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArraySort'>>, 'ARRAY_SUM': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArraySum'>>, 'ARRAY_UNION_AGG': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ArrayUnionAgg'>>, 'AVG': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Avg'>>, 'CASE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Case'>>, 'CAST': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Cast'>>, 'CAST_TO_STR_TYPE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.CastToStrType'>>, 'CEIL': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Ceil'>>, 'CEILING': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Ceil'>>, 'COALESCE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Coalesce'>>, 'IFNULL': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Coalesce'>>, 'NVL': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Coalesce'>>, 'CONCAT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Concat'>>, 'CONCAT_WS': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ConcatWs'>>, 'COUNT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Count'>>, 'COUNT_IF': <bound method Func.from_arg_list of <class 'sqlglot.expressions.CountIf'>>, 'CURRENT_DATE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.CurrentDate'>>, 'CURRENT_DATETIME': <bound method Func.from_arg_list of <class 'sqlglot.expressions.CurrentDatetime'>>, 'CURRENT_TIME': <bound method Func.from_arg_list of <class 'sqlglot.expressions.CurrentTime'>>, 'CURRENT_TIMESTAMP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.CurrentTimestamp'>>, 'CURRENT_USER': <bound method Func.from_arg_list of <class 'sqlglot.expressions.CurrentUser'>>, 'DATE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Date'>>, 'DATE_ADD': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DateAdd'>>, 'DATEDIFF': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DateDiff'>>, 'DATE_DIFF': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DateDiff'>>, 'DATEFROMPARTS': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DateFromParts'>>, 'DATE_STR_TO_DATE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DateStrToDate'>>, 'DATE_SUB': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DateSub'>>, 'DATE_TO_DATE_STR': <function Parser.<lambda>>, 'DATE_TO_DI': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DateToDi'>>, 'DATE_TRUNC': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DateTrunc'>>, 'DATETIME_ADD': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DatetimeAdd'>>, 'DATETIME_DIFF': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DatetimeDiff'>>, 'DATETIME_SUB': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DatetimeSub'>>, 'DATETIME_TRUNC': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DatetimeTrunc'>>, 'DAY': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Day'>>, 'DAY_OF_MONTH': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DayOfMonth'>>, 'DAYOFMONTH': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DayOfMonth'>>, 'DAY_OF_WEEK': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DayOfWeek'>>, 'DAYOFWEEK': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DayOfWeek'>>, 'DAY_OF_YEAR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DayOfYear'>>, 'DAYOFYEAR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DayOfYear'>>, 'DECODE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Decode'>>, 'DI_TO_DATE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.DiToDate'>>, 'ENCODE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Encode'>>, 'EXP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Exp'>>, 'EXPLODE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Explode'>>, 'EXTRACT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Extract'>>, 'FLOOR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Floor'>>, 'FROM_BASE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.FromBase'>>, 'FROM_BASE64': <bound method Func.from_arg_list of <class 'sqlglot.expressions.FromBase64'>>, 'GENERATE_SERIES': <bound method Func.from_arg_list of <class 'sqlglot.expressions.GenerateSeries'>>, 'GREATEST': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Greatest'>>, 'GROUP_CONCAT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.GroupConcat'>>, 'HEX': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Hex'>>, 'HLL': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Hll'>>, 'IF': <bound method Func.from_arg_list of <class 'sqlglot.expressions.If'>>, 'INITCAP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Initcap'>>, 'IS_NAN': <bound method Func.from_arg_list of <class 'sqlglot.expressions.IsNan'>>, 'ISNAN': <bound method Func.from_arg_list of <class 'sqlglot.expressions.IsNan'>>, 'JSON_ARRAY_CONTAINS': <bound method Func.from_arg_list of <class 'sqlglot.expressions.JSONArrayContains'>>, 'JSONB_EXTRACT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.JSONBExtract'>>, 'JSONB_EXTRACT_SCALAR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.JSONBExtractScalar'>>, 'JSON_EXTRACT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.JSONExtract'>>, 'JSON_EXTRACT_SCALAR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.JSONExtractScalar'>>, 'JSON_FORMAT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.JSONFormat'>>, 'J_S_O_N_OBJECT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.JSONObject'>>, 'LAST_DATE_OF_MONTH': <bound method Func.from_arg_list of <class 'sqlglot.expressions.LastDateOfMonth'>>, 'LEAST': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Least'>>, 'LEFT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Left'>>, 'LENGTH': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Length'>>, 'LEN': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Length'>>, 'LEVENSHTEIN': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Levenshtein'>>, 'LN': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Ln'>>, 'LOG': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Log'>>, 'LOG10': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Log10'>>, 'LOG2': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Log2'>>, 'LOGICAL_AND': <bound method Func.from_arg_list of <class 'sqlglot.expressions.LogicalAnd'>>, 'BOOL_AND': <bound method Func.from_arg_list of <class 'sqlglot.expressions.LogicalAnd'>>, 'BOOLAND_AGG': <bound method Func.from_arg_list of <class 'sqlglot.expressions.LogicalAnd'>>, 'LOGICAL_OR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.LogicalOr'>>, 'BOOL_OR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.LogicalOr'>>, 'BOOLOR_AGG': <bound method Func.from_arg_list of <class 'sqlglot.expressions.LogicalOr'>>, 'LOWER': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Lower'>>, 'LCASE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Lower'>>, 'MD5': <bound method Func.from_arg_list of <class 'sqlglot.expressions.MD5'>>, 'MD5_DIGEST': <bound method Func.from_arg_list of <class 'sqlglot.expressions.MD5Digest'>>, 'MAP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Map'>>, 'MAP_FROM_ENTRIES': <bound method Func.from_arg_list of <class 'sqlglot.expressions.MapFromEntries'>>, 'MATCH_AGAINST': <bound method Func.from_arg_list of <class 'sqlglot.expressions.MatchAgainst'>>, 'MAX': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Max'>>, 'MIN': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Min'>>, 'MONTH': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Month'>>, 'MONTHS_BETWEEN': <bound method Func.from_arg_list of <class 'sqlglot.expressions.MonthsBetween'>>, 'NEXT_VALUE_FOR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.NextValueFor'>>, 'NUMBER_TO_STR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.NumberToStr'>>, 'NVL2': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Nvl2'>>, 'OPEN_J_S_O_N': <bound method Func.from_arg_list of <class 'sqlglot.expressions.OpenJSON'>>, 'PARAMETERIZED_AGG': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ParameterizedAgg'>>, 'PERCENTILE_CONT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.PercentileCont'>>, 'PERCENTILE_DISC': <bound method Func.from_arg_list of <class 'sqlglot.expressions.PercentileDisc'>>, 'POSEXPLODE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Posexplode'>>, 'POWER': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Pow'>>, 'POW': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Pow'>>, 'QUANTILE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Quantile'>>, 'RANGE_N': <bound method Func.from_arg_list of <class 'sqlglot.expressions.RangeN'>>, 'READ_CSV': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ReadCSV'>>, 'REDUCE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Reduce'>>, 'REGEXP_EXTRACT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.RegexpExtract'>>, 'REGEXP_I_LIKE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.RegexpILike'>>, 'REGEXP_LIKE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.RegexpLike'>>, 'REGEXP_REPLACE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.RegexpReplace'>>, 'REGEXP_SPLIT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.RegexpSplit'>>, 'REPEAT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Repeat'>>, 'RIGHT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Right'>>, 'ROUND': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Round'>>, 'ROW_NUMBER': <bound method Func.from_arg_list of <class 'sqlglot.expressions.RowNumber'>>, 'SHA': <bound method Func.from_arg_list of <class 'sqlglot.expressions.SHA'>>, 'SHA1': <bound method Func.from_arg_list of <class 'sqlglot.expressions.SHA'>>, 'SHA2': <bound method Func.from_arg_list of <class 'sqlglot.expressions.SHA2'>>, 'SAFE_CONCAT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.SafeConcat'>>, 'SAFE_DIVIDE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.SafeDivide'>>, 'SET_AGG': <bound method Func.from_arg_list of <class 'sqlglot.expressions.SetAgg'>>, 'SORT_ARRAY': <bound method Func.from_arg_list of <class 'sqlglot.expressions.SortArray'>>, 'SPLIT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Split'>>, 'SQRT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Sqrt'>>, 'STANDARD_HASH': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StandardHash'>>, 'STAR_MAP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StarMap'>>, 'STARTS_WITH': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StartsWith'>>, 'STARTSWITH': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StartsWith'>>, 'STDDEV': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Stddev'>>, 'STDDEV_POP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StddevPop'>>, 'STDDEV_SAMP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StddevSamp'>>, 'STR_POSITION': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StrPosition'>>, 'STR_TO_DATE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StrToDate'>>, 'STR_TO_TIME': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StrToTime'>>, 'STR_TO_UNIX': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StrToUnix'>>, 'STRUCT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Struct'>>, 'STRUCT_EXTRACT': <bound method Func.from_arg_list of <class 'sqlglot.expressions.StructExtract'>>, 'SUBSTRING': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Substring'>>, 'SUM': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Sum'>>, 'TIME_ADD': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimeAdd'>>, 'TIME_DIFF': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimeDiff'>>, 'TIME_STR_TO_DATE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimeStrToDate'>>, 'TIME_STR_TO_TIME': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimeStrToTime'>>, 'TIME_STR_TO_UNIX': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimeStrToUnix'>>, 'TIME_SUB': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimeSub'>>, 'TIME_TO_STR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimeToStr'>>, 'TIME_TO_TIME_STR': <function Parser.<lambda>>, 'TIME_TO_UNIX': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimeToUnix'>>, 'TIME_TRUNC': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimeTrunc'>>, 'TIMESTAMP_ADD': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimestampAdd'>>, 'TIMESTAMP_DIFF': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimestampDiff'>>, 'TIMESTAMP_SUB': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimestampSub'>>, 'TIMESTAMP_TRUNC': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TimestampTrunc'>>, 'TO_BASE64': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ToBase64'>>, 'TO_CHAR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.ToChar'>>, 'TRANSFORM': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Transform'>>, 'TRIM': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Trim'>>, 'TRY_CAST': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TryCast'>>, 'TS_OR_DI_TO_DI': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TsOrDiToDi'>>, 'TS_OR_DS_ADD': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TsOrDsAdd'>>, 'TS_OR_DS_TO_DATE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.TsOrDsToDate'>>, 'TS_OR_DS_TO_DATE_STR': <function Parser.<lambda>>, 'UNHEX': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Unhex'>>, 'UNIX_TO_STR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.UnixToStr'>>, 'UNIX_TO_TIME': <bound method Func.from_arg_list of <class 'sqlglot.expressions.UnixToTime'>>, 'UNIX_TO_TIME_STR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.UnixToTimeStr'>>, 'UPPER': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Upper'>>, 'UCASE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Upper'>>, 'VAR_MAP': <function parse_var_map>, 'VARIANCE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Variance'>>, 'VARIANCE_SAMP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Variance'>>, 'VAR_SAMP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Variance'>>, 'VARIANCE_POP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.VariancePop'>>, 'VAR_POP': <bound method Func.from_arg_list of <class 'sqlglot.expressions.VariancePop'>>, 'WEEK': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Week'>>, 'WEEK_OF_YEAR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.WeekOfYear'>>, 'WEEKOFYEAR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.WeekOfYear'>>, 'WHEN': <bound method Func.from_arg_list of <class 'sqlglot.expressions.When'>>, 'X_M_L_TABLE': <bound method Func.from_arg_list of <class 'sqlglot.expressions.XMLTable'>>, 'XOR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Xor'>>, 'YEAR': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Year'>>, 'GLOB': <function Parser.<lambda>>, 'LIKE': <function parse_like>, 'EDITDIST3': <bound method Func.from_arg_list of <class 'sqlglot.expressions.Levenshtein'>>}
SET_TRIE: Dict =
{'GLOBAL': {0: True}, 'LOCAL': {0: True}, 'SESSION': {0: True}, 'TRANSACTION': {0: True}}
Inherited Members
- sqlglot.parser.Parser
- Parser
- NO_PAREN_FUNCTIONS
- NESTED_TYPE_TOKENS
- ENUM_TYPE_TOKENS
- TYPE_TOKENS
- SUBQUERY_PREDICATES
- RESERVED_KEYWORDS
- DB_CREATABLES
- CREATABLES
- ID_VAR_TOKENS
- INTERVAL_VARS
- TABLE_ALIAS_TOKENS
- COMMENT_TABLE_ALIAS_TOKENS
- UPDATE_ALIAS_TOKENS
- TRIM_TYPES
- FUNC_TOKENS
- CONJUNCTION
- EQUALITY
- COMPARISON
- BITWISE
- TERM
- FACTOR
- TIMESTAMPS
- SET_OPERATIONS
- JOIN_METHODS
- JOIN_SIDES
- JOIN_KINDS
- JOIN_HINTS
- LAMBDAS
- COLUMN_OPERATORS
- EXPRESSION_PARSERS
- STATEMENT_PARSERS
- UNARY_PARSERS
- PRIMARY_PARSERS
- PLACEHOLDER_PARSERS
- RANGE_PARSERS
- PROPERTY_PARSERS
- CONSTRAINT_PARSERS
- ALTER_PARSERS
- SCHEMA_UNNAMED_CONSTRAINTS
- NO_PAREN_FUNCTION_PARSERS
- FUNCTIONS_WITH_ALIASED_ARGS
- FUNCTION_PARSERS
- QUERY_MODIFIER_PARSERS
- SET_PARSERS
- SHOW_PARSERS
- TYPE_LITERAL_PARSERS
- MODIFIABLES
- DDL_SELECT_TOKENS
- PRE_VOLATILE_TOKENS
- TRANSACTION_KIND
- TRANSACTION_CHARACTERISTICS
- INSERT_ALTERNATIVES
- CLONE_KINDS
- TABLE_INDEX_HINT_TOKENS
- WINDOW_ALIAS_TOKENS
- WINDOW_BEFORE_PAREN_TOKENS
- WINDOW_SIDES
- ADD_CONSTRAINT_TOKENS
- STRICT_CAST
- CONCAT_NULL_OUTPUTS_STRING
- PREFIXED_PIVOT_COLUMNS
- IDENTIFY_PIVOT_STRINGS
- LOG_BASE_FIRST
- LOG_DEFAULTS_TO_LN
- INDEX_OFFSET
- UNNEST_COLUMN_ONLY
- ALIAS_POST_TABLESAMPLE
- STRICT_STRING_CONCAT
- NORMALIZE_FUNCTIONS
- NULL_ORDERING
- FORMAT_MAPPING
- TIME_MAPPING
- error_level
- error_message_context
- max_errors
- reset
- parse
- parse_into
- check_errors
- raise_error
- expression
- validate_expression
- errors
- sql
78 class Generator(generator.Generator): 79 JOIN_HINTS = False 80 TABLE_HINTS = False 81 QUERY_HINTS = False 82 83 TYPE_MAPPING = { 84 **generator.Generator.TYPE_MAPPING, 85 exp.DataType.Type.BOOLEAN: "INTEGER", 86 exp.DataType.Type.TINYINT: "INTEGER", 87 exp.DataType.Type.SMALLINT: "INTEGER", 88 exp.DataType.Type.INT: "INTEGER", 89 exp.DataType.Type.BIGINT: "INTEGER", 90 exp.DataType.Type.FLOAT: "REAL", 91 exp.DataType.Type.DOUBLE: "REAL", 92 exp.DataType.Type.DECIMAL: "REAL", 93 exp.DataType.Type.CHAR: "TEXT", 94 exp.DataType.Type.NCHAR: "TEXT", 95 exp.DataType.Type.VARCHAR: "TEXT", 96 exp.DataType.Type.NVARCHAR: "TEXT", 97 exp.DataType.Type.BINARY: "BLOB", 98 exp.DataType.Type.VARBINARY: "BLOB", 99 } 100 101 TOKEN_MAPPING = { 102 TokenType.AUTO_INCREMENT: "AUTOINCREMENT", 103 } 104 105 TRANSFORMS = { 106 **generator.Generator.TRANSFORMS, 107 exp.Concat: concat_to_dpipe_sql, 108 exp.CountIf: count_if_to_sum, 109 exp.Create: transforms.preprocess([_transform_create]), 110 exp.CurrentDate: lambda *_: "CURRENT_DATE", 111 exp.CurrentTime: lambda *_: "CURRENT_TIME", 112 exp.CurrentTimestamp: lambda *_: "CURRENT_TIMESTAMP", 113 exp.DateAdd: _date_add_sql, 114 exp.DateStrToDate: lambda self, e: self.sql(e, "this"), 115 exp.ILike: no_ilike_sql, 116 exp.JSONExtract: arrow_json_extract_sql, 117 exp.JSONExtractScalar: arrow_json_extract_scalar_sql, 118 exp.JSONBExtract: arrow_json_extract_sql, 119 exp.JSONBExtractScalar: arrow_json_extract_scalar_sql, 120 exp.Levenshtein: rename_func("EDITDIST3"), 121 exp.LogicalOr: rename_func("MAX"), 122 exp.LogicalAnd: rename_func("MIN"), 123 exp.Pivot: no_pivot_sql, 124 exp.SafeConcat: concat_to_dpipe_sql, 125 exp.Select: transforms.preprocess( 126 [transforms.eliminate_distinct_on, transforms.eliminate_qualify] 127 ), 128 exp.TableSample: no_tablesample_sql, 129 exp.TimeStrToTime: lambda self, e: self.sql(e, "this"), 130 exp.TryCast: no_trycast_sql, 131 } 132 133 PROPERTIES_LOCATION = { 134 k: exp.Properties.Location.UNSUPPORTED 135 for k, v in generator.Generator.PROPERTIES_LOCATION.items() 136 } 137 138 LIMIT_FETCH = "LIMIT" 139 140 def cast_sql(self, expression: exp.Cast, safe_prefix: t.Optional[str] = None) -> str: 141 if expression.is_type("date"): 142 return self.func("DATE", expression.this) 143 144 return super().cast_sql(expression) 145 146 def datediff_sql(self, expression: exp.DateDiff) -> str: 147 unit = expression.args.get("unit") 148 unit = unit.name.upper() if unit else "DAY" 149 150 sql = f"(JULIANDAY({self.sql(expression, 'this')}) - JULIANDAY({self.sql(expression, 'expression')}))" 151 152 if unit == "MONTH": 153 sql = f"{sql} / 30.0" 154 elif unit == "YEAR": 155 sql = f"{sql} / 365.0" 156 elif unit == "HOUR": 157 sql = f"{sql} * 24.0" 158 elif unit == "MINUTE": 159 sql = f"{sql} * 1440.0" 160 elif unit == "SECOND": 161 sql = f"{sql} * 86400.0" 162 elif unit == "MILLISECOND": 163 sql = f"{sql} * 86400000.0" 164 elif unit == "MICROSECOND": 165 sql = f"{sql} * 86400000000.0" 166 elif unit == "NANOSECOND": 167 sql = f"{sql} * 8640000000000.0" 168 else: 169 self.unsupported("DATEDIFF unsupported for '{unit}'.") 170 171 return f"CAST({sql} AS INTEGER)" 172 173 # https://www.sqlite.org/lang_aggfunc.html#group_concat 174 def groupconcat_sql(self, expression: exp.GroupConcat) -> str: 175 this = expression.this 176 distinct = expression.find(exp.Distinct) 177 178 if distinct: 179 this = distinct.expressions[0] 180 distinct_sql = "DISTINCT " 181 else: 182 distinct_sql = "" 183 184 if isinstance(expression.this, exp.Order): 185 self.unsupported("SQLite GROUP_CONCAT doesn't support ORDER BY.") 186 if expression.this.this and not distinct: 187 this = expression.this.this 188 189 separator = expression.args.get("separator") 190 return f"GROUP_CONCAT({distinct_sql}{self.format_args(this, separator)})" 191 192 def least_sql(self, expression: exp.Least) -> str: 193 if len(expression.expressions) > 1: 194 return rename_func("MIN")(self, expression) 195 196 return self.sql(expression, "this") 197 198 def transaction_sql(self, expression: exp.Transaction) -> str: 199 this = expression.this 200 this = f" {this}" if this else "" 201 return f"BEGIN{this} TRANSACTION"
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
TYPE_MAPPING =
{<Type.NCHAR: 'NCHAR'>: 'TEXT', <Type.NVARCHAR: 'NVARCHAR'>: 'TEXT', <Type.MEDIUMTEXT: 'MEDIUMTEXT'>: 'TEXT', <Type.LONGTEXT: 'LONGTEXT'>: 'TEXT', <Type.MEDIUMBLOB: 'MEDIUMBLOB'>: 'BLOB', <Type.LONGBLOB: 'LONGBLOB'>: 'BLOB', <Type.INET: 'INET'>: 'INET', <Type.BOOLEAN: 'BOOLEAN'>: 'INTEGER', <Type.TINYINT: 'TINYINT'>: 'INTEGER', <Type.SMALLINT: 'SMALLINT'>: 'INTEGER', <Type.INT: 'INT'>: 'INTEGER', <Type.BIGINT: 'BIGINT'>: 'INTEGER', <Type.FLOAT: 'FLOAT'>: 'REAL', <Type.DOUBLE: 'DOUBLE'>: 'REAL', <Type.DECIMAL: 'DECIMAL'>: 'REAL', <Type.CHAR: 'CHAR'>: 'TEXT', <Type.VARCHAR: 'VARCHAR'>: 'TEXT', <Type.BINARY: 'BINARY'>: 'BLOB', <Type.VARBINARY: 'VARBINARY'>: 'BLOB'}
TRANSFORMS =
{<class 'sqlglot.expressions.DateAdd'>: <function _date_add_sql>, <class 'sqlglot.expressions.TsOrDsAdd'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.CaseSpecificColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.CharacterSetColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.CharacterSetProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.CheckColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.CollateColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.CopyGrantsProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.CommentColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.DateFormatColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.DefaultColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.EncodeColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.ExecuteAsProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.ExternalProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.HeapProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.InlineLengthColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.LanguageProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.LocationProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.LogProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.MaterializedProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.NoPrimaryIndexProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.OnCommitProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.OnUpdateColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.PathColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.ReturnsProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.SetProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.SettingsProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.SqlSecurityProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.StabilityProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.TemporaryProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.ToTableProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.TransientProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.TitleColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.UppercaseColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.VarMap'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.VolatileProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.WithJournalTableProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.Concat'>: <function concat_to_dpipe_sql>, <class 'sqlglot.expressions.CountIf'>: <function count_if_to_sum>, <class 'sqlglot.expressions.Create'>: <function preprocess.<locals>._to_sql>, <class 'sqlglot.expressions.CurrentDate'>: <function SQLite.Generator.<lambda>>, <class 'sqlglot.expressions.CurrentTime'>: <function SQLite.Generator.<lambda>>, <class 'sqlglot.expressions.CurrentTimestamp'>: <function SQLite.Generator.<lambda>>, <class 'sqlglot.expressions.DateStrToDate'>: <function SQLite.Generator.<lambda>>, <class 'sqlglot.expressions.ILike'>: <function no_ilike_sql>, <class 'sqlglot.expressions.JSONExtract'>: <function arrow_json_extract_sql>, <class 'sqlglot.expressions.JSONExtractScalar'>: <function arrow_json_extract_scalar_sql>, <class 'sqlglot.expressions.JSONBExtract'>: <function arrow_json_extract_sql>, <class 'sqlglot.expressions.JSONBExtractScalar'>: <function arrow_json_extract_scalar_sql>, <class 'sqlglot.expressions.Levenshtein'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.LogicalOr'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.LogicalAnd'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.Pivot'>: <function no_pivot_sql>, <class 'sqlglot.expressions.SafeConcat'>: <function concat_to_dpipe_sql>, <class 'sqlglot.expressions.Select'>: <function preprocess.<locals>._to_sql>, <class 'sqlglot.expressions.TableSample'>: <function no_tablesample_sql>, <class 'sqlglot.expressions.TimeStrToTime'>: <function SQLite.Generator.<lambda>>, <class 'sqlglot.expressions.TryCast'>: <function no_trycast_sql>}
PROPERTIES_LOCATION =
{<class 'sqlglot.expressions.AlgorithmProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.AutoIncrementProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.BlockCompressionProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.CharacterSetProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.ChecksumProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.CollateProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.CopyGrantsProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.Cluster'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.ClusteredByProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.DataBlocksizeProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.DefinerProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.DictRange'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.DictProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.DistKeyProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.DistStyleProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.EngineProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.ExecuteAsProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.ExternalProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.FallbackProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.FileFormatProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.FreespaceProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.HeapProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.IsolatedLoadingProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.JournalProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.LanguageProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.LikeProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.LocationProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.LockingProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.LogProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.MaterializedProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.MergeBlockRatioProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.NoPrimaryIndexProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.OnCommitProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.Order'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.PartitionedByProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.PrimaryKey'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.Property'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.ReturnsProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.RowFormatProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.RowFormatDelimitedProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.RowFormatSerdeProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.SchemaCommentProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.SerdeProperties'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.Set'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.SettingsProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.SetProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.SortKeyProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.SqlSecurityProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.StabilityProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.TemporaryProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.ToTableProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.TransientProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.MergeTreeTTL'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.VolatileProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.WithDataProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.WithJournalTableProperty'>: <Location.UNSUPPORTED: 'UNSUPPORTED'>}
def
cast_sql( self, expression: sqlglot.expressions.Cast, safe_prefix: Optional[str] = None) -> str:
146 def datediff_sql(self, expression: exp.DateDiff) -> str: 147 unit = expression.args.get("unit") 148 unit = unit.name.upper() if unit else "DAY" 149 150 sql = f"(JULIANDAY({self.sql(expression, 'this')}) - JULIANDAY({self.sql(expression, 'expression')}))" 151 152 if unit == "MONTH": 153 sql = f"{sql} / 30.0" 154 elif unit == "YEAR": 155 sql = f"{sql} / 365.0" 156 elif unit == "HOUR": 157 sql = f"{sql} * 24.0" 158 elif unit == "MINUTE": 159 sql = f"{sql} * 1440.0" 160 elif unit == "SECOND": 161 sql = f"{sql} * 86400.0" 162 elif unit == "MILLISECOND": 163 sql = f"{sql} * 86400000.0" 164 elif unit == "MICROSECOND": 165 sql = f"{sql} * 86400000000.0" 166 elif unit == "NANOSECOND": 167 sql = f"{sql} * 8640000000000.0" 168 else: 169 self.unsupported("DATEDIFF unsupported for '{unit}'.") 170 171 return f"CAST({sql} AS INTEGER)"
174 def groupconcat_sql(self, expression: exp.GroupConcat) -> str: 175 this = expression.this 176 distinct = expression.find(exp.Distinct) 177 178 if distinct: 179 this = distinct.expressions[0] 180 distinct_sql = "DISTINCT " 181 else: 182 distinct_sql = "" 183 184 if isinstance(expression.this, exp.Order): 185 self.unsupported("SQLite GROUP_CONCAT doesn't support ORDER BY.") 186 if expression.this.this and not distinct: 187 this = expression.this.this 188 189 separator = expression.args.get("separator") 190 return f"GROUP_CONCAT({distinct_sql}{self.format_args(this, separator)})"
@classmethod
def
can_identify(text: str, identify: str | bool = 'safe') -> bool:
253 @classmethod 254 def can_identify(cls, text: str, identify: str | bool = "safe") -> bool: 255 """Checks if text can be identified given an identify option. 256 257 Args: 258 text: The text to check. 259 identify: 260 "always" or `True`: Always returns true. 261 "safe": True if the identifier is case-insensitive. 262 263 Returns: 264 Whether or not the given text can be identified. 265 """ 266 if identify is True or identify == "always": 267 return True 268 269 if identify == "safe": 270 return not cls.case_sensitive(text) 271 272 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
- NULL_ORDERING_SUPPORTED
- LOCKING_READS_SUPPORTED
- EXPLICIT_UNION
- WRAP_DERIVED_VALUES
- CREATE_FUNCTION_RETURN_AS
- MATCHED_BY_SOURCE
- SINGLE_STRING_INTERVAL
- INTERVAL_ALLOWS_PLURAL_FORM
- TABLESAMPLE_WITH_METHOD
- TABLESAMPLE_SIZE_IS_PERCENT
- RENAME_TABLE_WITH_DB
- GROUPINGS_SEP
- INDEX_ON
- QUERY_HINT_SEP
- IS_BOOL_ALLOWED
- DUPLICATE_KEY_UPDATE_WITH_SET
- LIMIT_IS_TOP
- RETURNING_END
- COLUMN_JOIN_MARKS_SUPPORTED
- EXTRACT_ALLOWS_QUOTES
- VALUES_AS_TABLE
- STAR_MAPPING
- TIME_PART_SINGULARS
- STRUCT_DELIMITER
- PARAMETER_TOKEN
- RESERVED_KEYWORDS
- WITH_SEPARATED_COMMENTS
- UNWRAPPED_INTERVAL_VALUES
- SENTINEL_LINE_BREAK
- INDEX_OFFSET
- UNNEST_COLUMN_ONLY
- ALIAS_POST_TABLESAMPLE
- IDENTIFIERS_CAN_START_WITH_DIGIT
- STRICT_STRING_CONCAT
- NORMALIZE_FUNCTIONS
- NULL_ORDERING
- ESCAPE_LINE_BREAK
- pretty
- identify
- normalize
- pad
- unsupported_level
- max_unsupported
- leading_comma
- max_text_width
- comments
- normalize_functions
- unsupported_messages
- 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
- withtablehint_sql
- indextablehint_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
- safebracket_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
- if_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
- xor_sql
- connector_sql
- bitwiseand_sql
- bitwiseleftshift_sql
- bitwisenot_sql
- bitwiseor_sql
- bitwiserightshift_sql
- bitwisexor_sql
- currentdate_sql
- collate_sql
- command_sql
- comment_sql
- mergetreettlaction_sql
- mergetreettl_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
- clusteredbyproperty_sql
- anyvalue_sql
- querytransform_sql
- indexconstraintoption_sql
- indexcolumnconstraint_sql