sqlglot.dialects.tsql

 15class TSQL(Dialect):
 16    LOG_BASE_FIRST = False
 17    TYPED_DIVISION = True
 18    CONCAT_COALESCE = True
 19    CONCAT_WS_COALESCE = True
 20    NORMALIZATION_STRATEGY = NormalizationStrategy.CASE_INSENSITIVE
 21    ALTER_TABLE_ADD_REQUIRED_FOR_EACH_COLUMN = False
 22
 23    TIME_FORMAT = "'yyyy-mm-dd hh:mm:ss'"
 24
 25    EXPRESSION_METADATA = EXPRESSION_METADATA.copy()
 26
 27    DATE_PART_MAPPING = {
 28        **Dialect.DATE_PART_MAPPING,
 29        "QQ": "QUARTER",
 30        "M": "MONTH",
 31        "Y": "DAYOFYEAR",
 32        "WW": "WEEK",
 33        "N": "MINUTE",
 34        "SS": "SECOND",
 35        "MCS": "MICROSECOND",
 36        "TZOFFSET": "TIMEZONE_MINUTE",
 37        "TZ": "TIMEZONE_MINUTE",
 38        "ISO_WEEK": "WEEKISO",
 39        "ISOWK": "WEEKISO",
 40        "ISOWW": "WEEKISO",
 41    }
 42
 43    TIME_MAPPING = {
 44        "year": "%Y",
 45        "dayofyear": "%j",
 46        "day": "%d",
 47        "dy": "%d",
 48        "y": "%Y",
 49        "week": "%W",
 50        "ww": "%W",
 51        "wk": "%W",
 52        "isowk": "%V",
 53        "isoww": "%V",
 54        "iso_week": "%V",
 55        "hour": "%h",
 56        "hh": "%I",
 57        "minute": "%M",
 58        "mi": "%M",
 59        "n": "%M",
 60        "second": "%S",
 61        "ss": "%S",
 62        "s": "%-S",
 63        "millisecond": "%f",
 64        "ms": "%f",
 65        "weekday": "%w",
 66        "dw": "%w",
 67        "month": "%m",
 68        "mm": "%M",
 69        "m": "%-M",
 70        "Y": "%Y",
 71        "YYYY": "%Y",
 72        "YY": "%y",
 73        "MMMM": "%B",
 74        "MMM": "%b",
 75        "MM": "%m",
 76        "M": "%-m",
 77        "dddd": "%A",
 78        "dd": "%d",
 79        "d": "%-d",
 80        "HH": "%H",
 81        "H": "%-H",
 82        "h": "%-I",
 83        "ffffff": "%f",
 84        "yyyy": "%Y",
 85        "yy": "%y",
 86    }
 87
 88    CONVERT_FORMAT_MAPPING = {
 89        "0": "%b %d %Y %-I:%M%p",
 90        "1": "%m/%d/%y",
 91        "2": "%y.%m.%d",
 92        "3": "%d/%m/%y",
 93        "4": "%d.%m.%y",
 94        "5": "%d-%m-%y",
 95        "6": "%d %b %y",
 96        "7": "%b %d, %y",
 97        "8": "%H:%M:%S",
 98        "9": "%b %d %Y %-I:%M:%S:%f%p",
 99        "10": "mm-dd-yy",
100        "11": "yy/mm/dd",
101        "12": "yymmdd",
102        "13": "%d %b %Y %H:%M:ss:%f",
103        "14": "%H:%M:%S:%f",
104        "20": "%Y-%m-%d %H:%M:%S",
105        "21": "%Y-%m-%d %H:%M:%S.%f",
106        "22": "%m/%d/%y %-I:%M:%S %p",
107        "23": "%Y-%m-%d",
108        "24": "%H:%M:%S",
109        "25": "%Y-%m-%d %H:%M:%S.%f",
110        "100": "%b %d %Y %-I:%M%p",
111        "101": "%m/%d/%Y",
112        "102": "%Y.%m.%d",
113        "103": "%d/%m/%Y",
114        "104": "%d.%m.%Y",
115        "105": "%d-%m-%Y",
116        "106": "%d %b %Y",
117        "107": "%b %d, %Y",
118        "108": "%H:%M:%S",
119        "109": "%b %d %Y %-I:%M:%S:%f%p",
120        "110": "%m-%d-%Y",
121        "111": "%Y/%m/%d",
122        "112": "%Y%m%d",
123        "113": "%d %b %Y %H:%M:%S:%f",
124        "114": "%H:%M:%S:%f",
125        "120": "%Y-%m-%d %H:%M:%S",
126        "121": "%Y-%m-%d %H:%M:%S.%f",
127        "126": "%Y-%m-%dT%H:%M:%S.%f",
128    }
129
130    FORMAT_TIME_MAPPING = {
131        "y": "%B %Y",
132        "d": "%m/%d/%Y",
133        "H": "%-H",
134        "h": "%-I",
135        "s": "%Y-%m-%d %H:%M:%S",
136        "D": "%A,%B,%Y",
137        "f": "%A,%B,%Y %-I:%M %p",
138        "F": "%A,%B,%Y %-I:%M:%S %p",
139        "g": "%m/%d/%Y %-I:%M %p",
140        "G": "%m/%d/%Y %-I:%M:%S %p",
141        "M": "%B %-d",
142        "m": "%B %-d",
143        "O": "%Y-%m-%dT%H:%M:%S",
144        "u": "%Y-%M-%D %H:%M:%S%z",
145        "U": "%A, %B %D, %Y %H:%M:%S%z",
146        "T": "%-I:%M:%S %p",
147        "t": "%-I:%M",
148        "Y": "%a %Y",
149    }
150
151    class Tokenizer(tokens.Tokenizer):
152        IDENTIFIERS = [("[", "]"), '"']
153        QUOTES = ["'", '"']
154        HEX_STRINGS = [("0x", ""), ("0X", "")]
155        VAR_SINGLE_TOKENS = {"@", "$", "#"}
156
157        KEYWORDS = {
158            **tokens.Tokenizer.KEYWORDS,
159            "CLUSTERED INDEX": TokenType.INDEX,
160            "DATETIME2": TokenType.DATETIME2,
161            "DATETIMEOFFSET": TokenType.TIMESTAMPTZ,
162            "DECLARE": TokenType.DECLARE,
163            "EXEC": TokenType.EXECUTE,
164            "FOR SYSTEM_TIME": TokenType.TIMESTAMP_SNAPSHOT,
165            "GO": TokenType.COMMAND,
166            "IMAGE": TokenType.IMAGE,
167            "MONEY": TokenType.MONEY,
168            "NONCLUSTERED INDEX": TokenType.INDEX,
169            "NTEXT": TokenType.TEXT,
170            "OPTION": TokenType.OPTION,
171            "OUTPUT": TokenType.RETURNING,
172            "PRINT": TokenType.COMMAND,
173            "PROC": TokenType.PROCEDURE,
174            "REAL": TokenType.FLOAT,
175            "ROWVERSION": TokenType.ROWVERSION,
176            "SMALLDATETIME": TokenType.SMALLDATETIME,
177            "SMALLMONEY": TokenType.SMALLMONEY,
178            "SQL_VARIANT": TokenType.VARIANT,
179            "SYSTEM_USER": TokenType.CURRENT_USER,
180            "TOP": TokenType.TOP,
181            "TIMESTAMP": TokenType.ROWVERSION,
182            "TINYINT": TokenType.UTINYINT,
183            "UNIQUEIDENTIFIER": TokenType.UUID,
184            "UPDATE STATISTICS": TokenType.COMMAND,
185            "XML": TokenType.XML,
186        }
187        KEYWORDS.pop("/*+")
188
189        COMMANDS = {*tokens.Tokenizer.COMMANDS, TokenType.END} - {TokenType.EXECUTE}
190
191    Parser = TSQLParser
192
193    Generator = TSQLGenerator

151    class Tokenizer(tokens.Tokenizer):
152        IDENTIFIERS = [("[", "]"), '"']
153        QUOTES = ["'", '"']
154        HEX_STRINGS = [("0x", ""), ("0X", "")]
155        VAR_SINGLE_TOKENS = {"@", "$", "#"}
156
157        KEYWORDS = {
158            **tokens.Tokenizer.KEYWORDS,
159            "CLUSTERED INDEX": TokenType.INDEX,
160            "DATETIME2": TokenType.DATETIME2,
161            "DATETIMEOFFSET": TokenType.TIMESTAMPTZ,
162            "DECLARE": TokenType.DECLARE,
163            "EXEC": TokenType.EXECUTE,
164            "FOR SYSTEM_TIME": TokenType.TIMESTAMP_SNAPSHOT,
165            "GO": TokenType.COMMAND,
166            "IMAGE": TokenType.IMAGE,
167            "MONEY": TokenType.MONEY,
168            "NONCLUSTERED INDEX": TokenType.INDEX,
169            "NTEXT": TokenType.TEXT,
170            "OPTION": TokenType.OPTION,
171            "OUTPUT": TokenType.RETURNING,
172            "PRINT": TokenType.COMMAND,
173            "PROC": TokenType.PROCEDURE,
174            "REAL": TokenType.FLOAT,
175            "ROWVERSION": TokenType.ROWVERSION,
176            "SMALLDATETIME": TokenType.SMALLDATETIME,
177            "SMALLMONEY": TokenType.SMALLMONEY,
178            "SQL_VARIANT": TokenType.VARIANT,
179            "SYSTEM_USER": TokenType.CURRENT_USER,
180            "TOP": TokenType.TOP,
181            "TIMESTAMP": TokenType.ROWVERSION,
182            "TINYINT": TokenType.UTINYINT,
183            "UNIQUEIDENTIFIER": TokenType.UUID,
184            "UPDATE STATISTICS": TokenType.COMMAND,
185            "XML": TokenType.XML,
186        }
187        KEYWORDS.pop("/*+")
188
189        COMMANDS = {*tokens.Tokenizer.COMMANDS, TokenType.END} - {TokenType.EXECUTE}