Edit on GitHub

Add flag that controls whether the virtual layer's views will be created by the model specified gateway rather than the default gateway.

 1"""Add flag that controls whether the virtual layer's views will be created by the model specified gateway rather than the default gateway."""
 2
 3from sqlglot import exp
 4
 5
 6def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
 7    environments_table = "_environments"
 8    if schema:
 9        environments_table = f"{schema}.{environments_table}"
10
11    alter_table_exp = exp.Alter(
12        this=exp.to_table(environments_table),
13        kind="TABLE",
14        actions=[
15            exp.ColumnDef(
16                this=exp.to_column("gateway_managed"),
17                kind=exp.DataType.build("boolean"),
18            )
19        ],
20    )
21    engine_adapter.execute(alter_table_exp)
22
23
24def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
25    environments_table = "_environments"
26    if schema:
27        environments_table = f"{schema}.{environments_table}"
28
29    engine_adapter.update_table(
30        environments_table,
31        {"gateway_managed": False},
32        where=exp.true(),
33    )
def migrate_schemas(engine_adapter, schema, **kwargs): 
 7def migrate_schemas(engine_adapter, schema, **kwargs):  # type: ignore
 8    environments_table = "_environments"
 9    if schema:
10        environments_table = f"{schema}.{environments_table}"
11
12    alter_table_exp = exp.Alter(
13        this=exp.to_table(environments_table),
14        kind="TABLE",
15        actions=[
16            exp.ColumnDef(
17                this=exp.to_column("gateway_managed"),
18                kind=exp.DataType.build("boolean"),
19            )
20        ],
21    )
22    engine_adapter.execute(alter_table_exp)
def migrate_rows(engine_adapter, schema, **kwargs): 
25def migrate_rows(engine_adapter, schema, **kwargs):  # type: ignore
26    environments_table = "_environments"
27    if schema:
28        environments_table = f"{schema}.{environments_table}"
29
30    engine_adapter.update_table(
31        environments_table,
32        {"gateway_managed": False},
33        where=exp.true(),
34    )