1"""sqlglot expressions DDL."""
  2
  3from __future__ import annotations
  4
  5import typing as t
  6
  7from sqlglot.helper import trait
  8from sqlglot.expressions.core import Expression, Expr, Func
  9from sqlglot.expressions.query import Query, Selectable
 10
 11if t.TYPE_CHECKING:
 12    from sqlglot.expressions.query import CTE
 13
 14
 15@trait
 16class DDL(Selectable):
 17    @property
 18    def ctes(self) -> list[CTE]:
 19        """Returns a list of all the CTEs attached to this statement."""
 20        with_ = self.args.get("with_")
 21        return with_.expressions if with_ else []
 22
 23    @property
 24    def selects(self) -> list[Expr]:
 25        """If this statement contains a query (e.g. a CTAS), this returns the query's projections."""
 26        expression = self.expression
 27        return expression.selects if isinstance(expression, Query) else []
 28
 29    @property
 30    def named_selects(self) -> list[str]:
 31        """
 32        If this statement contains a query (e.g. a CTAS), this returns the output
 33        names of the query's projections.
 34        """
 35        expression = self.expression
 36        return expression.named_selects if isinstance(expression, Query) else []
 37
 38
 39class Create(Expression, DDL):
 40    arg_types = {
 41        "with_": False,
 42        "this": True,
 43        "kind": True,
 44        "expression": False,
 45        "exists": False,
 46        "properties": False,
 47        "replace": False,
 48        "refresh": False,
 49        "unique": False,
 50        "indexes": False,
 51        "no_schema_binding": False,
 52        "begin": False,
 53        "clone": False,
 54        "concurrently": False,
 55        "clustered": False,
 56    }
 57
 58    @property
 59    def kind(self) -> str | None:
 60        kind = self.args.get("kind")
 61        return kind and kind.upper()
 62
 63
 64class SequenceProperties(Expression):
 65    arg_types = {
 66        "increment": False,
 67        "minvalue": False,
 68        "maxvalue": False,
 69        "cache": False,
 70        "start": False,
 71        "owned": False,
 72        "options": False,
 73    }
 74
 75
 76class TriggerProperties(Expression):
 77    arg_types = {
 78        "table": True,
 79        "timing": True,
 80        "events": True,
 81        "execute": True,
 82        "constraint": False,
 83        "referenced_table": False,
 84        "deferrable": False,
 85        "initially": False,
 86        "referencing": False,
 87        "for_each": False,
 88        "when": False,
 89    }
 90
 91
 92class TriggerExecute(Expression):
 93    pass
 94
 95
 96class TriggerEvent(Expression):
 97    arg_types = {"this": True, "columns": False}
 98
 99
100class TriggerReferencing(Expression):
101    arg_types = {"old": False, "new": False}
102
103
104class TruncateTable(Expression):
105    arg_types = {
106        "expressions": True,
107        "is_database": False,
108        "exists": False,
109        "only": False,
110        "cluster": False,
111        "identity": False,
112        "option": False,
113        "partition": False,
114    }
115
116
117class Clone(Expression):
118    arg_types = {"this": True, "shallow": False, "copy": False}
119
120
121class Describe(Expression):
122    arg_types = {
123        "this": True,
124        "style": False,
125        "kind": False,
126        "properties": False,
127        "expressions": False,
128        "partition": False,
129        "format": False,
130        "as_json": False,
131    }
132
133
134class Attach(Expression):
135    arg_types = {"this": True, "exists": False, "expressions": False}
136
137
138class Detach(Expression):
139    arg_types = {
140        "this": True,
141        "kind": False,
142        "exists": False,
143        "cluster": False,
144        "permanent": False,
145        "sync": False,
146    }
147
148
149class Install(Expression):
150    arg_types = {"this": True, "from_": False, "force": False}
151
152
153class Summarize(Expression):
154    arg_types = {"this": True, "table": False}
155
156
157class Kill(Expression):
158    arg_types = {"this": True, "kind": False}
159
160
161class Pragma(Expression):
162    pass
163
164
165class Declare(Expression):
166    arg_types = {"expressions": True, "replace": False}
167
168
169class DeclareItem(Expression):
170    arg_types = {"this": True, "kind": False, "default": False}
171
172
173class Set(Expression):
174    arg_types = {"expressions": False, "unset": False, "tag": False}
175
176
177class Heredoc(Expression):
178    arg_types = {"this": True, "tag": False}
179
180
181class SetItem(Expression):
182    arg_types = {
183        "this": False,
184        "expressions": False,
185        "kind": False,
186        "collate": False,  # MySQL SET NAMES statement
187        "global_": False,
188    }
189
190
191class Show(Expression):
192    arg_types = {
193        "this": True,
194        "history": False,
195        "terse": False,
196        "target": False,
197        "offset": False,
198        "starts_with": False,
199        "limit": False,
200        "from_": False,
201        "like": False,
202        "where": False,
203        "db": False,
204        "scope": False,
205        "scope_kind": False,
206        "full": False,
207        "mutex": False,
208        "query": False,
209        "channel": False,
210        "global_": False,
211        "log": False,
212        "position": False,
213        "types": False,
214        "privileges": False,
215        "for_table": False,
216        "for_group": False,
217        "for_user": False,
218        "for_role": False,
219        "into_outfile": False,
220        "json": False,
221        "iceberg": False,
222    }
223
224
225class UserDefinedFunction(Expression):
226    arg_types = {"this": True, "expressions": False, "wrapped": False}
227
228
229class CharacterSet(Expression):
230    arg_types = {"this": True, "default": False}
231
232
233class AlterColumn(Expression):
234    arg_types = {
235        "this": True,
236        "dtype": False,
237        "collate": False,
238        "using": False,
239        "default": False,
240        "drop": False,
241        "comment": False,
242        "allow_null": False,
243        "visible": False,
244        "rename_to": False,
245    }
246
247
248class AlterIndex(Expression):
249    arg_types = {"this": True, "visible": True}
250
251
252class AlterDistStyle(Expression):
253    pass
254
255
256class AlterSortKey(Expression):
257    arg_types = {"this": False, "expressions": False, "compound": False}
258
259
260class AlterSet(Expression):
261    arg_types = {
262        "expressions": False,
263        "option": False,
264        "tablespace": False,
265        "access_method": False,
266        "file_format": False,
267        "copy_options": False,
268        "tag": False,
269        "location": False,
270        "serde": False,
271    }
272
273
274class RenameColumn(Expression):
275    arg_types = {"this": True, "to": True, "exists": False}
276
277
278class AlterRename(Expression):
279    pass
280
281
282class RenameIndex(Expression):
283    arg_types = {"this": True, "to": True}
284
285
286class AlterModifySqlSecurity(Expression):
287    arg_types = {"expressions": True}
288
289
290class SwapTable(Expression):
291    pass
292
293
294class Comment(Expression):
295    arg_types = {
296        "this": True,
297        "kind": True,
298        "expression": True,
299        "exists": False,
300        "materialized": False,
301    }
302
303
304class Comprehension(Expression):
305    arg_types = {
306        "this": True,
307        "expression": True,
308        "position": False,
309        "iterator": True,
310        "condition": False,
311    }
312
313
314class MergeTreeTTLAction(Expression):
315    arg_types = {
316        "this": True,
317        "delete": False,
318        "recompress": False,
319        "to_disk": False,
320        "to_volume": False,
321    }
322
323
324class MergeTreeTTL(Expression):
325    arg_types = {
326        "expressions": True,
327        "where": False,
328        "group": False,
329        "aggregates": False,
330    }
331
332
333class Drop(Expression):
334    arg_types = {
335        "this": False,
336        "kind": False,
337        "expressions": False,
338        "exists": False,
339        "temporary": False,
340        "materialized": False,
341        "cascade": False,
342        "restrict": False,
343        "constraints": False,
344        "purge": False,
345        "cluster": False,
346        "concurrently": False,
347        "sync": False,
348        "iceberg": False,
349    }
350
351    @property
352    def kind(self) -> str | None:
353        kind = self.args.get("kind")
354        return kind and kind.upper()
355
356
357class Command(Expression):
358    arg_types = {"this": True, "expression": False}
359
360
361class Transaction(Expression):
362    arg_types = {"this": False, "modes": False, "mark": False}
363
364
365class Commit(Expression):
366    arg_types = {"chain": False, "this": False, "durability": False}
367
368
369class Rollback(Expression):
370    arg_types = {"savepoint": False, "this": False}
371
372
373class Alter(Expression):
374    arg_types = {
375        "this": False,
376        "kind": True,
377        "actions": True,
378        "exists": False,
379        "only": False,
380        "options": False,
381        "cluster": False,
382        "not_valid": False,
383        "check": False,
384        "cascade": False,
385        "iceberg": False,
386    }
387
388    @property
389    def kind(self) -> str | None:
390        kind = self.args.get("kind")
391        return kind and kind.upper()
392
393    @property
394    def actions(self) -> list[Expr]:
395        return self.args.get("actions") or []
396
397
398class AlterSession(Expression):
399    arg_types = {"expressions": True, "unset": False}
400
401
402class Use(Expression):
403    arg_types = {"this": False, "expressions": False, "kind": False}
404
405
406class NextValueFor(Expression, Func):
407    arg_types = {"this": True, "order": False}
408
409
410class Execute(Expression):
411    arg_types = {"this": True, "expressions": False}
412
413    @property
414    def name(self) -> str:
415        return self.this.name
416
417
418class ExecuteSql(Execute):
419    pass