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 = self.func("JSON_QUERY", expression.this, expression.expression) 107 json_value = self.func("JSON_VALUE", expression.this, expression.expression) 108 return self.func("ISNULL", json_query, json_value) 109 110 111def _timestrtotime_sql(self: TSQLGenerator, expression: exp.TimeStrToTime): 112 sql = timestrtotime_sql(self, expression) 113 if expression.args.get("zone"): 114 # If there is a timezone, produce an expression like: 115 # CAST('2020-01-01 12:13:14-08:00' AS DATETIMEOFFSET) AT TIME ZONE 'UTC' 116 # If you dont have AT TIME ZONE 'UTC', wrapping that expression in another cast back to DATETIME2 just drops the timezone information 117 return self.sql(exp.AtTimeZone(this=sql, zone=exp.Literal.string("UTC"))) 118 return sql 119 120 121class TSQLGenerator(generator.Generator): 122 SELECT_KINDS: tuple[str, ...] = () 123 TRY_SUPPORTED = False 124 SUPPORTS_UESCAPE = False 125 SUPPORTS_DECODE_CASE = False 126 127 AFTER_HAVING_MODIFIER_TRANSFORMS = generator.AFTER_HAVING_MODIFIER_TRANSFORMS 128 129 LIMIT_IS_TOP = True 130 QUERY_HINTS = False 131 RETURNING_END = False 132 NVL2_SUPPORTED = False 133 ALTER_TABLE_INCLUDE_COLUMN_KEYWORD = False 134 LIMIT_FETCH = "FETCH" 135 COMPUTED_COLUMN_WITH_TYPE = False 136 CTE_RECURSIVE_KEYWORD_REQUIRED = False 137 ENSURE_BOOLS = True 138 NULL_ORDERING_SUPPORTED: bool | None = None 139 SUPPORTS_SINGLE_ARG_CONCAT = False 140 TABLESAMPLE_SEED_KEYWORD = "REPEATABLE" 141 SUPPORTS_SELECT_INTO = True 142 JSON_PATH_BRACKETED_KEY_SUPPORTED = False 143 SUPPORTS_TO_NUMBER = False 144 SET_OP_MODIFIERS = False 145 COPY_PARAMS_EQ_REQUIRED = True 146 PARSE_JSON_NAME: str | None = None 147 EXCEPT_INTERSECT_SUPPORT_ALL_CLAUSE = False 148 ALTER_SET_WRAPPED = True 149 ALTER_SET_TYPE = "" 150 151 EXPRESSIONS_WITHOUT_NESTED_CTES = { 152 exp.Create, 153 exp.Delete, 154 exp.Insert, 155 exp.Intersect, 156 exp.Except, 157 exp.Merge, 158 exp.Select, 159 exp.Subquery, 160 exp.Union, 161 exp.Update, 162 } 163 164 SUPPORTED_JSON_PATH_PARTS = { 165 exp.JSONPathKey, 166 exp.JSONPathRoot, 167 exp.JSONPathSubscript, 168 } 169 170 TYPE_MAPPING = { 171 **{ 172 k: v 173 for k, v in generator.Generator.TYPE_MAPPING.items() 174 if k not in (exp.DType.NCHAR, exp.DType.NVARCHAR) 175 }, 176 exp.DType.BOOLEAN: "BIT", 177 exp.DType.DATETIME2: "DATETIME2", 178 exp.DType.DECIMAL: "NUMERIC", 179 exp.DType.DOUBLE: "FLOAT", 180 exp.DType.INT: "INTEGER", 181 exp.DType.ROWVERSION: "ROWVERSION", 182 exp.DType.TEXT: "VARCHAR(MAX)", 183 exp.DType.TIMESTAMP: "DATETIME2", 184 exp.DType.TIMESTAMPNTZ: "DATETIME2", 185 exp.DType.TIMESTAMPTZ: "DATETIMEOFFSET", 186 exp.DType.SMALLDATETIME: "SMALLDATETIME", 187 exp.DType.UTINYINT: "TINYINT", 188 exp.DType.VARIANT: "SQL_VARIANT", 189 exp.DType.UUID: "UNIQUEIDENTIFIER", 190 } 191 192 TRANSFORMS = { 193 **{k: v for k, v in generator.Generator.TRANSFORMS.items() if k != exp.ReturnsProperty}, 194 exp.AnyValue: any_value_to_max_sql, 195 exp.Atan2: rename_func("ATN2"), 196 exp.ArrayToString: rename_func("STRING_AGG"), 197 exp.AutoIncrementColumnConstraint: lambda *_: "IDENTITY", 198 exp.Ceil: rename_func("CEILING"), 199 exp.Chr: rename_func("CHAR"), 200 exp.DateAdd: date_delta_sql("DATEADD"), 201 exp.CTE: transforms.preprocess([qualify_derived_table_outputs]), 202 exp.CurrentDate: rename_func("GETDATE"), 203 exp.CurrentTimestamp: rename_func("GETDATE"), 204 exp.CurrentTimestampLTZ: rename_func("SYSDATETIMEOFFSET"), 205 exp.DateStrToDate: datestrtodate_sql, 206 exp.Day: remove_ts_or_ds_to_date(), 207 exp.GeneratedAsIdentityColumnConstraint: generatedasidentitycolumnconstraint_sql, 208 exp.GroupConcat: _string_agg_sql, 209 exp.If: rename_func("IIF"), 210 exp.JSONExtract: _json_extract_sql, 211 exp.JSONExtractScalar: _json_extract_sql, 212 exp.LastDay: lambda self, e: self.func("EOMONTH", e.this), 213 exp.Ln: rename_func("LOG"), 214 exp.Max: max_or_greatest, 215 exp.MD5: lambda self, e: self.func("HASHBYTES", exp.Literal.string("MD5"), e.this), 216 exp.Min: min_or_least, 217 exp.Month: remove_ts_or_ds_to_date(), 218 exp.NumberToStr: _format_sql, 219 exp.Repeat: rename_func("REPLICATE"), 220 exp.CurrentSchema: rename_func("SCHEMA_NAME"), 221 exp.Select: transforms.preprocess( 222 [ 223 transforms.eliminate_distinct_on, 224 transforms.eliminate_semi_and_anti_joins, 225 transforms.eliminate_qualify, 226 transforms.unnest_generate_date_array_using_recursive_cte, 227 ] 228 ), 229 exp.Stddev: rename_func("STDEV"), 230 exp.StrPosition: lambda self, e: strposition_sql( 231 self, e, func_name="CHARINDEX", supports_position=True 232 ), 233 exp.Subquery: transforms.preprocess([qualify_derived_table_outputs]), 234 exp.SHA: lambda self, e: self.func("HASHBYTES", exp.Literal.string("SHA1"), e.this), 235 exp.SHA1Digest: lambda self, e: self.func("HASHBYTES", exp.Literal.string("SHA1"), e.this), 236 exp.SHA2: lambda self, e: self.func( 237 "HASHBYTES", exp.Literal.string(f"SHA2_{e.args.get('length', 256)}"), e.this 238 ), 239 exp.TemporaryProperty: lambda self, e: "", 240 exp.TimeStrToTime: _timestrtotime_sql, 241 exp.TimeToStr: _format_sql, 242 exp.TimestampAdd: date_delta_sql("DATEADD"), 243 exp.Trim: trim_sql, 244 exp.TsOrDsAdd: date_delta_sql("DATEADD", cast=True), 245 exp.TsOrDsDiff: date_delta_sql("DATEDIFF"), 246 exp.TimestampTrunc: lambda self, e: self.func("DATETRUNC", e.unit, e.this), 247 exp.Trunc: lambda self, e: self.func( 248 "ROUND", 249 e.this, 250 e.args.get("decimals") or exp.Literal.number(0), 251 exp.Literal.number(1), 252 ), 253 exp.Uuid: lambda *_: "NEWID()", 254 exp.Year: remove_ts_or_ds_to_date(), 255 exp.DateFromParts: rename_func("DATEFROMPARTS"), 256 } 257 258 PROPERTIES_LOCATION = { 259 **generator.Generator.PROPERTIES_LOCATION, 260 exp.VolatileProperty: exp.Properties.Location.UNSUPPORTED, 261 } 262 263 def scope_resolution(self, rhs: str, scope_name: str) -> str: 264 return f"{scope_name}::{rhs}" 265 266 def select_sql(self, expression: exp.Select) -> str: 267 limit = expression.args.get("limit") 268 offset = expression.args.get("offset") 269 270 if isinstance(limit, exp.Fetch) and not offset: 271 # Dialects like Oracle can FETCH directly from a row set but 272 # T-SQL requires an ORDER BY + OFFSET clause in order to FETCH 273 offset = exp.Offset(expression=exp.Literal.number(0)) 274 expression.set("offset", offset) 275 276 if offset: 277 if not expression.args.get("order"): 278 # ORDER BY is required in order to use OFFSET in a query, so we use 279 # a noop order by, since we don't really care about the order. 280 # See: https://www.microsoftpressstore.com/articles/article.aspx?p=2314819 281 expression.order_by(exp.select(exp.null()).subquery(), copy=False) 282 283 if isinstance(limit, exp.Limit): 284 # TOP and OFFSET can't be combined, we need use FETCH instead of TOP 285 # we replace here because otherwise TOP would be generated in select_sql 286 limit.replace(exp.Fetch(direction="FIRST", count=limit.expression)) 287 288 return super().select_sql(expression) 289 290 def convert_sql(self, expression: exp.Convert) -> str: 291 name = "TRY_CONVERT" if expression.args.get("safe") else "CONVERT" 292 return self.func(name, expression.this, expression.expression, expression.args.get("style")) 293 294 def queryoption_sql(self, expression: exp.QueryOption) -> str: 295 option = self.sql(expression, "this") 296 value = self.sql(expression, "expression") 297 if value: 298 optional_equal_sign = "= " if option in OPTIONS_THAT_REQUIRE_EQUAL else "" 299 return f"{option} {optional_equal_sign}{value}" 300 return option 301 302 def lateral_op(self, expression: exp.Lateral) -> str: 303 cross_apply = expression.args.get("cross_apply") 304 if cross_apply is True: 305 return "CROSS APPLY" 306 if cross_apply is False: 307 return "OUTER APPLY" 308 309 # TODO: perhaps we can check if the parent is a Join and transpile it appropriately 310 self.unsupported("LATERAL clause is not supported.") 311 return "LATERAL" 312 313 def splitpart_sql(self, expression: exp.SplitPart) -> str: 314 this = expression.this 315 split_count = len(this.name.split(".")) 316 delimiter = expression.args.get("delimiter") 317 part_index = expression.args.get("part_index") 318 319 if ( 320 not all(isinstance(arg, exp.Literal) for arg in (this, delimiter, part_index)) 321 or (delimiter and delimiter.name != ".") 322 or not part_index 323 or split_count > 4 324 ): 325 self.unsupported( 326 "SPLIT_PART can be transpiled to PARSENAME only for '.' delimiter and literal values" 327 ) 328 return "" 329 330 return self.func( 331 "PARSENAME", this, exp.Literal.number(split_count + 1 - part_index.to_py()) 332 ) 333 334 def extract_sql(self, expression: exp.Extract) -> str: 335 part = expression.this 336 name = DATE_PART_UNMAPPING.get(part.name.upper()) or part 337 338 return self.func("DATEPART", name, expression.expression) 339 340 def timefromparts_sql(self, expression: exp.TimeFromParts) -> str: 341 nano = expression.args.get("nano") 342 if nano is not None: 343 nano.pop() 344 self.unsupported("Specifying nanoseconds is not supported in TIMEFROMPARTS.") 345 346 if expression.args.get("fractions") is None: 347 expression.set("fractions", exp.Literal.number(0)) 348 if expression.args.get("precision") is None: 349 expression.set("precision", exp.Literal.number(0)) 350 351 return rename_func("TIMEFROMPARTS")(self, expression) 352 353 def timestampfromparts_sql(self, expression: exp.TimestampFromParts) -> str: 354 zone = expression.args.get("zone") 355 if zone is not None: 356 zone.pop() 357 self.unsupported("Time zone is not supported in DATETIMEFROMPARTS.") 358 359 nano = expression.args.get("nano") 360 if nano is not None: 361 nano.pop() 362 self.unsupported("Specifying nanoseconds is not supported in DATETIMEFROMPARTS.") 363 364 if expression.args.get("milli") is None: 365 expression.set("milli", exp.Literal.number(0)) 366 367 return rename_func("DATETIMEFROMPARTS")(self, expression) 368 369 def setitem_sql(self, expression: exp.SetItem) -> str: 370 this = expression.this 371 if isinstance(this, exp.EQ) and not isinstance(this.left, exp.Parameter): 372 # T-SQL does not use '=' in SET command, except when the LHS is a variable. 373 return f"{self.sql(this.left)} {self.sql(this.right)}" 374 375 return super().setitem_sql(expression) 376 377 def boolean_sql(self, expression: exp.Boolean) -> str: 378 if type(expression.parent) in BIT_TYPES or isinstance( 379 expression.find_ancestor(exp.Values, exp.Select), exp.Values 380 ): 381 return "1" if expression.this else "0" 382 383 return "(1 = 1)" if expression.this else "(1 = 0)" 384 385 def is_sql(self, expression: exp.Is) -> str: 386 negate = expression.args.get("negate") 387 if isinstance(expression.expression, exp.Boolean): 388 return self.binary(expression, "<>" if negate else "=") 389 return self.binary(expression, "IS NOT" if negate else "IS") 390 391 def createable_sql(self, expression: exp.Create, locations: defaultdict) -> str: 392 sql = self.sql(expression, "this") 393 properties = expression.args.get("properties") 394 395 start = self._identifier_start 396 if ( 397 not sql.startswith("#") 398 and not sql.startswith(f"{start}#") 399 and any( 400 isinstance(prop, exp.TemporaryProperty) 401 for prop in (properties.expressions if properties else []) 402 ) 403 ): 404 sql = f"{start}#{sql[len(start) :]}" if sql.startswith(start) else f"#{sql}" 405 406 return sql 407 408 def create_sql(self, expression: exp.Create) -> str: 409 kind = expression.kind 410 exists = expression.args.get("exists") 411 expression.set("exists", None) 412 413 like_property = expression.find(exp.LikeProperty) 414 if like_property: 415 ctas_expression = like_property.this 416 else: 417 ctas_expression = expression.expression 418 419 if kind == "VIEW": 420 expression.this.set("catalog", None) 421 with_ = expression.args.get("with_") 422 if ctas_expression and with_: 423 # We've already preprocessed the Create expression to bubble up any nested CTEs, 424 # but CREATE VIEW actually requires the WITH clause to come after it so we need 425 # to amend the AST by moving the CTEs to the CREATE VIEW statement's query. 426 ctas_expression.set("with_", with_.pop()) 427 elif ( 428 kind == "FUNCTION" 429 and isinstance(ctas_expression, exp.Return) 430 and isinstance(body := ctas_expression.this.unnest(), exp.Query) 431 and (with_ := expression.args.get("with_")) 432 ): 433 # Similar to the VIEW branch, the table-valued functions require the WITH clause 434 # to stay inside the RETURN body, so we move back any CTEs that were bubbled up. 435 body.set("with_", with_.pop()) 436 437 table = expression.find(exp.Table) 438 439 # Convert CTAS statement to SELECT .. INTO .. 440 if kind == "TABLE" and ctas_expression: 441 if isinstance(ctas_expression, exp.UNWRAPPED_QUERIES): 442 ctas_expression = ctas_expression.subquery() 443 444 properties = expression.args.get("properties") or exp.Properties() 445 is_temp = any(isinstance(p, exp.TemporaryProperty) for p in properties.expressions) 446 447 select_into = exp.select("*").from_(exp.alias_(ctas_expression, "temp", table=True)) 448 select_into.set("into", exp.Into(this=table, temporary=is_temp)) 449 450 if like_property: 451 select_into.limit(0, copy=False) 452 453 sql = self.sql(select_into) 454 else: 455 sql = super().create_sql(expression) 456 457 if exists: 458 identifier = self.sql(exp.Literal.string(exp.table_name(table) if table else "")) 459 sql_with_ctes = self.prepend_ctes(expression, sql) 460 sql_literal = self.sql(exp.Literal.string(sql_with_ctes)) 461 if kind == "SCHEMA": 462 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = {identifier}) EXEC({sql_literal})""" 463 elif kind == "TABLE": 464 assert table 465 where = exp.and_( 466 exp.column("TABLE_NAME").eq(table.name), 467 exp.column("TABLE_SCHEMA").eq(table.db) if table.db else None, 468 exp.column("TABLE_CATALOG").eq(table.catalog) if table.catalog else None, 469 ) 470 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE {where}) EXEC({sql_literal})""" 471 elif kind == "INDEX": 472 index = self.sql(exp.Literal.string(expression.this.text("this"))) 473 return f"""IF NOT EXISTS (SELECT * FROM sys.indexes WHERE object_id = object_id({identifier}) AND name = {index}) EXEC({sql_literal})""" 474 elif expression.args.get("replace"): 475 sql = sql.replace("CREATE OR REPLACE ", "CREATE OR ALTER ", 1) 476 477 return self.prepend_ctes(expression, sql) 478 479 @generator.unsupported_args("unlogged", "expressions") 480 def into_sql(self, expression: exp.Into) -> str: 481 if expression.args.get("temporary"): 482 # If the Into expression has a temporary property, push this down to the Identifier 483 table = expression.find(exp.Table) 484 if table and isinstance(table.this, exp.Identifier): 485 table.this.set("temporary", True) 486 487 return f"{self.seg('INTO')} {self.sql(expression, 'this')}" 488 489 def count_sql(self, expression: exp.Count) -> str: 490 func_name = "COUNT_BIG" if expression.args.get("big_int") else "COUNT" 491 return rename_func(func_name)(self, expression) 492 493 def datediff_sql(self, expression: exp.DateDiff) -> str: 494 func_name = "DATEDIFF_BIG" if expression.args.get("big_int") else "DATEDIFF" 495 return date_delta_sql(func_name)(self, expression) 496 497 def offset_sql(self, expression: exp.Offset) -> str: 498 return f"{super().offset_sql(expression)} ROWS" 499 500 def version_sql(self, expression: exp.Version) -> str: 501 name = "SYSTEM_TIME" if expression.name == "TIMESTAMP" else expression.name 502 this = f"FOR {name}" 503 expr = expression.expression 504 kind = expression.text("kind") 505 if kind in ("FROM", "BETWEEN"): 506 args = expr.expressions 507 sep = "TO" if kind == "FROM" else "AND" 508 expr_sql = f"{self.sql(seq_get(args, 0))} {sep} {self.sql(seq_get(args, 1))}" 509 else: 510 expr_sql = self.sql(expr) 511 512 expr_sql = f" {expr_sql}" if expr_sql else "" 513 return f"{this} {kind}{expr_sql}" 514 515 def returnsproperty_sql(self, expression: exp.ReturnsProperty) -> str: 516 table = expression.args.get("table") 517 table = f"{table} " if table else "" 518 return f"RETURNS {table}{self.sql(expression, 'this')}" 519 520 def returning_sql(self, expression: exp.Returning) -> str: 521 into = self.sql(expression, "into") 522 into = self.seg(f"INTO {into}") if into else "" 523 return f"{self.seg('OUTPUT')} {self.expressions(expression, flat=True)}{into}" 524 525 def transaction_sql(self, expression: exp.Transaction) -> str: 526 this = self.sql(expression, "this") 527 this = f" {this}" if this else "" 528 mark = self.sql(expression, "mark") 529 mark = f" WITH MARK {mark}" if mark else "" 530 return f"BEGIN TRANSACTION{this}{mark}" 531 532 def commit_sql(self, expression: exp.Commit) -> str: 533 this = self.sql(expression, "this") 534 this = f" {this}" if this else "" 535 durability = expression.args.get("durability") 536 durability = ( 537 f" WITH (DELAYED_DURABILITY = {'ON' if durability else 'OFF'})" 538 if durability is not None 539 else "" 540 ) 541 return f"COMMIT TRANSACTION{this}{durability}" 542 543 def rollback_sql(self, expression: exp.Rollback) -> str: 544 this = self.sql(expression, "this") 545 this = f" {this}" if this else "" 546 return f"ROLLBACK TRANSACTION{this}" 547 548 def identifier_sql(self, expression: exp.Identifier) -> str: 549 identifier = super().identifier_sql(expression) 550 551 if expression.args.get("global_"): 552 prefix = "##" 553 elif expression.args.get("temporary"): 554 prefix = "#" 555 else: 556 return identifier 557 558 start = self._identifier_start 559 if expression.quoted and identifier.startswith(start): 560 return f"{start}{prefix}{identifier[len(start) :]}" 561 562 return f"{prefix}{identifier}" 563 564 def constraint_sql(self, expression: exp.Constraint) -> str: 565 this = self.sql(expression, "this") 566 expressions = self.expressions(expression, flat=True, sep=" ") 567 return f"CONSTRAINT {this} {expressions}" 568 569 def length_sql(self, expression: exp.Length) -> str: 570 return self._uncast_text(expression, "LEN") 571 572 def right_sql(self, expression: exp.Right) -> str: 573 return self._uncast_text(expression, "RIGHT") 574 575 def left_sql(self, expression: exp.Left) -> str: 576 return self._uncast_text(expression, "LEFT") 577 578 def _uncast_text(self, expression: exp.Expr, name: str) -> str: 579 this = expression.this 580 if isinstance(this, exp.Cast) and this.is_type(exp.DType.TEXT): 581 this_sql = self.sql(this, "this") 582 else: 583 this_sql = self.sql(this) 584 expression_sql = self.sql(expression, "expression") 585 return self.func(name, this_sql, expression_sql if expression_sql else None) 586 587 def partition_sql(self, expression: exp.Partition) -> str: 588 return f"WITH (PARTITIONS({self.expressions(expression, flat=True)}))" 589 590 def alter_sql(self, expression: exp.Alter) -> str: 591 action = seq_get(expression.args.get("actions") or [], 0) 592 if isinstance(action, exp.AlterRename): 593 return f"EXEC sp_rename '{self.sql(expression.this)}', '{action.this.name}'" 594 return super().alter_sql(expression) 595 596 def drop_sql(self, expression: exp.Drop) -> str: 597 if expression.args["kind"] == "VIEW": 598 expression.this.set("catalog", None) 599 return super().drop_sql(expression) 600 601 def options_modifier(self, expression: exp.Expr) -> str: 602 options = self.expressions(expression, key="options") 603 return f" OPTION{self.wrap(options)}" if options else "" 604 605 def dpipe_sql(self, expression: exp.DPipe) -> str: 606 return self.sql(reduce(lambda x, y: exp.Add(this=x, expression=y), expression.flatten())) 607 608 def isascii_sql(self, expression: exp.IsAscii) -> str: 609 return f"(PATINDEX(CONVERT(VARCHAR(MAX), 0x255b5e002d7f5d25) COLLATE Latin1_General_BIN, {self.sql(expression.this)}) = 0)" 610 611 def columndef_sql(self, expression: exp.ColumnDef, sep: str = " ") -> str: 612 this = super().columndef_sql(expression, sep) 613 default = self.sql(expression, "default") 614 default = f" = {default}" if default else "" 615 output = self.sql(expression, "output") 616 output = f" {output}" if output else "" 617 return f"{this}{default}{output}" 618 619 def coalesce_sql(self, expression: exp.Coalesce) -> str: 620 func_name = "ISNULL" if expression.args.get("is_null") else "COALESCE" 621 return rename_func(func_name)(self, expression) 622 623 def storedprocedure_sql(self, expression: exp.StoredProcedure) -> str: 624 this = self.sql(expression, "this") 625 expressions = self.expressions(expression) 626 expressions = ( 627 self.wrap(expressions) if expression.args.get("wrapped") else f" {expressions}" 628 ) 629 return f"{this}{expressions}" if expressions.strip() != "" else this 630 631 def ifblock_sql(self, expression: exp.IfBlock) -> str: 632 this = self.sql(expression, "this") 633 true = self.sql(expression, "true") 634 true = f" {true}" if true else " " 635 false = self.sql(expression, "false") 636 false = f"; ELSE BEGIN {false}" if false else "" 637 return f"IF {this} BEGIN{true}{false}" 638 639 def whileblock_sql(self, expression: exp.WhileBlock) -> str: 640 this = self.sql(expression, "this") 641 body = self.sql(expression, "body") 642 body = f" {body}" if body else " " 643 return f"WHILE {this} BEGIN{body}" 644 645 def execute_sql(self, expression: exp.Execute) -> str: 646 this = self.sql(expression, "this") 647 expressions = self.expressions(expression) 648 expressions = f" {expressions}" if expressions else "" 649 return_status = self.sql(expression, "return_status") 650 return_status = f"{return_status} = " if return_status else "" 651 return f"EXECUTE {return_status}{this}{expressions}" 652 653 def executesql_sql(self, expression: exp.ExecuteSql) -> str: 654 return self.execute_sql(expression)
DATE_PART_UNMAPPING =
{'WEEKISO': 'ISO_WEEK', 'DAYOFWEEK': 'WEEKDAY', 'TIMEZONE_MINUTE': 'TZOFFSET'}
BIT_TYPES =
{<class 'sqlglot.expressions.core.Alias'>, <class 'sqlglot.expressions.core.NEQ'>, <class 'sqlglot.expressions.core.Is'>, <class 'sqlglot.expressions.query.Select'>, <class 'sqlglot.expressions.core.In'>, <class 'sqlglot.expressions.core.EQ'>}
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.
122class TSQLGenerator(generator.Generator): 123 SELECT_KINDS: tuple[str, ...] = () 124 TRY_SUPPORTED = False 125 SUPPORTS_UESCAPE = False 126 SUPPORTS_DECODE_CASE = False 127 128 AFTER_HAVING_MODIFIER_TRANSFORMS = generator.AFTER_HAVING_MODIFIER_TRANSFORMS 129 130 LIMIT_IS_TOP = True 131 QUERY_HINTS = False 132 RETURNING_END = False 133 NVL2_SUPPORTED = False 134 ALTER_TABLE_INCLUDE_COLUMN_KEYWORD = False 135 LIMIT_FETCH = "FETCH" 136 COMPUTED_COLUMN_WITH_TYPE = False 137 CTE_RECURSIVE_KEYWORD_REQUIRED = False 138 ENSURE_BOOLS = True 139 NULL_ORDERING_SUPPORTED: bool | None = None 140 SUPPORTS_SINGLE_ARG_CONCAT = False 141 TABLESAMPLE_SEED_KEYWORD = "REPEATABLE" 142 SUPPORTS_SELECT_INTO = True 143 JSON_PATH_BRACKETED_KEY_SUPPORTED = False 144 SUPPORTS_TO_NUMBER = False 145 SET_OP_MODIFIERS = False 146 COPY_PARAMS_EQ_REQUIRED = True 147 PARSE_JSON_NAME: str | None = None 148 EXCEPT_INTERSECT_SUPPORT_ALL_CLAUSE = False 149 ALTER_SET_WRAPPED = True 150 ALTER_SET_TYPE = "" 151 152 EXPRESSIONS_WITHOUT_NESTED_CTES = { 153 exp.Create, 154 exp.Delete, 155 exp.Insert, 156 exp.Intersect, 157 exp.Except, 158 exp.Merge, 159 exp.Select, 160 exp.Subquery, 161 exp.Union, 162 exp.Update, 163 } 164 165 SUPPORTED_JSON_PATH_PARTS = { 166 exp.JSONPathKey, 167 exp.JSONPathRoot, 168 exp.JSONPathSubscript, 169 } 170 171 TYPE_MAPPING = { 172 **{ 173 k: v 174 for k, v in generator.Generator.TYPE_MAPPING.items() 175 if k not in (exp.DType.NCHAR, exp.DType.NVARCHAR) 176 }, 177 exp.DType.BOOLEAN: "BIT", 178 exp.DType.DATETIME2: "DATETIME2", 179 exp.DType.DECIMAL: "NUMERIC", 180 exp.DType.DOUBLE: "FLOAT", 181 exp.DType.INT: "INTEGER", 182 exp.DType.ROWVERSION: "ROWVERSION", 183 exp.DType.TEXT: "VARCHAR(MAX)", 184 exp.DType.TIMESTAMP: "DATETIME2", 185 exp.DType.TIMESTAMPNTZ: "DATETIME2", 186 exp.DType.TIMESTAMPTZ: "DATETIMEOFFSET", 187 exp.DType.SMALLDATETIME: "SMALLDATETIME", 188 exp.DType.UTINYINT: "TINYINT", 189 exp.DType.VARIANT: "SQL_VARIANT", 190 exp.DType.UUID: "UNIQUEIDENTIFIER", 191 } 192 193 TRANSFORMS = { 194 **{k: v for k, v in generator.Generator.TRANSFORMS.items() if k != exp.ReturnsProperty}, 195 exp.AnyValue: any_value_to_max_sql, 196 exp.Atan2: rename_func("ATN2"), 197 exp.ArrayToString: rename_func("STRING_AGG"), 198 exp.AutoIncrementColumnConstraint: lambda *_: "IDENTITY", 199 exp.Ceil: rename_func("CEILING"), 200 exp.Chr: rename_func("CHAR"), 201 exp.DateAdd: date_delta_sql("DATEADD"), 202 exp.CTE: transforms.preprocess([qualify_derived_table_outputs]), 203 exp.CurrentDate: rename_func("GETDATE"), 204 exp.CurrentTimestamp: rename_func("GETDATE"), 205 exp.CurrentTimestampLTZ: rename_func("SYSDATETIMEOFFSET"), 206 exp.DateStrToDate: datestrtodate_sql, 207 exp.Day: remove_ts_or_ds_to_date(), 208 exp.GeneratedAsIdentityColumnConstraint: generatedasidentitycolumnconstraint_sql, 209 exp.GroupConcat: _string_agg_sql, 210 exp.If: rename_func("IIF"), 211 exp.JSONExtract: _json_extract_sql, 212 exp.JSONExtractScalar: _json_extract_sql, 213 exp.LastDay: lambda self, e: self.func("EOMONTH", e.this), 214 exp.Ln: rename_func("LOG"), 215 exp.Max: max_or_greatest, 216 exp.MD5: lambda self, e: self.func("HASHBYTES", exp.Literal.string("MD5"), e.this), 217 exp.Min: min_or_least, 218 exp.Month: remove_ts_or_ds_to_date(), 219 exp.NumberToStr: _format_sql, 220 exp.Repeat: rename_func("REPLICATE"), 221 exp.CurrentSchema: rename_func("SCHEMA_NAME"), 222 exp.Select: transforms.preprocess( 223 [ 224 transforms.eliminate_distinct_on, 225 transforms.eliminate_semi_and_anti_joins, 226 transforms.eliminate_qualify, 227 transforms.unnest_generate_date_array_using_recursive_cte, 228 ] 229 ), 230 exp.Stddev: rename_func("STDEV"), 231 exp.StrPosition: lambda self, e: strposition_sql( 232 self, e, func_name="CHARINDEX", supports_position=True 233 ), 234 exp.Subquery: transforms.preprocess([qualify_derived_table_outputs]), 235 exp.SHA: lambda self, e: self.func("HASHBYTES", exp.Literal.string("SHA1"), e.this), 236 exp.SHA1Digest: lambda self, e: self.func("HASHBYTES", exp.Literal.string("SHA1"), e.this), 237 exp.SHA2: lambda self, e: self.func( 238 "HASHBYTES", exp.Literal.string(f"SHA2_{e.args.get('length', 256)}"), e.this 239 ), 240 exp.TemporaryProperty: lambda self, e: "", 241 exp.TimeStrToTime: _timestrtotime_sql, 242 exp.TimeToStr: _format_sql, 243 exp.TimestampAdd: date_delta_sql("DATEADD"), 244 exp.Trim: trim_sql, 245 exp.TsOrDsAdd: date_delta_sql("DATEADD", cast=True), 246 exp.TsOrDsDiff: date_delta_sql("DATEDIFF"), 247 exp.TimestampTrunc: lambda self, e: self.func("DATETRUNC", e.unit, e.this), 248 exp.Trunc: lambda self, e: self.func( 249 "ROUND", 250 e.this, 251 e.args.get("decimals") or exp.Literal.number(0), 252 exp.Literal.number(1), 253 ), 254 exp.Uuid: lambda *_: "NEWID()", 255 exp.Year: remove_ts_or_ds_to_date(), 256 exp.DateFromParts: rename_func("DATEFROMPARTS"), 257 } 258 259 PROPERTIES_LOCATION = { 260 **generator.Generator.PROPERTIES_LOCATION, 261 exp.VolatileProperty: exp.Properties.Location.UNSUPPORTED, 262 } 263 264 def scope_resolution(self, rhs: str, scope_name: str) -> str: 265 return f"{scope_name}::{rhs}" 266 267 def select_sql(self, expression: exp.Select) -> str: 268 limit = expression.args.get("limit") 269 offset = expression.args.get("offset") 270 271 if isinstance(limit, exp.Fetch) and not offset: 272 # Dialects like Oracle can FETCH directly from a row set but 273 # T-SQL requires an ORDER BY + OFFSET clause in order to FETCH 274 offset = exp.Offset(expression=exp.Literal.number(0)) 275 expression.set("offset", offset) 276 277 if offset: 278 if not expression.args.get("order"): 279 # ORDER BY is required in order to use OFFSET in a query, so we use 280 # a noop order by, since we don't really care about the order. 281 # See: https://www.microsoftpressstore.com/articles/article.aspx?p=2314819 282 expression.order_by(exp.select(exp.null()).subquery(), copy=False) 283 284 if isinstance(limit, exp.Limit): 285 # TOP and OFFSET can't be combined, we need use FETCH instead of TOP 286 # we replace here because otherwise TOP would be generated in select_sql 287 limit.replace(exp.Fetch(direction="FIRST", count=limit.expression)) 288 289 return super().select_sql(expression) 290 291 def convert_sql(self, expression: exp.Convert) -> str: 292 name = "TRY_CONVERT" if expression.args.get("safe") else "CONVERT" 293 return self.func(name, expression.this, expression.expression, expression.args.get("style")) 294 295 def queryoption_sql(self, expression: exp.QueryOption) -> str: 296 option = self.sql(expression, "this") 297 value = self.sql(expression, "expression") 298 if value: 299 optional_equal_sign = "= " if option in OPTIONS_THAT_REQUIRE_EQUAL else "" 300 return f"{option} {optional_equal_sign}{value}" 301 return option 302 303 def lateral_op(self, expression: exp.Lateral) -> str: 304 cross_apply = expression.args.get("cross_apply") 305 if cross_apply is True: 306 return "CROSS APPLY" 307 if cross_apply is False: 308 return "OUTER APPLY" 309 310 # TODO: perhaps we can check if the parent is a Join and transpile it appropriately 311 self.unsupported("LATERAL clause is not supported.") 312 return "LATERAL" 313 314 def splitpart_sql(self, expression: exp.SplitPart) -> str: 315 this = expression.this 316 split_count = len(this.name.split(".")) 317 delimiter = expression.args.get("delimiter") 318 part_index = expression.args.get("part_index") 319 320 if ( 321 not all(isinstance(arg, exp.Literal) for arg in (this, delimiter, part_index)) 322 or (delimiter and delimiter.name != ".") 323 or not part_index 324 or split_count > 4 325 ): 326 self.unsupported( 327 "SPLIT_PART can be transpiled to PARSENAME only for '.' delimiter and literal values" 328 ) 329 return "" 330 331 return self.func( 332 "PARSENAME", this, exp.Literal.number(split_count + 1 - part_index.to_py()) 333 ) 334 335 def extract_sql(self, expression: exp.Extract) -> str: 336 part = expression.this 337 name = DATE_PART_UNMAPPING.get(part.name.upper()) or part 338 339 return self.func("DATEPART", name, expression.expression) 340 341 def timefromparts_sql(self, expression: exp.TimeFromParts) -> str: 342 nano = expression.args.get("nano") 343 if nano is not None: 344 nano.pop() 345 self.unsupported("Specifying nanoseconds is not supported in TIMEFROMPARTS.") 346 347 if expression.args.get("fractions") is None: 348 expression.set("fractions", exp.Literal.number(0)) 349 if expression.args.get("precision") is None: 350 expression.set("precision", exp.Literal.number(0)) 351 352 return rename_func("TIMEFROMPARTS")(self, expression) 353 354 def timestampfromparts_sql(self, expression: exp.TimestampFromParts) -> str: 355 zone = expression.args.get("zone") 356 if zone is not None: 357 zone.pop() 358 self.unsupported("Time zone is not supported in DATETIMEFROMPARTS.") 359 360 nano = expression.args.get("nano") 361 if nano is not None: 362 nano.pop() 363 self.unsupported("Specifying nanoseconds is not supported in DATETIMEFROMPARTS.") 364 365 if expression.args.get("milli") is None: 366 expression.set("milli", exp.Literal.number(0)) 367 368 return rename_func("DATETIMEFROMPARTS")(self, expression) 369 370 def setitem_sql(self, expression: exp.SetItem) -> str: 371 this = expression.this 372 if isinstance(this, exp.EQ) and not isinstance(this.left, exp.Parameter): 373 # T-SQL does not use '=' in SET command, except when the LHS is a variable. 374 return f"{self.sql(this.left)} {self.sql(this.right)}" 375 376 return super().setitem_sql(expression) 377 378 def boolean_sql(self, expression: exp.Boolean) -> str: 379 if type(expression.parent) in BIT_TYPES or isinstance( 380 expression.find_ancestor(exp.Values, exp.Select), exp.Values 381 ): 382 return "1" if expression.this else "0" 383 384 return "(1 = 1)" if expression.this else "(1 = 0)" 385 386 def is_sql(self, expression: exp.Is) -> str: 387 negate = expression.args.get("negate") 388 if isinstance(expression.expression, exp.Boolean): 389 return self.binary(expression, "<>" if negate else "=") 390 return self.binary(expression, "IS NOT" if negate else "IS") 391 392 def createable_sql(self, expression: exp.Create, locations: defaultdict) -> str: 393 sql = self.sql(expression, "this") 394 properties = expression.args.get("properties") 395 396 start = self._identifier_start 397 if ( 398 not sql.startswith("#") 399 and not sql.startswith(f"{start}#") 400 and any( 401 isinstance(prop, exp.TemporaryProperty) 402 for prop in (properties.expressions if properties else []) 403 ) 404 ): 405 sql = f"{start}#{sql[len(start) :]}" if sql.startswith(start) else f"#{sql}" 406 407 return sql 408 409 def create_sql(self, expression: exp.Create) -> str: 410 kind = expression.kind 411 exists = expression.args.get("exists") 412 expression.set("exists", None) 413 414 like_property = expression.find(exp.LikeProperty) 415 if like_property: 416 ctas_expression = like_property.this 417 else: 418 ctas_expression = expression.expression 419 420 if kind == "VIEW": 421 expression.this.set("catalog", None) 422 with_ = expression.args.get("with_") 423 if ctas_expression and with_: 424 # We've already preprocessed the Create expression to bubble up any nested CTEs, 425 # but CREATE VIEW actually requires the WITH clause to come after it so we need 426 # to amend the AST by moving the CTEs to the CREATE VIEW statement's query. 427 ctas_expression.set("with_", with_.pop()) 428 elif ( 429 kind == "FUNCTION" 430 and isinstance(ctas_expression, exp.Return) 431 and isinstance(body := ctas_expression.this.unnest(), exp.Query) 432 and (with_ := expression.args.get("with_")) 433 ): 434 # Similar to the VIEW branch, the table-valued functions require the WITH clause 435 # to stay inside the RETURN body, so we move back any CTEs that were bubbled up. 436 body.set("with_", with_.pop()) 437 438 table = expression.find(exp.Table) 439 440 # Convert CTAS statement to SELECT .. INTO .. 441 if kind == "TABLE" and ctas_expression: 442 if isinstance(ctas_expression, exp.UNWRAPPED_QUERIES): 443 ctas_expression = ctas_expression.subquery() 444 445 properties = expression.args.get("properties") or exp.Properties() 446 is_temp = any(isinstance(p, exp.TemporaryProperty) for p in properties.expressions) 447 448 select_into = exp.select("*").from_(exp.alias_(ctas_expression, "temp", table=True)) 449 select_into.set("into", exp.Into(this=table, temporary=is_temp)) 450 451 if like_property: 452 select_into.limit(0, copy=False) 453 454 sql = self.sql(select_into) 455 else: 456 sql = super().create_sql(expression) 457 458 if exists: 459 identifier = self.sql(exp.Literal.string(exp.table_name(table) if table else "")) 460 sql_with_ctes = self.prepend_ctes(expression, sql) 461 sql_literal = self.sql(exp.Literal.string(sql_with_ctes)) 462 if kind == "SCHEMA": 463 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = {identifier}) EXEC({sql_literal})""" 464 elif kind == "TABLE": 465 assert table 466 where = exp.and_( 467 exp.column("TABLE_NAME").eq(table.name), 468 exp.column("TABLE_SCHEMA").eq(table.db) if table.db else None, 469 exp.column("TABLE_CATALOG").eq(table.catalog) if table.catalog else None, 470 ) 471 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE {where}) EXEC({sql_literal})""" 472 elif kind == "INDEX": 473 index = self.sql(exp.Literal.string(expression.this.text("this"))) 474 return f"""IF NOT EXISTS (SELECT * FROM sys.indexes WHERE object_id = object_id({identifier}) AND name = {index}) EXEC({sql_literal})""" 475 elif expression.args.get("replace"): 476 sql = sql.replace("CREATE OR REPLACE ", "CREATE OR ALTER ", 1) 477 478 return self.prepend_ctes(expression, sql) 479 480 @generator.unsupported_args("unlogged", "expressions") 481 def into_sql(self, expression: exp.Into) -> str: 482 if expression.args.get("temporary"): 483 # If the Into expression has a temporary property, push this down to the Identifier 484 table = expression.find(exp.Table) 485 if table and isinstance(table.this, exp.Identifier): 486 table.this.set("temporary", True) 487 488 return f"{self.seg('INTO')} {self.sql(expression, 'this')}" 489 490 def count_sql(self, expression: exp.Count) -> str: 491 func_name = "COUNT_BIG" if expression.args.get("big_int") else "COUNT" 492 return rename_func(func_name)(self, expression) 493 494 def datediff_sql(self, expression: exp.DateDiff) -> str: 495 func_name = "DATEDIFF_BIG" if expression.args.get("big_int") else "DATEDIFF" 496 return date_delta_sql(func_name)(self, expression) 497 498 def offset_sql(self, expression: exp.Offset) -> str: 499 return f"{super().offset_sql(expression)} ROWS" 500 501 def version_sql(self, expression: exp.Version) -> str: 502 name = "SYSTEM_TIME" if expression.name == "TIMESTAMP" else expression.name 503 this = f"FOR {name}" 504 expr = expression.expression 505 kind = expression.text("kind") 506 if kind in ("FROM", "BETWEEN"): 507 args = expr.expressions 508 sep = "TO" if kind == "FROM" else "AND" 509 expr_sql = f"{self.sql(seq_get(args, 0))} {sep} {self.sql(seq_get(args, 1))}" 510 else: 511 expr_sql = self.sql(expr) 512 513 expr_sql = f" {expr_sql}" if expr_sql else "" 514 return f"{this} {kind}{expr_sql}" 515 516 def returnsproperty_sql(self, expression: exp.ReturnsProperty) -> str: 517 table = expression.args.get("table") 518 table = f"{table} " if table else "" 519 return f"RETURNS {table}{self.sql(expression, 'this')}" 520 521 def returning_sql(self, expression: exp.Returning) -> str: 522 into = self.sql(expression, "into") 523 into = self.seg(f"INTO {into}") if into else "" 524 return f"{self.seg('OUTPUT')} {self.expressions(expression, flat=True)}{into}" 525 526 def transaction_sql(self, expression: exp.Transaction) -> str: 527 this = self.sql(expression, "this") 528 this = f" {this}" if this else "" 529 mark = self.sql(expression, "mark") 530 mark = f" WITH MARK {mark}" if mark else "" 531 return f"BEGIN TRANSACTION{this}{mark}" 532 533 def commit_sql(self, expression: exp.Commit) -> str: 534 this = self.sql(expression, "this") 535 this = f" {this}" if this else "" 536 durability = expression.args.get("durability") 537 durability = ( 538 f" WITH (DELAYED_DURABILITY = {'ON' if durability else 'OFF'})" 539 if durability is not None 540 else "" 541 ) 542 return f"COMMIT TRANSACTION{this}{durability}" 543 544 def rollback_sql(self, expression: exp.Rollback) -> str: 545 this = self.sql(expression, "this") 546 this = f" {this}" if this else "" 547 return f"ROLLBACK TRANSACTION{this}" 548 549 def identifier_sql(self, expression: exp.Identifier) -> str: 550 identifier = super().identifier_sql(expression) 551 552 if expression.args.get("global_"): 553 prefix = "##" 554 elif expression.args.get("temporary"): 555 prefix = "#" 556 else: 557 return identifier 558 559 start = self._identifier_start 560 if expression.quoted and identifier.startswith(start): 561 return f"{start}{prefix}{identifier[len(start) :]}" 562 563 return f"{prefix}{identifier}" 564 565 def constraint_sql(self, expression: exp.Constraint) -> str: 566 this = self.sql(expression, "this") 567 expressions = self.expressions(expression, flat=True, sep=" ") 568 return f"CONSTRAINT {this} {expressions}" 569 570 def length_sql(self, expression: exp.Length) -> str: 571 return self._uncast_text(expression, "LEN") 572 573 def right_sql(self, expression: exp.Right) -> str: 574 return self._uncast_text(expression, "RIGHT") 575 576 def left_sql(self, expression: exp.Left) -> str: 577 return self._uncast_text(expression, "LEFT") 578 579 def _uncast_text(self, expression: exp.Expr, name: str) -> str: 580 this = expression.this 581 if isinstance(this, exp.Cast) and this.is_type(exp.DType.TEXT): 582 this_sql = self.sql(this, "this") 583 else: 584 this_sql = self.sql(this) 585 expression_sql = self.sql(expression, "expression") 586 return self.func(name, this_sql, expression_sql if expression_sql else None) 587 588 def partition_sql(self, expression: exp.Partition) -> str: 589 return f"WITH (PARTITIONS({self.expressions(expression, flat=True)}))" 590 591 def alter_sql(self, expression: exp.Alter) -> str: 592 action = seq_get(expression.args.get("actions") or [], 0) 593 if isinstance(action, exp.AlterRename): 594 return f"EXEC sp_rename '{self.sql(expression.this)}', '{action.this.name}'" 595 return super().alter_sql(expression) 596 597 def drop_sql(self, expression: exp.Drop) -> str: 598 if expression.args["kind"] == "VIEW": 599 expression.this.set("catalog", None) 600 return super().drop_sql(expression) 601 602 def options_modifier(self, expression: exp.Expr) -> str: 603 options = self.expressions(expression, key="options") 604 return f" OPTION{self.wrap(options)}" if options else "" 605 606 def dpipe_sql(self, expression: exp.DPipe) -> str: 607 return self.sql(reduce(lambda x, y: exp.Add(this=x, expression=y), expression.flatten())) 608 609 def isascii_sql(self, expression: exp.IsAscii) -> str: 610 return f"(PATINDEX(CONVERT(VARCHAR(MAX), 0x255b5e002d7f5d25) COLLATE Latin1_General_BIN, {self.sql(expression.this)}) = 0)" 611 612 def columndef_sql(self, expression: exp.ColumnDef, sep: str = " ") -> str: 613 this = super().columndef_sql(expression, sep) 614 default = self.sql(expression, "default") 615 default = f" = {default}" if default else "" 616 output = self.sql(expression, "output") 617 output = f" {output}" if output else "" 618 return f"{this}{default}{output}" 619 620 def coalesce_sql(self, expression: exp.Coalesce) -> str: 621 func_name = "ISNULL" if expression.args.get("is_null") else "COALESCE" 622 return rename_func(func_name)(self, expression) 623 624 def storedprocedure_sql(self, expression: exp.StoredProcedure) -> str: 625 this = self.sql(expression, "this") 626 expressions = self.expressions(expression) 627 expressions = ( 628 self.wrap(expressions) if expression.args.get("wrapped") else f" {expressions}" 629 ) 630 return f"{this}{expressions}" if expressions.strip() != "" else this 631 632 def ifblock_sql(self, expression: exp.IfBlock) -> str: 633 this = self.sql(expression, "this") 634 true = self.sql(expression, "true") 635 true = f" {true}" if true else " " 636 false = self.sql(expression, "false") 637 false = f"; ELSE BEGIN {false}" if false else "" 638 return f"IF {this} BEGIN{true}{false}" 639 640 def whileblock_sql(self, expression: exp.WhileBlock) -> str: 641 this = self.sql(expression, "this") 642 body = self.sql(expression, "body") 643 body = f" {body}" if body else " " 644 return f"WHILE {this} BEGIN{body}" 645 646 def execute_sql(self, expression: exp.Execute) -> str: 647 this = self.sql(expression, "this") 648 expressions = self.expressions(expression) 649 expressions = f" {expressions}" if expressions else "" 650 return_status = self.sql(expression, "return_status") 651 return_status = f"{return_status} = " if return_status else "" 652 return f"EXECUTE {return_status}{this}{expressions}" 653 654 def executesql_sql(self, expression: exp.ExecuteSql) -> str: 655 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.query.Intersect'>, <class 'sqlglot.expressions.query.Except'>, <class 'sqlglot.expressions.dml.Delete'>, <class 'sqlglot.expressions.query.Union'>, <class 'sqlglot.expressions.dml.Insert'>, <class 'sqlglot.expressions.ddl.Create'>, <class 'sqlglot.expressions.query.Select'>, <class 'sqlglot.expressions.dml.Merge'>, <class 'sqlglot.expressions.query.Subquery'>, <class 'sqlglot.expressions.dml.Update'>}
SUPPORTED_JSON_PATH_PARTS =
{<class 'sqlglot.expressions.query.JSONPathKey'>, <class 'sqlglot.expressions.query.JSONPathRoot'>, <class 'sqlglot.expressions.query.JSONPathSubscript'>}
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.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.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'>}
267 def select_sql(self, expression: exp.Select) -> str: 268 limit = expression.args.get("limit") 269 offset = expression.args.get("offset") 270 271 if isinstance(limit, exp.Fetch) and not offset: 272 # Dialects like Oracle can FETCH directly from a row set but 273 # T-SQL requires an ORDER BY + OFFSET clause in order to FETCH 274 offset = exp.Offset(expression=exp.Literal.number(0)) 275 expression.set("offset", offset) 276 277 if offset: 278 if not expression.args.get("order"): 279 # ORDER BY is required in order to use OFFSET in a query, so we use 280 # a noop order by, since we don't really care about the order. 281 # See: https://www.microsoftpressstore.com/articles/article.aspx?p=2314819 282 expression.order_by(exp.select(exp.null()).subquery(), copy=False) 283 284 if isinstance(limit, exp.Limit): 285 # TOP and OFFSET can't be combined, we need use FETCH instead of TOP 286 # we replace here because otherwise TOP would be generated in select_sql 287 limit.replace(exp.Fetch(direction="FIRST", count=limit.expression)) 288 289 return super().select_sql(expression)
295 def queryoption_sql(self, expression: exp.QueryOption) -> str: 296 option = self.sql(expression, "this") 297 value = self.sql(expression, "expression") 298 if value: 299 optional_equal_sign = "= " if option in OPTIONS_THAT_REQUIRE_EQUAL else "" 300 return f"{option} {optional_equal_sign}{value}" 301 return option
303 def lateral_op(self, expression: exp.Lateral) -> str: 304 cross_apply = expression.args.get("cross_apply") 305 if cross_apply is True: 306 return "CROSS APPLY" 307 if cross_apply is False: 308 return "OUTER APPLY" 309 310 # TODO: perhaps we can check if the parent is a Join and transpile it appropriately 311 self.unsupported("LATERAL clause is not supported.") 312 return "LATERAL"
314 def splitpart_sql(self, expression: exp.SplitPart) -> str: 315 this = expression.this 316 split_count = len(this.name.split(".")) 317 delimiter = expression.args.get("delimiter") 318 part_index = expression.args.get("part_index") 319 320 if ( 321 not all(isinstance(arg, exp.Literal) for arg in (this, delimiter, part_index)) 322 or (delimiter and delimiter.name != ".") 323 or not part_index 324 or split_count > 4 325 ): 326 self.unsupported( 327 "SPLIT_PART can be transpiled to PARSENAME only for '.' delimiter and literal values" 328 ) 329 return "" 330 331 return self.func( 332 "PARSENAME", this, exp.Literal.number(split_count + 1 - part_index.to_py()) 333 )
341 def timefromparts_sql(self, expression: exp.TimeFromParts) -> str: 342 nano = expression.args.get("nano") 343 if nano is not None: 344 nano.pop() 345 self.unsupported("Specifying nanoseconds is not supported in TIMEFROMPARTS.") 346 347 if expression.args.get("fractions") is None: 348 expression.set("fractions", exp.Literal.number(0)) 349 if expression.args.get("precision") is None: 350 expression.set("precision", exp.Literal.number(0)) 351 352 return rename_func("TIMEFROMPARTS")(self, expression)
def
timestampfromparts_sql(self, expression: sqlglot.expressions.temporal.TimestampFromParts) -> str:
354 def timestampfromparts_sql(self, expression: exp.TimestampFromParts) -> str: 355 zone = expression.args.get("zone") 356 if zone is not None: 357 zone.pop() 358 self.unsupported("Time zone is not supported in DATETIMEFROMPARTS.") 359 360 nano = expression.args.get("nano") 361 if nano is not None: 362 nano.pop() 363 self.unsupported("Specifying nanoseconds is not supported in DATETIMEFROMPARTS.") 364 365 if expression.args.get("milli") is None: 366 expression.set("milli", exp.Literal.number(0)) 367 368 return rename_func("DATETIMEFROMPARTS")(self, expression)
370 def setitem_sql(self, expression: exp.SetItem) -> str: 371 this = expression.this 372 if isinstance(this, exp.EQ) and not isinstance(this.left, exp.Parameter): 373 # T-SQL does not use '=' in SET command, except when the LHS is a variable. 374 return f"{self.sql(this.left)} {self.sql(this.right)}" 375 376 return super().setitem_sql(expression)
def
createable_sql( self, expression: sqlglot.expressions.ddl.Create, locations: collections.defaultdict) -> str:
392 def createable_sql(self, expression: exp.Create, locations: defaultdict) -> str: 393 sql = self.sql(expression, "this") 394 properties = expression.args.get("properties") 395 396 start = self._identifier_start 397 if ( 398 not sql.startswith("#") 399 and not sql.startswith(f"{start}#") 400 and any( 401 isinstance(prop, exp.TemporaryProperty) 402 for prop in (properties.expressions if properties else []) 403 ) 404 ): 405 sql = f"{start}#{sql[len(start) :]}" if sql.startswith(start) else f"#{sql}" 406 407 return sql
409 def create_sql(self, expression: exp.Create) -> str: 410 kind = expression.kind 411 exists = expression.args.get("exists") 412 expression.set("exists", None) 413 414 like_property = expression.find(exp.LikeProperty) 415 if like_property: 416 ctas_expression = like_property.this 417 else: 418 ctas_expression = expression.expression 419 420 if kind == "VIEW": 421 expression.this.set("catalog", None) 422 with_ = expression.args.get("with_") 423 if ctas_expression and with_: 424 # We've already preprocessed the Create expression to bubble up any nested CTEs, 425 # but CREATE VIEW actually requires the WITH clause to come after it so we need 426 # to amend the AST by moving the CTEs to the CREATE VIEW statement's query. 427 ctas_expression.set("with_", with_.pop()) 428 elif ( 429 kind == "FUNCTION" 430 and isinstance(ctas_expression, exp.Return) 431 and isinstance(body := ctas_expression.this.unnest(), exp.Query) 432 and (with_ := expression.args.get("with_")) 433 ): 434 # Similar to the VIEW branch, the table-valued functions require the WITH clause 435 # to stay inside the RETURN body, so we move back any CTEs that were bubbled up. 436 body.set("with_", with_.pop()) 437 438 table = expression.find(exp.Table) 439 440 # Convert CTAS statement to SELECT .. INTO .. 441 if kind == "TABLE" and ctas_expression: 442 if isinstance(ctas_expression, exp.UNWRAPPED_QUERIES): 443 ctas_expression = ctas_expression.subquery() 444 445 properties = expression.args.get("properties") or exp.Properties() 446 is_temp = any(isinstance(p, exp.TemporaryProperty) for p in properties.expressions) 447 448 select_into = exp.select("*").from_(exp.alias_(ctas_expression, "temp", table=True)) 449 select_into.set("into", exp.Into(this=table, temporary=is_temp)) 450 451 if like_property: 452 select_into.limit(0, copy=False) 453 454 sql = self.sql(select_into) 455 else: 456 sql = super().create_sql(expression) 457 458 if exists: 459 identifier = self.sql(exp.Literal.string(exp.table_name(table) if table else "")) 460 sql_with_ctes = self.prepend_ctes(expression, sql) 461 sql_literal = self.sql(exp.Literal.string(sql_with_ctes)) 462 if kind == "SCHEMA": 463 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = {identifier}) EXEC({sql_literal})""" 464 elif kind == "TABLE": 465 assert table 466 where = exp.and_( 467 exp.column("TABLE_NAME").eq(table.name), 468 exp.column("TABLE_SCHEMA").eq(table.db) if table.db else None, 469 exp.column("TABLE_CATALOG").eq(table.catalog) if table.catalog else None, 470 ) 471 return f"""IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE {where}) EXEC({sql_literal})""" 472 elif kind == "INDEX": 473 index = self.sql(exp.Literal.string(expression.this.text("this"))) 474 return f"""IF NOT EXISTS (SELECT * FROM sys.indexes WHERE object_id = object_id({identifier}) AND name = {index}) EXEC({sql_literal})""" 475 elif expression.args.get("replace"): 476 sql = sql.replace("CREATE OR REPLACE ", "CREATE OR ALTER ", 1) 477 478 return self.prepend_ctes(expression, sql)
@generator.unsupported_args('unlogged', 'expressions')
def
into_sql(self, expression: sqlglot.expressions.query.Into) -> str:
480 @generator.unsupported_args("unlogged", "expressions") 481 def into_sql(self, expression: exp.Into) -> str: 482 if expression.args.get("temporary"): 483 # If the Into expression has a temporary property, push this down to the Identifier 484 table = expression.find(exp.Table) 485 if table and isinstance(table.this, exp.Identifier): 486 table.this.set("temporary", True) 487 488 return f"{self.seg('INTO')} {self.sql(expression, 'this')}"
501 def version_sql(self, expression: exp.Version) -> str: 502 name = "SYSTEM_TIME" if expression.name == "TIMESTAMP" else expression.name 503 this = f"FOR {name}" 504 expr = expression.expression 505 kind = expression.text("kind") 506 if kind in ("FROM", "BETWEEN"): 507 args = expr.expressions 508 sep = "TO" if kind == "FROM" else "AND" 509 expr_sql = f"{self.sql(seq_get(args, 0))} {sep} {self.sql(seq_get(args, 1))}" 510 else: 511 expr_sql = self.sql(expr) 512 513 expr_sql = f" {expr_sql}" if expr_sql else "" 514 return f"{this} {kind}{expr_sql}"
533 def commit_sql(self, expression: exp.Commit) -> str: 534 this = self.sql(expression, "this") 535 this = f" {this}" if this else "" 536 durability = expression.args.get("durability") 537 durability = ( 538 f" WITH (DELAYED_DURABILITY = {'ON' if durability else 'OFF'})" 539 if durability is not None 540 else "" 541 ) 542 return f"COMMIT TRANSACTION{this}{durability}"
549 def identifier_sql(self, expression: exp.Identifier) -> str: 550 identifier = super().identifier_sql(expression) 551 552 if expression.args.get("global_"): 553 prefix = "##" 554 elif expression.args.get("temporary"): 555 prefix = "#" 556 else: 557 return identifier 558 559 start = self._identifier_start 560 if expression.quoted and identifier.startswith(start): 561 return f"{start}{prefix}{identifier[len(start) :]}" 562 563 return f"{prefix}{identifier}"
612 def columndef_sql(self, expression: exp.ColumnDef, sep: str = " ") -> str: 613 this = super().columndef_sql(expression, sep) 614 default = self.sql(expression, "default") 615 default = f" = {default}" if default else "" 616 output = self.sql(expression, "output") 617 output = f" {output}" if output else "" 618 return f"{this}{default}{output}"
624 def storedprocedure_sql(self, expression: exp.StoredProcedure) -> str: 625 this = self.sql(expression, "this") 626 expressions = self.expressions(expression) 627 expressions = ( 628 self.wrap(expressions) if expression.args.get("wrapped") else f" {expressions}" 629 ) 630 return f"{this}{expressions}" if expressions.strip() != "" else this
632 def ifblock_sql(self, expression: exp.IfBlock) -> str: 633 this = self.sql(expression, "this") 634 true = self.sql(expression, "true") 635 true = f" {true}" if true else " " 636 false = self.sql(expression, "false") 637 false = f"; ELSE BEGIN {false}" if false else "" 638 return f"IF {this} BEGIN{true}{false}"
646 def execute_sql(self, expression: exp.Execute) -> str: 647 this = self.sql(expression, "this") 648 expressions = self.expressions(expression) 649 expressions = f" {expressions}" if expressions else "" 650 return_status = self.sql(expression, "return_status") 651 return_status = f"{return_status} = " if return_status else "" 652 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
- 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
- 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
- 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
- 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
- altermodifysqlsecurity_sql
- usingproperty_sql
- renameindex_sql