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 MacroOverloads(Expression):
230    arg_types = {"expressions": True}
231
232
233class MacroOverload(Expression):
234    arg_types = {"this": True, "expressions": False, "is_table": False}
235
236
237class CharacterSet(Expression):
238    arg_types = {"this": True, "default": False}
239
240
241class AlterColumn(Expression):
242    arg_types = {
243        "this": True,
244        "dtype": False,
245        "collate": False,
246        "using": False,
247        "default": False,
248        "drop": False,
249        "comment": False,
250        "allow_null": False,
251        "visible": False,
252        "rename_to": False,
253    }
254
255
256class ModifyColumn(Expression):
257    arg_types = {"this": True, "rename_from": False}
258
259
260class AlterIndex(Expression):
261    arg_types = {"this": True, "visible": True}
262
263
264class AlterDistStyle(Expression):
265    pass
266
267
268class AlterSortKey(Expression):
269    arg_types = {"this": False, "expressions": False, "compound": False}
270
271
272class AlterSet(Expression):
273    arg_types = {
274        "expressions": False,
275        "option": False,
276        "tablespace": False,
277        "access_method": False,
278        "file_format": False,
279        "copy_options": False,
280        "tag": False,
281        "location": False,
282        "serde": False,
283    }
284
285
286class RenameColumn(Expression):
287    arg_types = {"this": True, "to": True, "exists": False}
288
289
290class AlterRename(Expression):
291    pass
292
293
294class RenameIndex(Expression):
295    arg_types = {"this": True, "to": True}
296
297
298class AlterModifySqlSecurity(Expression):
299    arg_types = {"expressions": True}
300
301
302class SwapTable(Expression):
303    pass
304
305
306class Comment(Expression):
307    arg_types = {
308        "this": True,
309        "kind": True,
310        "expression": True,
311        "exists": False,
312        "materialized": False,
313    }
314
315
316class Comprehension(Expression):
317    arg_types = {
318        "this": True,
319        "expression": True,
320        "position": False,
321        "iterator": True,
322        "condition": False,
323    }
324
325
326class MergeTreeTTLAction(Expression):
327    arg_types = {
328        "this": True,
329        "delete": False,
330        "recompress": False,
331        "to_disk": False,
332        "to_volume": False,
333    }
334
335
336class MergeTreeTTL(Expression):
337    arg_types = {
338        "expressions": True,
339        "where": False,
340        "group": False,
341        "aggregates": False,
342    }
343
344
345class Drop(Expression):
346    arg_types = {
347        "this": False,
348        "kind": False,
349        "expressions": False,
350        "exists": False,
351        "temporary": False,
352        "materialized": False,
353        "cascade": False,
354        "restrict": False,
355        "constraints": False,
356        "purge": False,
357        "cluster": False,
358        "concurrently": False,
359        "sync": False,
360        "iceberg": False,
361    }
362
363    @property
364    def kind(self) -> str | None:
365        kind = self.args.get("kind")
366        return kind and kind.upper()
367
368
369class DropPrimaryKey(Expression):
370    arg_types = {}
371
372
373class Command(Expression):
374    arg_types = {"this": True, "expression": False}
375
376
377class Transaction(Expression):
378    arg_types = {"this": False, "modes": False, "mark": False}
379
380
381class Commit(Expression):
382    arg_types = {"chain": False, "this": False, "durability": False}
383
384
385class Rollback(Expression):
386    arg_types = {"savepoint": False, "this": False}
387
388
389class Alter(Expression):
390    arg_types = {
391        "this": False,
392        "kind": True,
393        "actions": True,
394        "exists": False,
395        "only": False,
396        "options": False,
397        "cluster": False,
398        "not_valid": False,
399        "check": False,
400        "cascade": False,
401        "iceberg": False,
402    }
403
404    @property
405    def kind(self) -> str | None:
406        kind = self.args.get("kind")
407        return kind and kind.upper()
408
409    @property
410    def actions(self) -> list[Expr]:
411        return self.args.get("actions") or []
412
413
414class AlterSession(Expression):
415    arg_types = {"expressions": True, "unset": False}
416
417
418class Use(Expression):
419    arg_types = {"this": False, "expressions": False, "kind": False}
420
421
422class NextValueFor(Expression, Func):
423    arg_types = {"this": True, "order": False}
424
425
426class Execute(Expression):
427    arg_types = {"this": True, "expressions": False}
428
429    @property
430    def name(self) -> str:
431        return self.this.name
432
433
434class ExecuteSql(Execute):
435    pass