sqlglot.generators.tsql
1from __future__ import annotations 2 3from functools import reduce 4 5from sqlglot import exp, generator, transforms 6from sqlglot.dialects.dialect import ( 7 any_value_to_max_sql, 8 date_delta_sql, 9 datestrtodate_sql, 10 generatedasidentitycolumnconstraint_sql, 11 max_or_greatest, 12 min_or_least, 13 remove_ts_or_ds_to_date, 14 rename_func, 15 strposition_sql, 16 timestrtotime_sql, 17 trim_sql, 18) 19from sqlglot.helper import seq_get 20from sqlglot.optimizer.scope import find_in_scope 21from sqlglot.parsers.tsql import OPTIONS_THAT_REQUIRE_EQUAL 22from sqlglot.time import format_time 23from collections import defaultdict 24 25DATE_PART_UNMAPPING = { 26 "WEEKISO": "ISO_WEEK", 27 "DAYOFWEEK": "WEEKDAY", 28 "TIMEZONE_MINUTE": "TZOFFSET", 29} 30 31BIT_TYPES = {exp.EQ, exp.NEQ, exp.Is, exp.In, exp.Select, exp.Alias} 32 33 34def _format_sql(self: TSQLGenerator, expression: exp.NumberToStr | exp.TimeToStr) -> str: 35 fmt = expression.args["format"] 36 37 if not isinstance(expression, exp.NumberToStr): 38 if fmt.is_string: 39 from sqlglot.dialects.tsql import TSQL 40 41 mapped_fmt = format_time(fmt.name, TSQL.INVERSE_TIME_MAPPING) 42 fmt_sql = self.sql(exp.Literal.string(mapped_fmt)) 43 else: 44 fmt_sql = self.format_time(expression) or self.sql(fmt) 45 else: 46 fmt_sql = self.sql(fmt) 47 48 return self.func("FORMAT", expression.this, fmt_sql, expression.args.get("culture")) 49 50 51def _string_agg_sql(self: TSQLGenerator, expression: exp.GroupConcat) -> str: 52 this = expression.this 53 distinct = find_in_scope(expression, exp.Distinct) 54 if distinct: 55 # exp.Distinct can appear below an exp.Order or an exp.GroupConcat expression 56 self.unsupported("T-SQL STRING_AGG doesn't support DISTINCT.") 57 this = distinct.pop().expressions[0] 58 59 order = "" 60 if isinstance(expression.this, exp.Order): 61 if expression.this.this: 62 this = expression.this.this.pop() 63 # Order has a leading space 64 order = f" WITHIN GROUP ({self.sql(expression.this)[1:]})" 65 66 separator = expression.args.get("separator") or exp.Literal.string(",") 67 return f"STRING_AGG({self.format_args(this, separator)}){order}" 68 69 70def qualify_derived_table_outputs(expression: exp.Expr) -> exp.Expr: 71 """Ensures all (unnamed) output columns are aliased for CTEs and Subqueries.""" 72 alias = expression.args.get("alias") 73 74 if ( 75 isinstance(expression, (exp.CTE, exp.Subquery)) 76 and isinstance(alias, exp.TableAlias) 77 and not alias.columns 78 ): 79 from sqlglot.dialects.tsql import TSQL 80 from sqlglot.optimizer.qualify_columns import qualify_outputs 81 82 # We keep track of the unaliased column projection indexes instead of the expressions 83 # themselves, because the latter are going to be replaced by new nodes when the aliases 84 # are added and hence we won't be able to reach these newly added Alias parents 85 query = expression.this 86 unaliased_column_indexes = ( 87 i for i, c in enumerate(query.selects) if isinstance(c, exp.Column) and not c.alias 88 ) 89 90 qualify_outputs(query, dialect=TSQL()) 91 92 # Preserve the quoting information of columns for newly added Alias nodes 93 query_selects = query.selects 94 for select_index in unaliased_column_indexes: 95 alias = query_selects[select_index] 96 column = alias.this 97 if isinstance(column.this, exp.Identifier): 98 alias.args["alias"].set("quoted", column.this.quoted) 99 100 return expression 101 102 103def _json_extract_sql( 104 self: TSQLGenerator, expression: exp.JSONExtract | exp.JSONExtractScalar 105) -> str: 106 # JSON_QUERY returns objects and arrays, JSON_VALUE returns scalars. A scalar-only 107 # extraction maps to JSON_VALUE; a source that also returns non-scalar values as 108 # text (e.g. SQLite's ->>) needs to try both, like a generic JSONExtract does 109 if isinstance(expression, exp.JSONExtractScalar) and expression.args.get("scalar_only"): 110 return self.func("JSON_VALUE", expression.this, expression.expression) 111 112 json_query = self.func("JSON_QUERY", expression.this, expression.expression) 113 if expression.args.get("json_query"): 114 return json_query 115 116 json_value = self.func("JSON_VALUE", expression.this, expression.expression) 117 return self.func("ISNULL", json_query, json_value) 118 119 120def _timestrtotime_sql(self: TSQLGenerator, expression: exp.TimeStrToTime): 121 sql = timestrtotime_sql(self, expression) 122 if expression.args.get("zone"): 123 # If there is a timezone, produce an expression like: 124 # CAST('2020-01-01 12:13:14-08:00' AS DATETIMEOFFSET) AT TIME ZONE 'UTC' 125 # If you dont have AT TIME ZONE 'UTC', wrapping that expression in another cast back to DATETIME2 just drops the timezone information 126 return self.sql(exp.AtTimeZone(this=sql, zone=exp.Literal.string("UTC"))) 127 return sql 128 129 130class TSQLGenerator(generator.Generator): 131 SELECT_KINDS: tuple[str, ...] = () 132 TRY_SUPPORTED = False 133 SUPPORTS_UESCAPE = False 134 SUPPORTS_DECODE_CASE = False 135 136 AFTER_HAVING_MODIFIER_TRANSFORMS = generator.AFTER_HAVING_MODIFIER_TRANSFORMS 137 138 LIMIT_IS_TOP = True 139 QUERY_HINTS = False 140 RETURNING_END = False 141 NVL2_SUPPORTED = False 142 ALTER_TABLE_INCLUDE_COLUMN_KEYWORD = False 143 LIMIT_FETCH = "FETCH" 144 COMPUTED_COLUMN_WITH_TYPE = False 145 CTE_RECURSIVE_KEYWORD_REQUIRED = False 146 ENSURE_BOOLS = True 147 NULL_ORDERING_SUPPORTED: bool | None = None 148 SUPPORTS_SINGLE_ARG_CONCAT = False 149 TABLESAMPLE_SEED_KEYWORD = "REPEATABLE" 150 SUPPORTS_SELECT_INTO = True 151 JSON_PATH_BRACKETED_KEY_SUPPORTED = False 152 SUPPORTS_TO_NUMBER = False 153 SET_OP_MODIFIERS = False 154 COPY_PARAMS_EQ_REQUIRED = True 155 PARSE_JSON_NAME: str | None = None 156 EXCEPT_INTERSECT_SUPPORT_ALL_CLAUSE = False 157 ALTER_SET_WRAPPED = True 158 ALTER_SET_TYPE = "" 159 SUPPORTS_ALTER_COLUMN_NULLABILITY = True 160 161 EXPRESSIONS_WITHOUT_NESTED_CTES = { 162 exp.Create, 163 exp.Delete, 164 exp.Insert, 165 exp.Intersect, 166 exp.Except, 167 exp.Merge, 168 exp.Select, 169 exp.Subquery, 170 exp.Union, 171 exp.Update, 172 } 173 174 SUPPORTED_JSON_PATH_PARTS = { 175 exp.JSONPathKey, 176 exp.JSONPathRoot, 177 exp.JSONPathSubscript, 178 } 179 180 TYPE_MAPPING = { 181 **{ 182 k: v 183 for k, v in generator.Generator.TYPE_MAPPING.items() 184 if k not in (exp.DType.NCHAR, exp.DType.NVARCHAR) 185 }, 186 exp.DType.BOOLEAN: "BIT", 187 exp.DType.DATETIME2: "DATETIME2", 188 exp.DType.DECIMAL: "NUMERIC", 189 exp.DType.DOUBLE: "FLOAT", 190 exp.DType.INT: "INTEGER", 191 exp.DType.ROWVERSION: "ROWVERSION", 192 exp.DType.TEXT: "VARCHAR(MAX)", 193 exp.DType.TIMESTAMP: "DATETIME2", 194 exp.DType.TIMESTAMPNTZ: "DATETIME2", 195 exp.DType.TIMESTAMPTZ: "DATETIMEOFFSET", 196 exp.DType.SMALLDATETIME: "SMALLDATETIME", 197 exp.DType.UTINYINT: "TINYINT", 198 exp.DType.VARIANT: "SQL_VARIANT", 199 exp.DType.UUID: "UNIQUEIDENTIFIER", 200 } 201 202 TRANSFORMS = { 203 **{k: v for k, v in generator.Generator.TRANSFORMS.items() if k != exp.ReturnsProperty}, 204 exp.AnyValue: any_value_to_max_sql, 205 exp.Atan2: rename_func("ATN2"), 206 exp.ArrayToString: rename_func("STRING_AGG"), 207 exp.AutoIncrementColumnConstraint: lambda *_: "IDENTITY", 208 exp.Ceil: rename_func("CEILING"), 209 exp.Chr: rename_func("CHAR"), 210 exp.DateAdd: date_delta_sql("DATEADD"), 211 exp.CTE: transforms.preprocess([qualify_derived_table_outputs]), 212 exp.CurrentDate: rename_func("GETDATE"), 213 exp.CurrentTimestamp: rename_func("GETDATE"), 214 exp.CurrentTimestampLTZ: rename_func("SYSDATETIMEOFFSET"), 215 exp.DateStrToDate: datestrtodate_sql, 216 exp.Day: remove_ts_or_ds_to_date(), 217 exp.GeneratedAsIdentityColumnConstraint: generatedasidentitycolumnconstraint_sql, 218 exp.GroupConcat: _string_agg_sql, 219 exp.If: rename_func("IIF"), 220 exp.JSONExtract: _json_extract_sql, 221 exp.JSONExtractScalar: _json_extract_sql, 222 exp.LastDay: lambda self, e: self.func("EOMONTH", e.this), 223 exp.Ln: rename_func("LOG"), 224 exp.Max: max_or_greatest, 225 exp.MD5: lambda self, e: self.func("HASHBYTES", exp.Literal.string("MD5"), e.this), 226 exp.Min: min_or_least, 227 exp.Month: remove_ts_or_ds_to_date(), 228 exp.NumberToStr: _format_sql, 229 exp.Repeat: rename_func("REPLICATE"), 230 exp.CurrentSchema: rename_func("SCHEMA_NAME"), 231 exp.Select: transforms.preprocess( 232 [ 233 transforms.eliminate_distinct_on, 234 transforms.eliminate_semi_and_anti_joins, 235 transforms.eliminate_qualify, 236 transforms.unnest_generate_date_array_using_recursive_cte, 237 ] 238 ), 239 exp.Stddev: rename_func("STDEV"), 240 exp.StrPosition: lambda self, e: strposition_sql( 241 self, e, func_name="CHARINDEX", supports_position=True 242 ), 243 exp.Subquery: transforms.preprocess([qualify_derived_table_outputs]), 244 exp.SHA: lambda self, e: self.func("HASHBYTES", exp.Literal.string("SHA1"), e.this), 245 exp.SHA1Digest: lambda self, e: self.func("HASHBYTES", exp.Literal.string("SHA1"), e.this), 246 exp.SHA2: lambda self, e: self.func( 247 "HASHBYTES", exp.Literal.string(f"SHA2_{e.args.get('length', 256)}"), e.this 248 ), 249 exp.TemporaryProperty: lambda self, e: "", 250 exp.TimeStrToTime: _timestrtotime_sql, 251 exp.TimeToStr: _format_sql, 252 exp.TimestampAdd: date_delta_sql("DATEADD"), 253 exp.Trim: trim_sql, 254 exp.TsOrDsAdd: date_delta_sql("DATEADD", cast=True), 255 exp.TsOrDsDiff: date_delta_sql("DATEDIFF"), 256 exp.TimestampTrunc: lambda self, e: self.func("DATETRUNC", e.unit, e.this), 257 exp.Trunc: lambda self, e: self.func( 258 "ROUND", 259 e.this, 260 e.args.get("decimals") or exp.Literal.number(0), 261 exp.Literal.number(1), 262 ), 263 exp.Uuid: lambda *_: "NEWID()", 264 exp.Year: remove_ts_or_ds_to_date(), 265 exp.DateFromParts: rename_func("DATEFROMPARTS"), 266 } 267 268 PROPERTIES_LOCATION = { 269 **generator.Generator.PROPERTIES_LOCATION, 270 exp.VolatileProperty: exp.Properties.Location.UNSUPPORTED, 271 } 272 273 def scope_resolution(self, rhs: str, scope_name: str) -> str: 274 return f"{scope_name}::{rhs}" 275 276 def select_sql(self, expression: exp.Select) -> str: 277 limit = expression.args.get("limit") 278 offset = expression.args.get("offset") 279 280 if isinstance(limit, exp.Fetch) and not offset: 281 # Dialects like Oracle can FETCH directly from a row set but 282 # T-SQL requires an ORDER BY + OFFSET clause in order to FETCH 283 offset = exp.Offset(expression=exp.Literal.number(0)) 284 expression.set("offset", offset) 285 286 if offset: 287 if not expression.args.get("order"): 288 # ORDER BY is required in order to use OFFSET in a query, so we use 289 # a noop order by, since we don't really care about the order. 290 # See: https://www.microsoftpressstore.com/articles/article.aspx?p=2314819 291 expression.order_by(exp.select(exp.null()).subquery(), copy=False) 292 293 if isinstance(limit, exp.Limit): 294 # TOP and OFFSET can't be combined, we need use FETCH instead of TOP 295 # we replace here because otherwise TOP would be generated in select_sql 296 limit.replace(exp.Fetch(direction="FIRST", count=limit.expression)) 297 298 return super().select_sql(expression) 299 300 def convert_sql(self, expression: exp.Convert) -> str: 301 name = "TRY_CONVERT" if expression.args.get("safe") else "CONVERT" 302 return self.func(name, expression.this, expression.expression, expression.args.get("style")) 303 304 def queryoption_sql(self, expression: exp.QueryOption) -> str: 305 option = self.sql(expression, "this") 306 value = self.sql(expression, "expression") 307 if value: 308 optional_equal_sign = "= " if option in OPTIONS_THAT_REQUIRE_EQUAL else "" 309 return f"{option} {optional_equal_sign}{value}" 310 return option 311 312 def lateral_op(self, expression: exp.Lateral) -> str: 313 cross_apply = expression.args.get("cross_apply") 314 if cross_apply is True: 315 return "CROSS APPLY" 316 if cross_apply is False: 317 return "OUTER APPLY" 318 319 # TODO: perhaps we can check if the parent is a Join and transpile it appropriately 320 self.unsupported("LATERAL clause is not supported.") 321 return "LATERAL" 322 323 def splitpart_sql(self, expression: exp.SplitPart) -> str: 324 this = expression.this 325 split_count = len(this.name.split(".")) 326 delimiter = expression.args.get("delimiter") 327 part_index = expression.args.get("part_index") 328 329 if ( 330 not all(isinstance(arg, exp.Literal) for arg in (this, delimiter, part_index)) 331 or (delimiter and delimiter.name != ".") 332 or not part_index 333 or split_count > 4 334 ): 335 self.unsupported( 336 "SPLIT_PART can be transpiled to PARSENAME only for '.' delimiter and literal values" 337 ) 338 return "" 339 340 return self.func( 341 "PARSENAME", this, exp.Literal.number(split_count + 1 - part_index.to_py()) 342 ) 343 344 def extract_sql(self, expression: exp.Extract) -> str: 345 part = expression.this 346 name = DATE_PART_UNMAPPING.get(part.name.upper()) or part 347 348 return self.func("DATEPART", name, expression.expression) 349 350 def timefromparts_sql(self, expression: exp.TimeFromParts) -> str: 351 nano = expression.args.get("nano") 352 if nano is not None: 353 nano.pop() 354 self.unsupported("Specifying nanoseconds is not supported in TIMEFROMPARTS.") 355 356 if expression.args.get("fractions") is None: 357 expression.set("fractions", exp.Literal.number(0)) 358 if expression.args.get("precision") is None: 359 expression.set("precision", exp.Literal.number(0)) 360 361 return rename_func("TIMEFROMPARTS")(self, expression) 362 363 def timestampfromparts_sql(self, expression: exp.TimestampFromParts) -> str: 364 zone = expression.args.get("zone") 365 if zone is not None: 366 zone.pop() 367 self.unsupported("Time zone is not supported in DATETIMEFROMPARTS.") 368 369 nano = expression.args.get("nano") 370 if nano is not None: 371 nano.pop() 372 self.unsupported("Specifying nanoseconds is not supported in DATETIMEFROMPARTS.") 373 374 if expression.args.get("milli") is None: 375 expression.set("milli", exp.Literal.number(0)) 376 377 return rename_func("DATETIMEFROMPARTS")(self, expression) 378 379 def setitem_sql(self, expression: exp.SetItem) -> str: 380 this = expression.this 381 if isinstance(this, exp.EQ) and not isinstance(this.left, exp.Parameter): 382 # T-SQL does not use '=' in SET command, except when the LHS is a variable. 383 return f"{self.sql(this.left)} {self.sql(this.right)}" 384 385 return super().setitem_sql(expression) 386 387 def boolean_sql(self, expression: exp.Boolean) -> str: 388 if type(expression.parent) in BIT_TYPES or isinstance( 389 expression.find_ancestor(exp.Values, exp.Select), exp.Values 390 ): 391 return "1" if expression.this else "0" 392 393 return "(1 = 1)" if expression.this else "(1 = 0)" 394 395 def is_sql(self, expression: exp.Is) -> str: 396 negate = expression.args.get("negate") 397 if isinstance(expression.expression, exp.Boolean): 398 return self.binary(expression, "<>" if negate else "=") 399 return self.binary(expression, "IS NOT" if negate else "IS") 400 401 def createable_sql(self, expression: exp.Create, locations: defaultdict) -> str: 402 sql = self.sql(expression, "this") 403 properties = expression.args.get("properties") 404 405 start = self._identifier_start 406 if ( 407 not sql.startswith("#") 408 and not sql.startswith(f"{start}#") 409 and any( 410 isinstance(prop, exp.TemporaryProperty) 411 for prop in (properties.expressions if properties else []) 412 ) 413 ): 414 sql = f"{start}#{sql[len(start) :]}" if sql.startswith(start) else f"#{sql}" 415 416 return sql 417 418 def create_sql(self, expression: exp.Create) -> str: 419 kind = expression.kind 420 exists = expression.args.get("exists") 421 expression.set("exists", None) 422 423 like_property = expression.find(exp.LikeProperty) 424 if like_property: 425 ctas_expression = like_property.this 426 else: 427 ctas_expression = expression.expression 428 429 if kind == "VIEW": 430 expression.this.set("catalog", None) 431 with_ = expression.args.get("with_") 432 if ctas_expression and with_: 433 # We've already preprocessed the Create expression to bubble up any nested CTEs, 434 # but CREATE VIEW actually requires the WITH clause to come after it so we need 435 # to amend the AST by moving the CTEs to the CREATE VIEW statement's query. 436 ctas_expression.set("with_", with_.pop()) 437 elif ( 438 kind == "FUNCTION" 439 and isinstance(ctas_expression, exp.Return) 440 and isinstance(body := ctas_expression.this.unnest(), exp.Query) 441 and (with_ := expression.args.get("with_")) 442 ): 443 # Similar to the VIEW branch, the table-valued functions require the WITH clause 444 # to stay inside the RETURN body, so we move back any CTEs that were bubbled up. 445 body.set("with_", with_.pop()) 446 447 table = expression.find(exp.Table) 448 449 # Convert CTAS statement to SELECT .. INTO .. 450 if kind == "TABLE" and ctas_expression: 451 if isinstance(ctas_expression, exp.UNWRAPPED_QUERIES): 452 ctas_expression = ctas_expression.subquery() 453 454 properties = expression.args.get("properties") or exp.Properties() 455 is_temp = any(isinstance(p, exp.TemporaryProperty) for p in properties.expressions) 456 457 select_into = exp.select("*").from_(exp.alias_(ctas_expression, "temp", table=True)) 458 select_into.set("into", exp.Into(this=table, temporary=is_temp)) 459 460 if like_property: 461 select_into.limit(0, copy=False) 462 463 sql = self.sql(select_into) 464 else: 465 sql = super().create_sql(expression) 466 467 if exists: 468 identifier = self.sql(exp.Literal.string(exp.table_name(table) if table else "")) 469 sql_with_ctes = self.prepend_ctes(expression, sql) 470 sql_literal = self.sql(exp.Literal.string(sql_with_ctes)) 471 if kind == "SCHEMA": 472 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = {identifier}) EXEC({sql_literal})""" 473 elif kind == "TABLE": 474 assert table 475 where = exp.and_( 476 exp.column("TABLE_NAME").eq(table.name), 477 exp.column("TABLE_SCHEMA").eq(table.db) if table.db else None, 478 exp.column("TABLE_CATALOG").eq(table.catalog) if table.catalog else None, 479 ) 480 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE {where}) EXEC({sql_literal})""" 481 elif kind == "INDEX": 482 index = self.sql(exp.Literal.string(expression.this.text("this"))) 483 return f"""IF NOT EXISTS (SELECT * FROM sys.indexes WHERE object_id = object_id({identifier}) AND name = {index}) EXEC({sql_literal})""" 484 elif expression.args.get("replace"): 485 sql = sql.replace("CREATE OR REPLACE ", "CREATE OR ALTER ", 1) 486 487 return self.prepend_ctes(expression, sql) 488 489 @generator.unsupported_args("unlogged", "expressions") 490 def into_sql(self, expression: exp.Into) -> str: 491 if expression.args.get("temporary"): 492 # If the Into expression has a temporary property, push this down to the Identifier 493 table = expression.find(exp.Table) 494 if table and isinstance(table.this, exp.Identifier): 495 table.this.set("temporary", True) 496 497 return f"{self.seg('INTO')} {self.sql(expression, 'this')}" 498 499 def count_sql(self, expression: exp.Count) -> str: 500 func_name = "COUNT_BIG" if expression.args.get("big_int") else "COUNT" 501 return rename_func(func_name)(self, expression) 502 503 def datediff_sql(self, expression: exp.DateDiff) -> str: 504 func_name = "DATEDIFF_BIG" if expression.args.get("big_int") else "DATEDIFF" 505 return date_delta_sql(func_name)(self, expression) 506 507 def offset_sql(self, expression: exp.Offset) -> str: 508 return f"{super().offset_sql(expression)} ROWS" 509 510 def version_sql(self, expression: exp.Version) -> str: 511 name = "SYSTEM_TIME" if expression.name == "TIMESTAMP" else expression.name 512 this = f"FOR {name}" 513 expr = expression.expression 514 kind = expression.text("kind") 515 if kind in ("FROM", "BETWEEN"): 516 args = expr.expressions 517 sep = "TO" if kind == "FROM" else "AND" 518 expr_sql = f"{self.sql(seq_get(args, 0))} {sep} {self.sql(seq_get(args, 1))}" 519 else: 520 expr_sql = self.sql(expr) 521 522 expr_sql = f" {expr_sql}" if expr_sql else "" 523 return f"{this} {kind}{expr_sql}" 524 525 def returnsproperty_sql(self, expression: exp.ReturnsProperty) -> str: 526 table = expression.args.get("table") 527 table = f"{table} " if table else "" 528 return f"RETURNS {table}{self.sql(expression, 'this')}" 529 530 def returning_sql(self, expression: exp.Returning) -> str: 531 into = self.sql(expression, "into") 532 into = self.seg(f"INTO {into}") if into else "" 533 return f"{self.seg('OUTPUT')} {self.expressions(expression, flat=True)}{into}" 534 535 def transaction_sql(self, expression: exp.Transaction) -> str: 536 this = self.sql(expression, "this") 537 this = f" {this}" if this else "" 538 mark = self.sql(expression, "mark") 539 mark = f" WITH MARK {mark}" if mark else "" 540 return f"BEGIN TRANSACTION{this}{mark}" 541 542 def commit_sql(self, expression: exp.Commit) -> str: 543 this = self.sql(expression, "this") 544 this = f" {this}" if this else "" 545 durability = expression.args.get("durability") 546 durability = ( 547 f" WITH (DELAYED_DURABILITY = {'ON' if durability else 'OFF'})" 548 if durability is not None 549 else "" 550 ) 551 return f"COMMIT TRANSACTION{this}{durability}" 552 553 def rollback_sql(self, expression: exp.Rollback) -> str: 554 this = self.sql(expression, "this") 555 this = f" {this}" if this else "" 556 return f"ROLLBACK TRANSACTION{this}" 557 558 def identifier_sql(self, expression: exp.Identifier) -> str: 559 identifier = super().identifier_sql(expression) 560 561 if expression.args.get("global_"): 562 prefix = "##" 563 elif expression.args.get("temporary"): 564 prefix = "#" 565 else: 566 return identifier 567 568 start = self._identifier_start 569 if expression.quoted and identifier.startswith(start): 570 return f"{start}{prefix}{identifier[len(start) :]}" 571 572 return f"{prefix}{identifier}" 573 574 def constraint_sql(self, expression: exp.Constraint) -> str: 575 this = self.sql(expression, "this") 576 expressions = self.expressions(expression, flat=True, sep=" ") 577 return f"CONSTRAINT {this} {expressions}" 578 579 def length_sql(self, expression: exp.Length) -> str: 580 return self._uncast_text(expression, "LEN") 581 582 def right_sql(self, expression: exp.Right) -> str: 583 return self._uncast_text(expression, "RIGHT") 584 585 def left_sql(self, expression: exp.Left) -> str: 586 return self._uncast_text(expression, "LEFT") 587 588 def _uncast_text(self, expression: exp.Expr, name: str) -> str: 589 this = expression.this 590 if isinstance(this, exp.Cast) and this.is_type(exp.DType.TEXT): 591 this_sql = self.sql(this, "this") 592 else: 593 this_sql = self.sql(this) 594 expression_sql = self.sql(expression, "expression") 595 return self.func(name, this_sql, expression_sql if expression_sql else None) 596 597 def partition_sql(self, expression: exp.Partition) -> str: 598 return f"WITH (PARTITIONS({self.expressions(expression, flat=True)}))" 599 600 def alter_sql(self, expression: exp.Alter) -> str: 601 action = seq_get(expression.args.get("actions") or [], 0) 602 if isinstance(action, exp.AlterRename): 603 return f"EXEC sp_rename '{self.sql(expression.this)}', '{action.this.name}'" 604 return super().alter_sql(expression) 605 606 def drop_sql(self, expression: exp.Drop) -> str: 607 if expression.args["kind"] == "VIEW": 608 for table in expression.args.get("tables") or []: 609 table.set("catalog", None) 610 return super().drop_sql(expression) 611 612 def options_modifier(self, expression: exp.Expr) -> str: 613 options = self.expressions(expression, key="options") 614 return f" OPTION{self.wrap(options)}" if options else "" 615 616 def dpipe_sql(self, expression: exp.DPipe) -> str: 617 return self.sql(reduce(lambda x, y: exp.Add(this=x, expression=y), expression.flatten())) 618 619 def isascii_sql(self, expression: exp.IsAscii) -> str: 620 return f"(PATINDEX(CONVERT(VARCHAR(MAX), 0x255b5e002d7f5d25) COLLATE Latin1_General_BIN, {self.sql(expression.this)}) = 0)" 621 622 def columndef_sql(self, expression: exp.ColumnDef, sep: str = " ") -> str: 623 this = super().columndef_sql(expression, sep) 624 default = self.sql(expression, "default") 625 default = f" = {default}" if default else "" 626 output = self.sql(expression, "output") 627 output = f" {output}" if output else "" 628 return f"{this}{default}{output}" 629 630 def coalesce_sql(self, expression: exp.Coalesce) -> str: 631 func_name = "ISNULL" if expression.args.get("is_null") else "COALESCE" 632 return rename_func(func_name)(self, expression) 633 634 def storedprocedure_sql(self, expression: exp.StoredProcedure) -> str: 635 this = self.sql(expression, "this") 636 expressions = self.expressions(expression) 637 expressions = ( 638 self.wrap(expressions) if expression.args.get("wrapped") else f" {expressions}" 639 ) 640 return f"{this}{expressions}" if expressions.strip() != "" else this 641 642 def ifblock_sql(self, expression: exp.IfBlock) -> str: 643 this = self.sql(expression, "this") 644 true = self.sql(expression, "true") 645 true = f" {true}" if true else " " 646 false = self.sql(expression, "false") 647 false = f"; ELSE BEGIN {false}" if false else "" 648 return f"IF {this} BEGIN{true}{false}" 649 650 def whileblock_sql(self, expression: exp.WhileBlock) -> str: 651 this = self.sql(expression, "this") 652 body = self.sql(expression, "body") 653 body = f" {body}" if body else " " 654 return f"WHILE {this} BEGIN{body}" 655 656 def execute_sql(self, expression: exp.Execute) -> str: 657 this = self.sql(expression, "this") 658 expressions = self.expressions(expression) 659 expressions = f" {expressions}" if expressions else "" 660 return_status = self.sql(expression, "return_status") 661 return_status = f"{return_status} = " if return_status else "" 662 return f"EXECUTE {return_status}{this}{expressions}" 663 664 def executesql_sql(self, expression: exp.ExecuteSql) -> str: 665 return self.execute_sql(expression)
DATE_PART_UNMAPPING =
{'WEEKISO': 'ISO_WEEK', 'DAYOFWEEK': 'WEEKDAY', 'TIMEZONE_MINUTE': 'TZOFFSET'}
BIT_TYPES =
{<class 'sqlglot.expressions.core.EQ'>, <class 'sqlglot.expressions.core.Is'>, <class 'sqlglot.expressions.core.NEQ'>, <class 'sqlglot.expressions.core.Alias'>, <class 'sqlglot.expressions.query.Select'>, <class 'sqlglot.expressions.core.In'>}
def
qualify_derived_table_outputs( expression: sqlglot.expressions.core.Expr) -> sqlglot.expressions.core.Expr:
71def qualify_derived_table_outputs(expression: exp.Expr) -> exp.Expr: 72 """Ensures all (unnamed) output columns are aliased for CTEs and Subqueries.""" 73 alias = expression.args.get("alias") 74 75 if ( 76 isinstance(expression, (exp.CTE, exp.Subquery)) 77 and isinstance(alias, exp.TableAlias) 78 and not alias.columns 79 ): 80 from sqlglot.dialects.tsql import TSQL 81 from sqlglot.optimizer.qualify_columns import qualify_outputs 82 83 # We keep track of the unaliased column projection indexes instead of the expressions 84 # themselves, because the latter are going to be replaced by new nodes when the aliases 85 # are added and hence we won't be able to reach these newly added Alias parents 86 query = expression.this 87 unaliased_column_indexes = ( 88 i for i, c in enumerate(query.selects) if isinstance(c, exp.Column) and not c.alias 89 ) 90 91 qualify_outputs(query, dialect=TSQL()) 92 93 # Preserve the quoting information of columns for newly added Alias nodes 94 query_selects = query.selects 95 for select_index in unaliased_column_indexes: 96 alias = query_selects[select_index] 97 column = alias.this 98 if isinstance(column.this, exp.Identifier): 99 alias.args["alias"].set("quoted", column.this.quoted) 100 101 return expression
Ensures all (unnamed) output columns are aliased for CTEs and Subqueries.
131class TSQLGenerator(generator.Generator): 132 SELECT_KINDS: tuple[str, ...] = () 133 TRY_SUPPORTED = False 134 SUPPORTS_UESCAPE = False 135 SUPPORTS_DECODE_CASE = False 136 137 AFTER_HAVING_MODIFIER_TRANSFORMS = generator.AFTER_HAVING_MODIFIER_TRANSFORMS 138 139 LIMIT_IS_TOP = True 140 QUERY_HINTS = False 141 RETURNING_END = False 142 NVL2_SUPPORTED = False 143 ALTER_TABLE_INCLUDE_COLUMN_KEYWORD = False 144 LIMIT_FETCH = "FETCH" 145 COMPUTED_COLUMN_WITH_TYPE = False 146 CTE_RECURSIVE_KEYWORD_REQUIRED = False 147 ENSURE_BOOLS = True 148 NULL_ORDERING_SUPPORTED: bool | None = None 149 SUPPORTS_SINGLE_ARG_CONCAT = False 150 TABLESAMPLE_SEED_KEYWORD = "REPEATABLE" 151 SUPPORTS_SELECT_INTO = True 152 JSON_PATH_BRACKETED_KEY_SUPPORTED = False 153 SUPPORTS_TO_NUMBER = False 154 SET_OP_MODIFIERS = False 155 COPY_PARAMS_EQ_REQUIRED = True 156 PARSE_JSON_NAME: str | None = None 157 EXCEPT_INTERSECT_SUPPORT_ALL_CLAUSE = False 158 ALTER_SET_WRAPPED = True 159 ALTER_SET_TYPE = "" 160 SUPPORTS_ALTER_COLUMN_NULLABILITY = True 161 162 EXPRESSIONS_WITHOUT_NESTED_CTES = { 163 exp.Create, 164 exp.Delete, 165 exp.Insert, 166 exp.Intersect, 167 exp.Except, 168 exp.Merge, 169 exp.Select, 170 exp.Subquery, 171 exp.Union, 172 exp.Update, 173 } 174 175 SUPPORTED_JSON_PATH_PARTS = { 176 exp.JSONPathKey, 177 exp.JSONPathRoot, 178 exp.JSONPathSubscript, 179 } 180 181 TYPE_MAPPING = { 182 **{ 183 k: v 184 for k, v in generator.Generator.TYPE_MAPPING.items() 185 if k not in (exp.DType.NCHAR, exp.DType.NVARCHAR) 186 }, 187 exp.DType.BOOLEAN: "BIT", 188 exp.DType.DATETIME2: "DATETIME2", 189 exp.DType.DECIMAL: "NUMERIC", 190 exp.DType.DOUBLE: "FLOAT", 191 exp.DType.INT: "INTEGER", 192 exp.DType.ROWVERSION: "ROWVERSION", 193 exp.DType.TEXT: "VARCHAR(MAX)", 194 exp.DType.TIMESTAMP: "DATETIME2", 195 exp.DType.TIMESTAMPNTZ: "DATETIME2", 196 exp.DType.TIMESTAMPTZ: "DATETIMEOFFSET", 197 exp.DType.SMALLDATETIME: "SMALLDATETIME", 198 exp.DType.UTINYINT: "TINYINT", 199 exp.DType.VARIANT: "SQL_VARIANT", 200 exp.DType.UUID: "UNIQUEIDENTIFIER", 201 } 202 203 TRANSFORMS = { 204 **{k: v for k, v in generator.Generator.TRANSFORMS.items() if k != exp.ReturnsProperty}, 205 exp.AnyValue: any_value_to_max_sql, 206 exp.Atan2: rename_func("ATN2"), 207 exp.ArrayToString: rename_func("STRING_AGG"), 208 exp.AutoIncrementColumnConstraint: lambda *_: "IDENTITY", 209 exp.Ceil: rename_func("CEILING"), 210 exp.Chr: rename_func("CHAR"), 211 exp.DateAdd: date_delta_sql("DATEADD"), 212 exp.CTE: transforms.preprocess([qualify_derived_table_outputs]), 213 exp.CurrentDate: rename_func("GETDATE"), 214 exp.CurrentTimestamp: rename_func("GETDATE"), 215 exp.CurrentTimestampLTZ: rename_func("SYSDATETIMEOFFSET"), 216 exp.DateStrToDate: datestrtodate_sql, 217 exp.Day: remove_ts_or_ds_to_date(), 218 exp.GeneratedAsIdentityColumnConstraint: generatedasidentitycolumnconstraint_sql, 219 exp.GroupConcat: _string_agg_sql, 220 exp.If: rename_func("IIF"), 221 exp.JSONExtract: _json_extract_sql, 222 exp.JSONExtractScalar: _json_extract_sql, 223 exp.LastDay: lambda self, e: self.func("EOMONTH", e.this), 224 exp.Ln: rename_func("LOG"), 225 exp.Max: max_or_greatest, 226 exp.MD5: lambda self, e: self.func("HASHBYTES", exp.Literal.string("MD5"), e.this), 227 exp.Min: min_or_least, 228 exp.Month: remove_ts_or_ds_to_date(), 229 exp.NumberToStr: _format_sql, 230 exp.Repeat: rename_func("REPLICATE"), 231 exp.CurrentSchema: rename_func("SCHEMA_NAME"), 232 exp.Select: transforms.preprocess( 233 [ 234 transforms.eliminate_distinct_on, 235 transforms.eliminate_semi_and_anti_joins, 236 transforms.eliminate_qualify, 237 transforms.unnest_generate_date_array_using_recursive_cte, 238 ] 239 ), 240 exp.Stddev: rename_func("STDEV"), 241 exp.StrPosition: lambda self, e: strposition_sql( 242 self, e, func_name="CHARINDEX", supports_position=True 243 ), 244 exp.Subquery: transforms.preprocess([qualify_derived_table_outputs]), 245 exp.SHA: lambda self, e: self.func("HASHBYTES", exp.Literal.string("SHA1"), e.this), 246 exp.SHA1Digest: lambda self, e: self.func("HASHBYTES", exp.Literal.string("SHA1"), e.this), 247 exp.SHA2: lambda self, e: self.func( 248 "HASHBYTES", exp.Literal.string(f"SHA2_{e.args.get('length', 256)}"), e.this 249 ), 250 exp.TemporaryProperty: lambda self, e: "", 251 exp.TimeStrToTime: _timestrtotime_sql, 252 exp.TimeToStr: _format_sql, 253 exp.TimestampAdd: date_delta_sql("DATEADD"), 254 exp.Trim: trim_sql, 255 exp.TsOrDsAdd: date_delta_sql("DATEADD", cast=True), 256 exp.TsOrDsDiff: date_delta_sql("DATEDIFF"), 257 exp.TimestampTrunc: lambda self, e: self.func("DATETRUNC", e.unit, e.this), 258 exp.Trunc: lambda self, e: self.func( 259 "ROUND", 260 e.this, 261 e.args.get("decimals") or exp.Literal.number(0), 262 exp.Literal.number(1), 263 ), 264 exp.Uuid: lambda *_: "NEWID()", 265 exp.Year: remove_ts_or_ds_to_date(), 266 exp.DateFromParts: rename_func("DATEFROMPARTS"), 267 } 268 269 PROPERTIES_LOCATION = { 270 **generator.Generator.PROPERTIES_LOCATION, 271 exp.VolatileProperty: exp.Properties.Location.UNSUPPORTED, 272 } 273 274 def scope_resolution(self, rhs: str, scope_name: str) -> str: 275 return f"{scope_name}::{rhs}" 276 277 def select_sql(self, expression: exp.Select) -> str: 278 limit = expression.args.get("limit") 279 offset = expression.args.get("offset") 280 281 if isinstance(limit, exp.Fetch) and not offset: 282 # Dialects like Oracle can FETCH directly from a row set but 283 # T-SQL requires an ORDER BY + OFFSET clause in order to FETCH 284 offset = exp.Offset(expression=exp.Literal.number(0)) 285 expression.set("offset", offset) 286 287 if offset: 288 if not expression.args.get("order"): 289 # ORDER BY is required in order to use OFFSET in a query, so we use 290 # a noop order by, since we don't really care about the order. 291 # See: https://www.microsoftpressstore.com/articles/article.aspx?p=2314819 292 expression.order_by(exp.select(exp.null()).subquery(), copy=False) 293 294 if isinstance(limit, exp.Limit): 295 # TOP and OFFSET can't be combined, we need use FETCH instead of TOP 296 # we replace here because otherwise TOP would be generated in select_sql 297 limit.replace(exp.Fetch(direction="FIRST", count=limit.expression)) 298 299 return super().select_sql(expression) 300 301 def convert_sql(self, expression: exp.Convert) -> str: 302 name = "TRY_CONVERT" if expression.args.get("safe") else "CONVERT" 303 return self.func(name, expression.this, expression.expression, expression.args.get("style")) 304 305 def queryoption_sql(self, expression: exp.QueryOption) -> str: 306 option = self.sql(expression, "this") 307 value = self.sql(expression, "expression") 308 if value: 309 optional_equal_sign = "= " if option in OPTIONS_THAT_REQUIRE_EQUAL else "" 310 return f"{option} {optional_equal_sign}{value}" 311 return option 312 313 def lateral_op(self, expression: exp.Lateral) -> str: 314 cross_apply = expression.args.get("cross_apply") 315 if cross_apply is True: 316 return "CROSS APPLY" 317 if cross_apply is False: 318 return "OUTER APPLY" 319 320 # TODO: perhaps we can check if the parent is a Join and transpile it appropriately 321 self.unsupported("LATERAL clause is not supported.") 322 return "LATERAL" 323 324 def splitpart_sql(self, expression: exp.SplitPart) -> str: 325 this = expression.this 326 split_count = len(this.name.split(".")) 327 delimiter = expression.args.get("delimiter") 328 part_index = expression.args.get("part_index") 329 330 if ( 331 not all(isinstance(arg, exp.Literal) for arg in (this, delimiter, part_index)) 332 or (delimiter and delimiter.name != ".") 333 or not part_index 334 or split_count > 4 335 ): 336 self.unsupported( 337 "SPLIT_PART can be transpiled to PARSENAME only for '.' delimiter and literal values" 338 ) 339 return "" 340 341 return self.func( 342 "PARSENAME", this, exp.Literal.number(split_count + 1 - part_index.to_py()) 343 ) 344 345 def extract_sql(self, expression: exp.Extract) -> str: 346 part = expression.this 347 name = DATE_PART_UNMAPPING.get(part.name.upper()) or part 348 349 return self.func("DATEPART", name, expression.expression) 350 351 def timefromparts_sql(self, expression: exp.TimeFromParts) -> str: 352 nano = expression.args.get("nano") 353 if nano is not None: 354 nano.pop() 355 self.unsupported("Specifying nanoseconds is not supported in TIMEFROMPARTS.") 356 357 if expression.args.get("fractions") is None: 358 expression.set("fractions", exp.Literal.number(0)) 359 if expression.args.get("precision") is None: 360 expression.set("precision", exp.Literal.number(0)) 361 362 return rename_func("TIMEFROMPARTS")(self, expression) 363 364 def timestampfromparts_sql(self, expression: exp.TimestampFromParts) -> str: 365 zone = expression.args.get("zone") 366 if zone is not None: 367 zone.pop() 368 self.unsupported("Time zone is not supported in DATETIMEFROMPARTS.") 369 370 nano = expression.args.get("nano") 371 if nano is not None: 372 nano.pop() 373 self.unsupported("Specifying nanoseconds is not supported in DATETIMEFROMPARTS.") 374 375 if expression.args.get("milli") is None: 376 expression.set("milli", exp.Literal.number(0)) 377 378 return rename_func("DATETIMEFROMPARTS")(self, expression) 379 380 def setitem_sql(self, expression: exp.SetItem) -> str: 381 this = expression.this 382 if isinstance(this, exp.EQ) and not isinstance(this.left, exp.Parameter): 383 # T-SQL does not use '=' in SET command, except when the LHS is a variable. 384 return f"{self.sql(this.left)} {self.sql(this.right)}" 385 386 return super().setitem_sql(expression) 387 388 def boolean_sql(self, expression: exp.Boolean) -> str: 389 if type(expression.parent) in BIT_TYPES or isinstance( 390 expression.find_ancestor(exp.Values, exp.Select), exp.Values 391 ): 392 return "1" if expression.this else "0" 393 394 return "(1 = 1)" if expression.this else "(1 = 0)" 395 396 def is_sql(self, expression: exp.Is) -> str: 397 negate = expression.args.get("negate") 398 if isinstance(expression.expression, exp.Boolean): 399 return self.binary(expression, "<>" if negate else "=") 400 return self.binary(expression, "IS NOT" if negate else "IS") 401 402 def createable_sql(self, expression: exp.Create, locations: defaultdict) -> str: 403 sql = self.sql(expression, "this") 404 properties = expression.args.get("properties") 405 406 start = self._identifier_start 407 if ( 408 not sql.startswith("#") 409 and not sql.startswith(f"{start}#") 410 and any( 411 isinstance(prop, exp.TemporaryProperty) 412 for prop in (properties.expressions if properties else []) 413 ) 414 ): 415 sql = f"{start}#{sql[len(start) :]}" if sql.startswith(start) else f"#{sql}" 416 417 return sql 418 419 def create_sql(self, expression: exp.Create) -> str: 420 kind = expression.kind 421 exists = expression.args.get("exists") 422 expression.set("exists", None) 423 424 like_property = expression.find(exp.LikeProperty) 425 if like_property: 426 ctas_expression = like_property.this 427 else: 428 ctas_expression = expression.expression 429 430 if kind == "VIEW": 431 expression.this.set("catalog", None) 432 with_ = expression.args.get("with_") 433 if ctas_expression and with_: 434 # We've already preprocessed the Create expression to bubble up any nested CTEs, 435 # but CREATE VIEW actually requires the WITH clause to come after it so we need 436 # to amend the AST by moving the CTEs to the CREATE VIEW statement's query. 437 ctas_expression.set("with_", with_.pop()) 438 elif ( 439 kind == "FUNCTION" 440 and isinstance(ctas_expression, exp.Return) 441 and isinstance(body := ctas_expression.this.unnest(), exp.Query) 442 and (with_ := expression.args.get("with_")) 443 ): 444 # Similar to the VIEW branch, the table-valued functions require the WITH clause 445 # to stay inside the RETURN body, so we move back any CTEs that were bubbled up. 446 body.set("with_", with_.pop()) 447 448 table = expression.find(exp.Table) 449 450 # Convert CTAS statement to SELECT .. INTO .. 451 if kind == "TABLE" and ctas_expression: 452 if isinstance(ctas_expression, exp.UNWRAPPED_QUERIES): 453 ctas_expression = ctas_expression.subquery() 454 455 properties = expression.args.get("properties") or exp.Properties() 456 is_temp = any(isinstance(p, exp.TemporaryProperty) for p in properties.expressions) 457 458 select_into = exp.select("*").from_(exp.alias_(ctas_expression, "temp", table=True)) 459 select_into.set("into", exp.Into(this=table, temporary=is_temp)) 460 461 if like_property: 462 select_into.limit(0, copy=False) 463 464 sql = self.sql(select_into) 465 else: 466 sql = super().create_sql(expression) 467 468 if exists: 469 identifier = self.sql(exp.Literal.string(exp.table_name(table) if table else "")) 470 sql_with_ctes = self.prepend_ctes(expression, sql) 471 sql_literal = self.sql(exp.Literal.string(sql_with_ctes)) 472 if kind == "SCHEMA": 473 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = {identifier}) EXEC({sql_literal})""" 474 elif kind == "TABLE": 475 assert table 476 where = exp.and_( 477 exp.column("TABLE_NAME").eq(table.name), 478 exp.column("TABLE_SCHEMA").eq(table.db) if table.db else None, 479 exp.column("TABLE_CATALOG").eq(table.catalog) if table.catalog else None, 480 ) 481 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE {where}) EXEC({sql_literal})""" 482 elif kind == "INDEX": 483 index = self.sql(exp.Literal.string(expression.this.text("this"))) 484 return f"""IF NOT EXISTS (SELECT * FROM sys.indexes WHERE object_id = object_id({identifier}) AND name = {index}) EXEC({sql_literal})""" 485 elif expression.args.get("replace"): 486 sql = sql.replace("CREATE OR REPLACE ", "CREATE OR ALTER ", 1) 487 488 return self.prepend_ctes(expression, sql) 489 490 @generator.unsupported_args("unlogged", "expressions") 491 def into_sql(self, expression: exp.Into) -> str: 492 if expression.args.get("temporary"): 493 # If the Into expression has a temporary property, push this down to the Identifier 494 table = expression.find(exp.Table) 495 if table and isinstance(table.this, exp.Identifier): 496 table.this.set("temporary", True) 497 498 return f"{self.seg('INTO')} {self.sql(expression, 'this')}" 499 500 def count_sql(self, expression: exp.Count) -> str: 501 func_name = "COUNT_BIG" if expression.args.get("big_int") else "COUNT" 502 return rename_func(func_name)(self, expression) 503 504 def datediff_sql(self, expression: exp.DateDiff) -> str: 505 func_name = "DATEDIFF_BIG" if expression.args.get("big_int") else "DATEDIFF" 506 return date_delta_sql(func_name)(self, expression) 507 508 def offset_sql(self, expression: exp.Offset) -> str: 509 return f"{super().offset_sql(expression)} ROWS" 510 511 def version_sql(self, expression: exp.Version) -> str: 512 name = "SYSTEM_TIME" if expression.name == "TIMESTAMP" else expression.name 513 this = f"FOR {name}" 514 expr = expression.expression 515 kind = expression.text("kind") 516 if kind in ("FROM", "BETWEEN"): 517 args = expr.expressions 518 sep = "TO" if kind == "FROM" else "AND" 519 expr_sql = f"{self.sql(seq_get(args, 0))} {sep} {self.sql(seq_get(args, 1))}" 520 else: 521 expr_sql = self.sql(expr) 522 523 expr_sql = f" {expr_sql}" if expr_sql else "" 524 return f"{this} {kind}{expr_sql}" 525 526 def returnsproperty_sql(self, expression: exp.ReturnsProperty) -> str: 527 table = expression.args.get("table") 528 table = f"{table} " if table else "" 529 return f"RETURNS {table}{self.sql(expression, 'this')}" 530 531 def returning_sql(self, expression: exp.Returning) -> str: 532 into = self.sql(expression, "into") 533 into = self.seg(f"INTO {into}") if into else "" 534 return f"{self.seg('OUTPUT')} {self.expressions(expression, flat=True)}{into}" 535 536 def transaction_sql(self, expression: exp.Transaction) -> str: 537 this = self.sql(expression, "this") 538 this = f" {this}" if this else "" 539 mark = self.sql(expression, "mark") 540 mark = f" WITH MARK {mark}" if mark else "" 541 return f"BEGIN TRANSACTION{this}{mark}" 542 543 def commit_sql(self, expression: exp.Commit) -> str: 544 this = self.sql(expression, "this") 545 this = f" {this}" if this else "" 546 durability = expression.args.get("durability") 547 durability = ( 548 f" WITH (DELAYED_DURABILITY = {'ON' if durability else 'OFF'})" 549 if durability is not None 550 else "" 551 ) 552 return f"COMMIT TRANSACTION{this}{durability}" 553 554 def rollback_sql(self, expression: exp.Rollback) -> str: 555 this = self.sql(expression, "this") 556 this = f" {this}" if this else "" 557 return f"ROLLBACK TRANSACTION{this}" 558 559 def identifier_sql(self, expression: exp.Identifier) -> str: 560 identifier = super().identifier_sql(expression) 561 562 if expression.args.get("global_"): 563 prefix = "##" 564 elif expression.args.get("temporary"): 565 prefix = "#" 566 else: 567 return identifier 568 569 start = self._identifier_start 570 if expression.quoted and identifier.startswith(start): 571 return f"{start}{prefix}{identifier[len(start) :]}" 572 573 return f"{prefix}{identifier}" 574 575 def constraint_sql(self, expression: exp.Constraint) -> str: 576 this = self.sql(expression, "this") 577 expressions = self.expressions(expression, flat=True, sep=" ") 578 return f"CONSTRAINT {this} {expressions}" 579 580 def length_sql(self, expression: exp.Length) -> str: 581 return self._uncast_text(expression, "LEN") 582 583 def right_sql(self, expression: exp.Right) -> str: 584 return self._uncast_text(expression, "RIGHT") 585 586 def left_sql(self, expression: exp.Left) -> str: 587 return self._uncast_text(expression, "LEFT") 588 589 def _uncast_text(self, expression: exp.Expr, name: str) -> str: 590 this = expression.this 591 if isinstance(this, exp.Cast) and this.is_type(exp.DType.TEXT): 592 this_sql = self.sql(this, "this") 593 else: 594 this_sql = self.sql(this) 595 expression_sql = self.sql(expression, "expression") 596 return self.func(name, this_sql, expression_sql if expression_sql else None) 597 598 def partition_sql(self, expression: exp.Partition) -> str: 599 return f"WITH (PARTITIONS({self.expressions(expression, flat=True)}))" 600 601 def alter_sql(self, expression: exp.Alter) -> str: 602 action = seq_get(expression.args.get("actions") or [], 0) 603 if isinstance(action, exp.AlterRename): 604 return f"EXEC sp_rename '{self.sql(expression.this)}', '{action.this.name}'" 605 return super().alter_sql(expression) 606 607 def drop_sql(self, expression: exp.Drop) -> str: 608 if expression.args["kind"] == "VIEW": 609 for table in expression.args.get("tables") or []: 610 table.set("catalog", None) 611 return super().drop_sql(expression) 612 613 def options_modifier(self, expression: exp.Expr) -> str: 614 options = self.expressions(expression, key="options") 615 return f" OPTION{self.wrap(options)}" if options else "" 616 617 def dpipe_sql(self, expression: exp.DPipe) -> str: 618 return self.sql(reduce(lambda x, y: exp.Add(this=x, expression=y), expression.flatten())) 619 620 def isascii_sql(self, expression: exp.IsAscii) -> str: 621 return f"(PATINDEX(CONVERT(VARCHAR(MAX), 0x255b5e002d7f5d25) COLLATE Latin1_General_BIN, {self.sql(expression.this)}) = 0)" 622 623 def columndef_sql(self, expression: exp.ColumnDef, sep: str = " ") -> str: 624 this = super().columndef_sql(expression, sep) 625 default = self.sql(expression, "default") 626 default = f" = {default}" if default else "" 627 output = self.sql(expression, "output") 628 output = f" {output}" if output else "" 629 return f"{this}{default}{output}" 630 631 def coalesce_sql(self, expression: exp.Coalesce) -> str: 632 func_name = "ISNULL" if expression.args.get("is_null") else "COALESCE" 633 return rename_func(func_name)(self, expression) 634 635 def storedprocedure_sql(self, expression: exp.StoredProcedure) -> str: 636 this = self.sql(expression, "this") 637 expressions = self.expressions(expression) 638 expressions = ( 639 self.wrap(expressions) if expression.args.get("wrapped") else f" {expressions}" 640 ) 641 return f"{this}{expressions}" if expressions.strip() != "" else this 642 643 def ifblock_sql(self, expression: exp.IfBlock) -> str: 644 this = self.sql(expression, "this") 645 true = self.sql(expression, "true") 646 true = f" {true}" if true else " " 647 false = self.sql(expression, "false") 648 false = f"; ELSE BEGIN {false}" if false else "" 649 return f"IF {this} BEGIN{true}{false}" 650 651 def whileblock_sql(self, expression: exp.WhileBlock) -> str: 652 this = self.sql(expression, "this") 653 body = self.sql(expression, "body") 654 body = f" {body}" if body else " " 655 return f"WHILE {this} BEGIN{body}" 656 657 def execute_sql(self, expression: exp.Execute) -> str: 658 this = self.sql(expression, "this") 659 expressions = self.expressions(expression) 660 expressions = f" {expressions}" if expressions else "" 661 return_status = self.sql(expression, "return_status") 662 return_status = f"{return_status} = " if return_status else "" 663 return f"EXECUTE {return_status}{this}{expressions}" 664 665 def executesql_sql(self, expression: exp.ExecuteSql) -> str: 666 return self.execute_sql(expression)
Generator converts a given syntax tree to the corresponding SQL string.
Arguments:
- pretty: Whether 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: Always quote except for specials cases. 'safe': Only quote identifiers that are case insensitive.
- normalize: Whether to normalize identifiers to lowercase. Default: False.
- pad: The pad size in a formatted string. For example, this affects the indentation of a projection in a query, relative to its nesting level. Default: 2.
- indent: The indentation size in a formatted string. For example, this affects the
indentation of subqueries and filters under a
WHEREclause. Default: 2. - normalize_functions: How to normalize 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: Whether 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 to preserve comments in the output SQL code. Default: True
EXPRESSIONS_WITHOUT_NESTED_CTES =
{<class 'sqlglot.expressions.ddl.Create'>, <class 'sqlglot.expressions.query.Subquery'>, <class 'sqlglot.expressions.dml.Delete'>, <class 'sqlglot.expressions.query.Intersect'>, <class 'sqlglot.expressions.query.Except'>, <class 'sqlglot.expressions.dml.Insert'>, <class 'sqlglot.expressions.query.Union'>, <class 'sqlglot.expressions.dml.Merge'>, <class 'sqlglot.expressions.query.Select'>, <class 'sqlglot.expressions.dml.Update'>}
SUPPORTED_JSON_PATH_PARTS =
{<class 'sqlglot.expressions.query.JSONPathSubscript'>, <class 'sqlglot.expressions.query.JSONPathKey'>, <class 'sqlglot.expressions.query.JSONPathRoot'>}
TYPE_MAPPING =
{<DType.DATETIME2: 'DATETIME2'>: 'DATETIME2', <DType.MEDIUMTEXT: 'MEDIUMTEXT'>: 'TEXT', <DType.LONGTEXT: 'LONGTEXT'>: 'TEXT', <DType.TINYTEXT: 'TINYTEXT'>: 'TEXT', <DType.BLOB: 'BLOB'>: 'VARBINARY', <DType.MEDIUMBLOB: 'MEDIUMBLOB'>: 'BLOB', <DType.LONGBLOB: 'LONGBLOB'>: 'BLOB', <DType.TINYBLOB: 'TINYBLOB'>: 'BLOB', <DType.INET: 'INET'>: 'INET', <DType.ROWVERSION: 'ROWVERSION'>: 'ROWVERSION', <DType.SMALLDATETIME: 'SMALLDATETIME'>: 'SMALLDATETIME', <DType.BOOLEAN: 'BOOLEAN'>: 'BIT', <DType.DECIMAL: 'DECIMAL'>: 'NUMERIC', <DType.DOUBLE: 'DOUBLE'>: 'FLOAT', <DType.INT: 'INT'>: 'INTEGER', <DType.TEXT: 'TEXT'>: 'VARCHAR(MAX)', <DType.TIMESTAMP: 'TIMESTAMP'>: 'DATETIME2', <DType.TIMESTAMPNTZ: 'TIMESTAMPNTZ'>: 'DATETIME2', <DType.TIMESTAMPTZ: 'TIMESTAMPTZ'>: 'DATETIMEOFFSET', <DType.UTINYINT: 'UTINYINT'>: 'TINYINT', <DType.VARIANT: 'VARIANT'>: 'SQL_VARIANT', <DType.UUID: 'UUID'>: 'UNIQUEIDENTIFIER'}
TRANSFORMS =
{<class 'sqlglot.expressions.query.JSONPathKey'>: <function <lambda>>, <class 'sqlglot.expressions.query.JSONPathRoot'>: <function <lambda>>, <class 'sqlglot.expressions.query.JSONPathSubscript'>: <function <lambda>>, <class 'sqlglot.expressions.core.Adjacent'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.AllowedValuesProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.AnalyzeColumns'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.AnalyzeWith'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.array.ArrayContainedBy'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.array.ArrayContainsAll'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.array.ArrayOverlaps'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.AssumeColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.AutoRefreshProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.BackupProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.BinaryColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.CaseSpecificColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.CalledOnNullInputProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.math.Ceil'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.constraints.CharacterSetColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.CharacterSetProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.ClusteredColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.CollateColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.CommentColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.functions.ConnectByRoot'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.string.ConvertToCharset'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.CopyGrantsProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.CredentialsProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.functions.CurrentCatalog'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.functions.SessionUser'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.DateFormatColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.DefaultColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.ApiProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.ApplicationProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.CatalogProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.ComputeProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.DatabaseProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.DynamicProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.EmptyProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.EncodeColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.EndStatement'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.EnviromentProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.HandlerProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.ParameterStyleProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.EphemeralColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.ExcludeColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.ExecuteAsProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.Except'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.ExternalProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.math.Floor'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.Get'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.GlobalProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.HeapProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.HybridProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.IcebergProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.InheritsProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.InlineLengthColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.InputModelProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.Intersect'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.datatypes.IntervalSpan'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.functions.Int64'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.json.JSONBContainsAnyTopKeys'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.json.JSONBContainsAllTopKeys'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.json.JSONBContainsTopKey'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.json.JSONBDeleteAtPath'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.json.JSONBPathExists'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.json.JSONObject'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.json.JSONObjectAgg'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.LanguageProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.LocationProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.LogProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.MaskingProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.MaterializedProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.functions.NetFunc'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.NetworkProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.NonClusteredColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.NoPrimaryIndexProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.NotForReplicationColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.OnCommitProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.OnProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.OnUpdateColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.core.Operator'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.OutputModelProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.core.ExtendsLeft'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.core.ExtendsRight'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.PathColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.PartitionedByBucket'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.PartitionByTruncate'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.core.PivotAny'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.array.PositionalColumn'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.ProjectionPolicyColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.InvisibleColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.ZeroFillColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.Put'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.RemoteWithConnectionModelProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.RowAccessProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.core.SafeFunc'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.SampleProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.SecureProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.SecurityIntegrationProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.SetConfigProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.SetProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.SettingsProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.SharingProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.SqlReadWriteProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.SqlSecurityProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.StabilityProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.Stream'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.StreamingTableProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.StrictProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.ddl.SwapTable'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.TableColumn'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.Tags'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.TemporaryProperty'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.constraints.TitleColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.array.ToMap'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.ToTableProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.TransformModelProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.TransientProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.VirtualProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.ddl.TriggerExecute'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.Union'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.UnloggedProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.UsingTemplateProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.UsingData'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.UppercaseColumnConstraint'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.temporal.UtcDate'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.temporal.UtcTime'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.temporal.UtcTimestamp'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.query.Variadic'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.array.VarMap'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.ViewAttributeProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.VolatileProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.WithJournalTableProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.WithProcedureOptions'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.WithSchemaBindingProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.constraints.WithOperator'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.properties.ForceProperty'>: <function Generator.<lambda>>, <class 'sqlglot.expressions.aggregate.AnyValue'>: <function any_value_to_max_sql>, <class 'sqlglot.expressions.math.Atan2'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.array.ArrayToString'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.constraints.AutoIncrementColumnConstraint'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.string.Chr'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.temporal.DateAdd'>: <function date_delta_sql.<locals>._delta_sql>, <class 'sqlglot.expressions.query.CTE'>: <function preprocess.<locals>._to_sql>, <class 'sqlglot.expressions.temporal.CurrentDate'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.temporal.CurrentTimestamp'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.temporal.CurrentTimestampLTZ'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.temporal.DateStrToDate'>: <function datestrtodate_sql>, <class 'sqlglot.expressions.temporal.Day'>: <function remove_ts_or_ds_to_date.<locals>.func>, <class 'sqlglot.expressions.constraints.GeneratedAsIdentityColumnConstraint'>: <function generatedasidentitycolumnconstraint_sql>, <class 'sqlglot.expressions.aggregate.GroupConcat'>: <function _string_agg_sql>, <class 'sqlglot.expressions.functions.If'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.json.JSONExtract'>: <function _json_extract_sql>, <class 'sqlglot.expressions.json.JSONExtractScalar'>: <function _json_extract_sql>, <class 'sqlglot.expressions.temporal.LastDay'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.math.Ln'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.aggregate.Max'>: <function max_or_greatest>, <class 'sqlglot.expressions.string.MD5'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.aggregate.Min'>: <function min_or_least>, <class 'sqlglot.expressions.temporal.Month'>: <function remove_ts_or_ds_to_date.<locals>.func>, <class 'sqlglot.expressions.string.NumberToStr'>: <function _format_sql>, <class 'sqlglot.expressions.string.Repeat'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.functions.CurrentSchema'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.query.Select'>: <function preprocess.<locals>._to_sql>, <class 'sqlglot.expressions.aggregate.Stddev'>: <function rename_func.<locals>.<lambda>>, <class 'sqlglot.expressions.string.StrPosition'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.query.Subquery'>: <function preprocess.<locals>._to_sql>, <class 'sqlglot.expressions.string.SHA'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.string.SHA1Digest'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.string.SHA2'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.temporal.TimeStrToTime'>: <function _timestrtotime_sql>, <class 'sqlglot.expressions.temporal.TimeToStr'>: <function _format_sql>, <class 'sqlglot.expressions.temporal.TimestampAdd'>: <function date_delta_sql.<locals>._delta_sql>, <class 'sqlglot.expressions.string.Trim'>: <function trim_sql>, <class 'sqlglot.expressions.temporal.TsOrDsAdd'>: <function date_delta_sql.<locals>._delta_sql>, <class 'sqlglot.expressions.temporal.TsOrDsDiff'>: <function date_delta_sql.<locals>._delta_sql>, <class 'sqlglot.expressions.temporal.TimestampTrunc'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.math.Trunc'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.functions.Uuid'>: <function TSQLGenerator.<lambda>>, <class 'sqlglot.expressions.temporal.Year'>: <function remove_ts_or_ds_to_date.<locals>.func>, <class 'sqlglot.expressions.temporal.DateFromParts'>: <function rename_func.<locals>.<lambda>>}
PROPERTIES_LOCATION =
{<class 'sqlglot.expressions.properties.AllowedValuesProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.AlgorithmProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.ApiProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.ApplicationProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.AutoIncrementProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.AutoRefreshProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.BackupProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.BlockCompressionProperty'>: <PropertiesLocation.POST_NAME: 'POST_NAME'>, <class 'sqlglot.expressions.properties.CalledOnNullInputProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.CatalogProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.CharacterSetProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.ChecksumProperty'>: <PropertiesLocation.POST_NAME: 'POST_NAME'>, <class 'sqlglot.expressions.properties.CollateProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.ComputeProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.CopyGrantsProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.query.Cluster'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.ClusteredByProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.ClusterProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.DistributedByProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.DuplicateKeyProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.DataBlocksizeProperty'>: <PropertiesLocation.POST_NAME: 'POST_NAME'>, <class 'sqlglot.expressions.properties.DatabaseProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.DataDeletionProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.DefinerProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.DictRange'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.DictProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.DynamicProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.DistKeyProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.DistStyleProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.EmptyProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.EncodeProperty'>: <PropertiesLocation.POST_EXPRESSION: 'POST_EXPRESSION'>, <class 'sqlglot.expressions.properties.EngineProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.EnviromentProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.HandlerProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.ParameterStyleProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.ExecuteAsProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.ExternalProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.FallbackProperty'>: <PropertiesLocation.POST_NAME: 'POST_NAME'>, <class 'sqlglot.expressions.properties.FileFormatProperty'>: <PropertiesLocation.POST_WITH: 'POST_WITH'>, <class 'sqlglot.expressions.properties.FreespaceProperty'>: <PropertiesLocation.POST_NAME: 'POST_NAME'>, <class 'sqlglot.expressions.properties.GlobalProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.HeapProperty'>: <PropertiesLocation.POST_WITH: 'POST_WITH'>, <class 'sqlglot.expressions.properties.HybridProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.InheritsProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.IcebergProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.IncludeProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.InputModelProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.IsolatedLoadingProperty'>: <PropertiesLocation.POST_NAME: 'POST_NAME'>, <class 'sqlglot.expressions.properties.JournalProperty'>: <PropertiesLocation.POST_NAME: 'POST_NAME'>, <class 'sqlglot.expressions.properties.LanguageProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.LikeProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.LocationProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.LockProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.LockingProperty'>: <PropertiesLocation.POST_ALIAS: 'POST_ALIAS'>, <class 'sqlglot.expressions.properties.LogProperty'>: <PropertiesLocation.POST_NAME: 'POST_NAME'>, <class 'sqlglot.expressions.properties.MaskingProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.MaterializedProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.MergeBlockRatioProperty'>: <PropertiesLocation.POST_NAME: 'POST_NAME'>, <class 'sqlglot.expressions.properties.ModuleProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.NetworkProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.NoPrimaryIndexProperty'>: <PropertiesLocation.POST_EXPRESSION: 'POST_EXPRESSION'>, <class 'sqlglot.expressions.properties.OnProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.OnCommitProperty'>: <PropertiesLocation.POST_EXPRESSION: 'POST_EXPRESSION'>, <class 'sqlglot.expressions.query.Order'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.OutputModelProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.PartitionedByProperty'>: <PropertiesLocation.POST_WITH: 'POST_WITH'>, <class 'sqlglot.expressions.properties.PartitionedOfProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.constraints.PrimaryKey'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.Property'>: <PropertiesLocation.POST_WITH: 'POST_WITH'>, <class 'sqlglot.expressions.properties.RefreshTriggerProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.RemoteWithConnectionModelProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.ReturnsProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.RollupProperty'>: <PropertiesLocation.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.properties.RowAccessProperty'>: <PropertiesLocation.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.properties.RowFormatProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.RowFormatDelimitedProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.RowFormatSerdeProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.SampleProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.SchemaCommentProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.SecureProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.SecurityIntegrationProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.SerdeProperties'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.ddl.Set'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.SettingsProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.SetProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.SetConfigProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.SharingProperty'>: <PropertiesLocation.POST_EXPRESSION: 'POST_EXPRESSION'>, <class 'sqlglot.expressions.ddl.SequenceProperties'>: <PropertiesLocation.POST_EXPRESSION: 'POST_EXPRESSION'>, <class 'sqlglot.expressions.ddl.TriggerProperties'>: <PropertiesLocation.POST_EXPRESSION: 'POST_EXPRESSION'>, <class 'sqlglot.expressions.properties.SortKeyProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.SqlReadWriteProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.SqlSecurityProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.StabilityProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.StorageHandlerProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.StreamingTableProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.StrictProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.Tags'>: <PropertiesLocation.POST_WITH: 'POST_WITH'>, <class 'sqlglot.expressions.properties.TemporaryProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.ToTableProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.TransientProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.TransformModelProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.ddl.MergeTreeTTL'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.UnloggedProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.UsingProperty'>: <PropertiesLocation.POST_EXPRESSION: 'POST_EXPRESSION'>, <class 'sqlglot.expressions.properties.UsingTemplateProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.ViewAttributeProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.VirtualProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>, <class 'sqlglot.expressions.properties.VolatileProperty'>: <PropertiesLocation.UNSUPPORTED: 'UNSUPPORTED'>, <class 'sqlglot.expressions.properties.WithDataProperty'>: <PropertiesLocation.POST_EXPRESSION: 'POST_EXPRESSION'>, <class 'sqlglot.expressions.properties.WithJournalTableProperty'>: <PropertiesLocation.POST_NAME: 'POST_NAME'>, <class 'sqlglot.expressions.properties.WithProcedureOptions'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.WithSchemaBindingProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.WithSystemVersioningProperty'>: <PropertiesLocation.POST_SCHEMA: 'POST_SCHEMA'>, <class 'sqlglot.expressions.properties.ForceProperty'>: <PropertiesLocation.POST_CREATE: 'POST_CREATE'>}
277 def select_sql(self, expression: exp.Select) -> str: 278 limit = expression.args.get("limit") 279 offset = expression.args.get("offset") 280 281 if isinstance(limit, exp.Fetch) and not offset: 282 # Dialects like Oracle can FETCH directly from a row set but 283 # T-SQL requires an ORDER BY + OFFSET clause in order to FETCH 284 offset = exp.Offset(expression=exp.Literal.number(0)) 285 expression.set("offset", offset) 286 287 if offset: 288 if not expression.args.get("order"): 289 # ORDER BY is required in order to use OFFSET in a query, so we use 290 # a noop order by, since we don't really care about the order. 291 # See: https://www.microsoftpressstore.com/articles/article.aspx?p=2314819 292 expression.order_by(exp.select(exp.null()).subquery(), copy=False) 293 294 if isinstance(limit, exp.Limit): 295 # TOP and OFFSET can't be combined, we need use FETCH instead of TOP 296 # we replace here because otherwise TOP would be generated in select_sql 297 limit.replace(exp.Fetch(direction="FIRST", count=limit.expression)) 298 299 return super().select_sql(expression)
305 def queryoption_sql(self, expression: exp.QueryOption) -> str: 306 option = self.sql(expression, "this") 307 value = self.sql(expression, "expression") 308 if value: 309 optional_equal_sign = "= " if option in OPTIONS_THAT_REQUIRE_EQUAL else "" 310 return f"{option} {optional_equal_sign}{value}" 311 return option
313 def lateral_op(self, expression: exp.Lateral) -> str: 314 cross_apply = expression.args.get("cross_apply") 315 if cross_apply is True: 316 return "CROSS APPLY" 317 if cross_apply is False: 318 return "OUTER APPLY" 319 320 # TODO: perhaps we can check if the parent is a Join and transpile it appropriately 321 self.unsupported("LATERAL clause is not supported.") 322 return "LATERAL"
324 def splitpart_sql(self, expression: exp.SplitPart) -> str: 325 this = expression.this 326 split_count = len(this.name.split(".")) 327 delimiter = expression.args.get("delimiter") 328 part_index = expression.args.get("part_index") 329 330 if ( 331 not all(isinstance(arg, exp.Literal) for arg in (this, delimiter, part_index)) 332 or (delimiter and delimiter.name != ".") 333 or not part_index 334 or split_count > 4 335 ): 336 self.unsupported( 337 "SPLIT_PART can be transpiled to PARSENAME only for '.' delimiter and literal values" 338 ) 339 return "" 340 341 return self.func( 342 "PARSENAME", this, exp.Literal.number(split_count + 1 - part_index.to_py()) 343 )
351 def timefromparts_sql(self, expression: exp.TimeFromParts) -> str: 352 nano = expression.args.get("nano") 353 if nano is not None: 354 nano.pop() 355 self.unsupported("Specifying nanoseconds is not supported in TIMEFROMPARTS.") 356 357 if expression.args.get("fractions") is None: 358 expression.set("fractions", exp.Literal.number(0)) 359 if expression.args.get("precision") is None: 360 expression.set("precision", exp.Literal.number(0)) 361 362 return rename_func("TIMEFROMPARTS")(self, expression)
def
timestampfromparts_sql(self, expression: sqlglot.expressions.temporal.TimestampFromParts) -> str:
364 def timestampfromparts_sql(self, expression: exp.TimestampFromParts) -> str: 365 zone = expression.args.get("zone") 366 if zone is not None: 367 zone.pop() 368 self.unsupported("Time zone is not supported in DATETIMEFROMPARTS.") 369 370 nano = expression.args.get("nano") 371 if nano is not None: 372 nano.pop() 373 self.unsupported("Specifying nanoseconds is not supported in DATETIMEFROMPARTS.") 374 375 if expression.args.get("milli") is None: 376 expression.set("milli", exp.Literal.number(0)) 377 378 return rename_func("DATETIMEFROMPARTS")(self, expression)
380 def setitem_sql(self, expression: exp.SetItem) -> str: 381 this = expression.this 382 if isinstance(this, exp.EQ) and not isinstance(this.left, exp.Parameter): 383 # T-SQL does not use '=' in SET command, except when the LHS is a variable. 384 return f"{self.sql(this.left)} {self.sql(this.right)}" 385 386 return super().setitem_sql(expression)
def
createable_sql( self, expression: sqlglot.expressions.ddl.Create, locations: collections.defaultdict) -> str:
402 def createable_sql(self, expression: exp.Create, locations: defaultdict) -> str: 403 sql = self.sql(expression, "this") 404 properties = expression.args.get("properties") 405 406 start = self._identifier_start 407 if ( 408 not sql.startswith("#") 409 and not sql.startswith(f"{start}#") 410 and any( 411 isinstance(prop, exp.TemporaryProperty) 412 for prop in (properties.expressions if properties else []) 413 ) 414 ): 415 sql = f"{start}#{sql[len(start) :]}" if sql.startswith(start) else f"#{sql}" 416 417 return sql
419 def create_sql(self, expression: exp.Create) -> str: 420 kind = expression.kind 421 exists = expression.args.get("exists") 422 expression.set("exists", None) 423 424 like_property = expression.find(exp.LikeProperty) 425 if like_property: 426 ctas_expression = like_property.this 427 else: 428 ctas_expression = expression.expression 429 430 if kind == "VIEW": 431 expression.this.set("catalog", None) 432 with_ = expression.args.get("with_") 433 if ctas_expression and with_: 434 # We've already preprocessed the Create expression to bubble up any nested CTEs, 435 # but CREATE VIEW actually requires the WITH clause to come after it so we need 436 # to amend the AST by moving the CTEs to the CREATE VIEW statement's query. 437 ctas_expression.set("with_", with_.pop()) 438 elif ( 439 kind == "FUNCTION" 440 and isinstance(ctas_expression, exp.Return) 441 and isinstance(body := ctas_expression.this.unnest(), exp.Query) 442 and (with_ := expression.args.get("with_")) 443 ): 444 # Similar to the VIEW branch, the table-valued functions require the WITH clause 445 # to stay inside the RETURN body, so we move back any CTEs that were bubbled up. 446 body.set("with_", with_.pop()) 447 448 table = expression.find(exp.Table) 449 450 # Convert CTAS statement to SELECT .. INTO .. 451 if kind == "TABLE" and ctas_expression: 452 if isinstance(ctas_expression, exp.UNWRAPPED_QUERIES): 453 ctas_expression = ctas_expression.subquery() 454 455 properties = expression.args.get("properties") or exp.Properties() 456 is_temp = any(isinstance(p, exp.TemporaryProperty) for p in properties.expressions) 457 458 select_into = exp.select("*").from_(exp.alias_(ctas_expression, "temp", table=True)) 459 select_into.set("into", exp.Into(this=table, temporary=is_temp)) 460 461 if like_property: 462 select_into.limit(0, copy=False) 463 464 sql = self.sql(select_into) 465 else: 466 sql = super().create_sql(expression) 467 468 if exists: 469 identifier = self.sql(exp.Literal.string(exp.table_name(table) if table else "")) 470 sql_with_ctes = self.prepend_ctes(expression, sql) 471 sql_literal = self.sql(exp.Literal.string(sql_with_ctes)) 472 if kind == "SCHEMA": 473 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = {identifier}) EXEC({sql_literal})""" 474 elif kind == "TABLE": 475 assert table 476 where = exp.and_( 477 exp.column("TABLE_NAME").eq(table.name), 478 exp.column("TABLE_SCHEMA").eq(table.db) if table.db else None, 479 exp.column("TABLE_CATALOG").eq(table.catalog) if table.catalog else None, 480 ) 481 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE {where}) EXEC({sql_literal})""" 482 elif kind == "INDEX": 483 index = self.sql(exp.Literal.string(expression.this.text("this"))) 484 return f"""IF NOT EXISTS (SELECT * FROM sys.indexes WHERE object_id = object_id({identifier}) AND name = {index}) EXEC({sql_literal})""" 485 elif expression.args.get("replace"): 486 sql = sql.replace("CREATE OR REPLACE ", "CREATE OR ALTER ", 1) 487 488 return self.prepend_ctes(expression, sql)
@generator.unsupported_args('unlogged', 'expressions')
def
into_sql(self, expression: sqlglot.expressions.query.Into) -> str:
490 @generator.unsupported_args("unlogged", "expressions") 491 def into_sql(self, expression: exp.Into) -> str: 492 if expression.args.get("temporary"): 493 # If the Into expression has a temporary property, push this down to the Identifier 494 table = expression.find(exp.Table) 495 if table and isinstance(table.this, exp.Identifier): 496 table.this.set("temporary", True) 497 498 return f"{self.seg('INTO')} {self.sql(expression, 'this')}"
511 def version_sql(self, expression: exp.Version) -> str: 512 name = "SYSTEM_TIME" if expression.name == "TIMESTAMP" else expression.name 513 this = f"FOR {name}" 514 expr = expression.expression 515 kind = expression.text("kind") 516 if kind in ("FROM", "BETWEEN"): 517 args = expr.expressions 518 sep = "TO" if kind == "FROM" else "AND" 519 expr_sql = f"{self.sql(seq_get(args, 0))} {sep} {self.sql(seq_get(args, 1))}" 520 else: 521 expr_sql = self.sql(expr) 522 523 expr_sql = f" {expr_sql}" if expr_sql else "" 524 return f"{this} {kind}{expr_sql}"
543 def commit_sql(self, expression: exp.Commit) -> str: 544 this = self.sql(expression, "this") 545 this = f" {this}" if this else "" 546 durability = expression.args.get("durability") 547 durability = ( 548 f" WITH (DELAYED_DURABILITY = {'ON' if durability else 'OFF'})" 549 if durability is not None 550 else "" 551 ) 552 return f"COMMIT TRANSACTION{this}{durability}"
559 def identifier_sql(self, expression: exp.Identifier) -> str: 560 identifier = super().identifier_sql(expression) 561 562 if expression.args.get("global_"): 563 prefix = "##" 564 elif expression.args.get("temporary"): 565 prefix = "#" 566 else: 567 return identifier 568 569 start = self._identifier_start 570 if expression.quoted and identifier.startswith(start): 571 return f"{start}{prefix}{identifier[len(start) :]}" 572 573 return f"{prefix}{identifier}"
623 def columndef_sql(self, expression: exp.ColumnDef, sep: str = " ") -> str: 624 this = super().columndef_sql(expression, sep) 625 default = self.sql(expression, "default") 626 default = f" = {default}" if default else "" 627 output = self.sql(expression, "output") 628 output = f" {output}" if output else "" 629 return f"{this}{default}{output}"
635 def storedprocedure_sql(self, expression: exp.StoredProcedure) -> str: 636 this = self.sql(expression, "this") 637 expressions = self.expressions(expression) 638 expressions = ( 639 self.wrap(expressions) if expression.args.get("wrapped") else f" {expressions}" 640 ) 641 return f"{this}{expressions}" if expressions.strip() != "" else this
643 def ifblock_sql(self, expression: exp.IfBlock) -> str: 644 this = self.sql(expression, "this") 645 true = self.sql(expression, "true") 646 true = f" {true}" if true else " " 647 false = self.sql(expression, "false") 648 false = f"; ELSE BEGIN {false}" if false else "" 649 return f"IF {this} BEGIN{true}{false}"
657 def execute_sql(self, expression: exp.Execute) -> str: 658 this = self.sql(expression, "this") 659 expressions = self.expressions(expression) 660 expressions = f" {expressions}" if expressions else "" 661 return_status = self.sql(expression, "return_status") 662 return_status = f"{return_status} = " if return_status else "" 663 return f"EXECUTE {return_status}{this}{expressions}"
Inherited Members
- sqlglot.generator.Generator
- Generator
- WINDOW_FUNCS_WITH_NULL_ORDERING
- IGNORE_NULLS_IN_FUNC
- IGNORE_NULLS_BEFORE_ORDER
- LOCKING_READS_SUPPORTED
- WRAP_DERIVED_VALUES
- CREATE_FUNCTION_RETURN_AS
- MATCHED_BY_SOURCE
- SUPPORTS_MERGE_WHERE
- SINGLE_STRING_INTERVAL
- INTERVAL_ALLOWS_PLURAL_FORM
- AUTO_REFRESH_BARE_INTERVALS
- LIMIT_ONLY_LITERALS
- RENAME_TABLE_WITH_DB
- GROUPINGS_SEP
- SUPPORTS_GROUPING_SETS_AS_SUFFIX
- INDEX_ON
- INOUT_SEPARATOR
- JOIN_HINTS
- DIRECTED_JOINS
- TABLE_HINTS
- QUERY_HINT_SEP
- IS_BOOL_ALLOWED
- DUPLICATE_KEY_UPDATE_WITH_SET
- EXTRACT_ALLOWS_QUOTES
- TZ_TO_WITH_TIME_ZONE
- VALUES_AS_TABLE
- UNNEST_WITH_ORDINALITY
- SEMI_ANTI_JOIN_WITH_SIDE
- SUPPORTS_TABLE_COPY
- TABLESAMPLE_REQUIRES_PARENS
- TABLESAMPLE_SIZE_IS_ROWS
- TABLESAMPLE_KEYWORDS
- TABLESAMPLE_WITH_METHOD
- HISTORICAL_DATA_POST_ALIAS
- COLLATE_IS_FUNC
- DATA_TYPE_SPECIFIERS_ALLOWED
- LAST_DAY_SUPPORTS_DATE_PART
- SUPPORTS_TABLE_ALIAS_COLUMNS
- SUPPORTS_NAMED_CTE_COLUMNS
- UNPIVOT_ALIASES_ARE_IDENTIFIERS
- PIVOT_ALIAS_WITH_AS
- JSON_KEY_VALUE_PAIR_SEP
- INSERT_OVERWRITE
- SUPPORTS_UNLOGGED_TABLES
- SUPPORTS_CREATE_TABLE_LIKE
- SUPPORTS_MODIFY_COLUMN
- SUPPORTS_CHANGE_COLUMN
- SUPPORTS_ALTER_COLUMN_IF_EXISTS
- LIKE_PROPERTY_INSIDE_SCHEMA
- MULTI_ARG_DISTINCT
- JSON_TYPE_REQUIRED_FOR_EXTRACTION
- JSON_PATH_SINGLE_QUOTE_ESCAPE
- JSON_PATH_KEY_QUOTED_FORCES_BRACKETS
- CAN_IMPLEMENT_ARRAY_ANY
- SUPPORTS_WINDOW_EXCLUDE
- COPY_PARAMS_ARE_WRAPPED
- COPY_HAS_INTO_KEYWORD
- UNICODE_SUBSTITUTE
- STAR_EXCEPT
- HEX_FUNC
- WITH_PROPERTIES_PREFIX
- QUOTE_JSON_PATH
- PAD_FILL_PATTERN_IS_REQUIRED
- SUPPORTS_EXPLODING_PROJECTIONS
- ARRAY_CONCAT_IS_VAR_LEN
- SUPPORTS_CONVERT_TIMEZONE
- SUPPORTS_MEDIAN
- SUPPORTS_UNIX_SECONDS
- NORMALIZE_EXTRACT_DATE_PARTS
- ARRAY_SIZE_NAME
- ARRAY_SIZE_DIM_REQUIRED
- SUPPORTS_BETWEEN_FLAGS
- SUPPORTS_LIKE_QUANTIFIERS
- MATCH_AGAINST_TABLE_PREFIX
- SET_ASSIGNMENT_REQUIRES_VARIABLE_KEYWORD
- DECLARE_DEFAULT_ASSIGNMENT
- UPDATE_STATEMENT_SUPPORTS_FROM
- STAR_EXCLUDE_REQUIRES_DERIVED_TABLE
- SUPPORTS_DROP_ALTER_ICEBERG_PROPERTY
- UNSUPPORTED_TYPES
- TYPE_PARAM_SETTINGS
- TIME_PART_SINGULARS
- TOKEN_MAPPING
- STRUCT_DELIMITER
- PARAMETER_TOKEN
- NAMED_PLACEHOLDER_TOKEN
- EXPRESSION_PRECEDES_PROPERTIES_CREATABLES
- RESERVED_KEYWORDS
- WITH_SEPARATED_COMMENTS
- EXCLUDE_COMMENTS
- UNWRAPPED_INTERVAL_VALUES
- PARAMETERIZABLE_TEXT_TYPES
- RESPECT_IGNORE_NULLS_UNSUPPORTED_EXPRESSIONS
- MOD_OPERATOR
- MOD_PAREN_PARENT_TYPES
- SAFE_JSON_PATH_KEY_RE
- SENTINEL_LINE_BREAK
- pretty
- identify
- normalize
- pad
- unsupported_level
- max_unsupported
- leading_comma
- max_text_width
- comments
- dialect
- normalize_functions
- unsupported_messages
- generate
- preprocess
- unsupported
- sep
- seg
- sanitize_comment
- maybe_comment
- wrap
- no_identify
- normalize_func
- indent
- sql
- uncache_sql
- cache_sql
- characterset_sql
- column_parts
- column_sql
- pseudocolumn_sql
- columnposition_sql
- columnconstraint_sql
- computedcolumnconstraint_sql
- autoincrementcolumnconstraint_sql
- compresscolumnconstraint_sql
- generatedasidentitycolumnconstraint_sql
- generatedasrowcolumnconstraint_sql
- periodforsystemtimeconstraint_sql
- notnullcolumnconstraint_sql
- primarykeycolumnconstraint_sql
- uniquecolumnconstraint_sql
- inoutcolumnconstraint_sql
- sequenceproperties_sql
- triggerproperties_sql
- triggerreferencing_sql
- triggerevent_sql
- clone_sql
- describe_sql
- heredoc_sql
- prepend_ctes
- with_sql
- cte_sql
- tablealias_sql
- bitstring_sql
- hexstring_sql
- bytestring_sql
- unicodestring_sql
- rawstring_sql
- datatypeparam_sql
- datatype_param_bound_limiter
- datatype_sql
- directory_sql
- delete_sql
- set_operation
- set_operations
- fetch_sql
- limitoptions_sql
- filter_sql
- hint_sql
- indexparameters_sql
- index_sql
- dynamicidentifier_sql
- hex_sql
- lowerhex_sql
- inputoutputformat_sql
- national_sql
- properties_sql
- root_properties
- properties
- with_properties
- locate_properties
- property_name
- property_sql
- uuidproperty_sql
- likeproperty_sql
- fallbackproperty_sql
- journalproperty_sql
- freespaceproperty_sql
- checksumproperty_sql
- mergeblockratioproperty_sql
- moduleproperty_sql
- datablocksizeproperty_sql
- blockcompressionproperty_sql
- isolatedloadingproperty_sql
- partitionboundspec_sql
- partitionedofproperty_sql
- lockingproperty_sql
- withdataproperty_sql
- withsystemversioningproperty_sql
- insert_sql
- introducer_sql
- kill_sql
- pseudotype_sql
- objectidentifier_sql
- onconflict_sql
- rowformatdelimitedproperty_sql
- withtablehint_sql
- indextablehint_sql
- historicaldata_sql
- table_parts
- table_sql
- tablefromrows_sql
- tablesample_sql
- pivot_sql
- tuple_sql
- update_sql
- values_sql
- var_sql
- from_sql
- groupingsets_sql
- rollup_sql
- rollupindex_sql
- rollupproperty_sql
- cube_sql
- group_sql
- having_sql
- connect_sql
- prior_sql
- join_sql
- lambda_sql
- lateral_sql
- limit_sql
- set_sql
- queryband_sql
- pragma_sql
- lock_sql
- literal_sql
- escape_str
- loaddata_sql
- null_sql
- booland_sql
- boolor_sql
- order_sql
- withfill_sql
- cluster_sql
- clusterproperty_sql
- distribute_sql
- sort_sql
- ordered_sql
- matchrecognizemeasure_sql
- matchrecognize_sql
- query_modifiers
- forclause_sql
- offset_limit_modifiers
- after_limit_modifiers
- schema_sql
- schema_columns_sql
- star_sql
- parameter_sql
- sessionparameter_sql
- placeholder_sql
- subquery_sql
- qualify_sql
- unnest_sql
- prewhere_sql
- where_sql
- window_sql
- partition_by_sql
- windowspec_sql
- withingroup_sql
- between_sql
- bracket_offset_expressions
- bracket_sql
- all_sql
- any_sql
- exists_sql
- case_sql
- nextvaluefor_sql
- trim_sql
- convert_concat_args
- concat_sql
- concatws_sql
- check_sql
- foreignkey_sql
- primarykey_sql
- timeserieskey_sql
- if_sql
- matchagainst_sql
- jsonkeyvalue_sql
- jsonpath_sql
- json_path_part
- formatjson_sql
- formatphrase_sql
- jsonarray_sql
- jsonarrayagg_sql
- jsoncolumndef_sql
- jsonschema_sql
- jsontable_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
- pivotalias_sql
- aliases_sql
- atindex_sql
- attimezone_sql
- fromtimezone_sql
- fromiso8601date_sql
- fromiso8601timestamp_sql
- fromiso8601timestampnanos_sql
- add_sql
- and_sql
- or_sql
- xor_sql
- connector_sql
- bitwiseand_sql
- bitwiseleftshift_sql
- bitwisenot_sql
- bitwiseor_sql
- bitwiserightshift_sql
- bitwisexor_sql
- cast_sql
- strtotime_sql
- strtodate_sql
- parsedatetime_sql
- currentdate_sql
- collate_sql
- command_sql
- comment_sql
- mergetreettlaction_sql
- mergetreettl_sql
- altercolumn_sql
- modifycolumn_sql
- alterindex_sql
- alterdiststyle_sql
- altersortkey_sql
- alterrename_sql
- renamecolumn_sql
- alterset_sql
- altersession_sql
- add_column_sql
- droppartition_sql
- dropprimarykey_sql
- addconstraint_sql
- addpartition_sql
- distinct_sql
- ignorenulls_sql
- respectnulls_sql
- havingmax_sql
- intdiv_sql
- div_sql
- safedivide_sql
- overlaps_sql
- distance_sql
- distancend_sql
- dot_sql
- eq_sql
- propertyeq_sql
- escape_sql
- glob_sql
- gt_sql
- gte_sql
- like_sql
- ilike_sql
- match_sql
- similarto_sql
- lt_sql
- lte_sql
- mod_sql
- mul_sql
- neq_sql
- nullsafeeq_sql
- nullsafeneq_sql
- sub_sql
- trycast_sql
- jsoncast_sql
- try_sql
- log_sql
- use_sql
- binary
- ceil_floor
- function_fallback_sql
- func
- format_args
- too_wide
- format_time
- expressions
- op_expressions
- naked_property
- tag_sql
- token_sql
- userdefinedfunction_sql
- macrooverloads_sql
- macrooverload_sql
- joinhint_sql
- kwarg_sql
- when_sql
- whens_sql
- merge_sql
- tochar_sql
- tonumber_sql
- dictproperty_sql
- dictrange_sql
- dictsubproperty_sql
- duplicatekeyproperty_sql
- uniquekeyproperty_sql
- distributedbyproperty_sql
- oncluster_sql
- clusteredbyproperty_sql
- anyvalue_sql
- querytransform_sql
- indexconstraintoption_sql
- checkcolumnconstraint_sql
- indexcolumnconstraint_sql
- nvl2_sql
- nthvalue_sql
- comprehension_sql
- columnprefix_sql
- opclass_sql
- predict_sql
- generateembedding_sql
- generatetext_sql
- generatetable_sql
- generatebool_sql
- generateint_sql
- generatedouble_sql
- mltranslate_sql
- mlforecast_sql
- aiforecast_sql
- featuresattime_sql
- vectorsearch_sql
- forin_sql
- refresh_sql
- toarray_sql
- tsordstotime_sql
- tsordstotimestamp_sql
- tsordstodatetime_sql
- tsordstodate_sql
- unixdate_sql
- lastday_sql
- dateadd_sql
- arrayany_sql
- struct_sql
- partitionrange_sql
- truncatetable_sql
- copyparameter_sql
- credentials_sql
- copy_sql
- semicolon_sql
- datadeletionproperty_sql
- maskingpolicycolumnconstraint_sql
- gapfill_sql
- scoperesolution_sql
- parsejson_sql
- rand_sql
- changes_sql
- pad_sql
- summarize_sql
- explodinggenerateseries_sql
- converttimezone_sql
- json_sql
- jsonvalue_sql
- skipjsoncolumn_sql
- conditionalinsert_sql
- multitableinserts_sql
- oncondition_sql
- jsonextractquote_sql
- jsonexists_sql
- arrayagg_sql
- slice_sql
- apply_sql
- grant_sql
- revoke_sql
- grantprivilege_sql
- grantprincipal_sql
- columns_sql
- overlay_sql
- todouble_sql
- string_sql
- median_sql
- overflowtruncatebehavior_sql
- unixseconds_sql
- arraysize_sql
- attach_sql
- detach_sql
- attachoption_sql
- watermarkcolumnconstraint_sql
- encodeproperty_sql
- includeproperty_sql
- xmlelement_sql
- xmlkeyvalueoption_sql
- partitionbyrangeproperty_sql
- partitionbyrangepropertydynamic_sql
- unpivotcolumns_sql
- analyzesample_sql
- analyzestatistics_sql
- analyzehistogram_sql
- analyzedelete_sql
- analyzelistchainedrows_sql
- analyzevalidate_sql
- analyze_sql
- xmltable_sql
- xmlnamespace_sql
- export_sql
- declare_sql
- declareitem_sql
- recursivewithsearch_sql
- parameterizedagg_sql
- anonymousaggfunc_sql
- combinedaggfunc_sql
- combinedparameterizedagg_sql
- show_sql
- install_sql
- get_put_sql
- translatecharacters_sql
- decodecase_sql
- semanticview_sql
- getextract_sql
- datefromunixdate_sql
- space_sql
- buildproperty_sql
- refreshtriggerproperty_sql
- modelattribute_sql
- directorystage_sql
- uuid_sql
- initcap_sql
- localtime_sql
- localtimestamp_sql
- weekstart_name
- weekstart_sql
- chr_sql
- block_sql
- functionspecification_sql
- casestatement_sql
- loopblock_sql
- repeatblock_sql
- leave_sql
- iterate_sql
- altermodifysqlsecurity_sql
- usingproperty_sql
- renameindex_sql