blob: 652cdefdb45abdf8faad9b556d24d8b51de77caa (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
from sqlglot import alias, exp
from sqlglot.errors import OptimizeError
from sqlglot.optimizer.scope import traverse_scope
def isolate_table_selects(expression):
for scope in traverse_scope(expression):
if len(scope.selected_sources) == 1:
continue
for (_, source) in scope.selected_sources.values():
if not isinstance(source, exp.Table):
continue
if not source.alias:
raise OptimizeError("Tables require an alias. Run qualify_tables optimization.")
source.replace(
exp.select("*")
.from_(
alias(source.copy(), source.name or source.alias, table=True),
copy=False,
)
.subquery(source.alias, copy=False)
)
return expression
|