Edit on GitHub

Add support for environment catalog name override.

 1"""Add support for environment catalog name override."""
 2
 3from sqlglot import exp
 4
 5
 6def migrate(state_sync, **kwargs):  # type: ignore
 7    engine_adapter = state_sync.engine_adapter
 8    environments_table = "_environments"
 9    if state_sync.schema:
10        environments_table = f"{state_sync.schema}.{environments_table}"
11
12    alter_table_exp = exp.AlterTable(
13        this=exp.to_table(environments_table),
14        actions=[
15            exp.ColumnDef(
16                this=exp.to_column("catalog_name_override"),
17                kind=exp.DataType.build("text"),
18            )
19        ],
20    )
21    engine_adapter.execute(alter_table_exp)
def migrate(state_sync, **kwargs): 
 7def migrate(state_sync, **kwargs):  # type: ignore
 8    engine_adapter = state_sync.engine_adapter
 9    environments_table = "_environments"
10    if state_sync.schema:
11        environments_table = f"{state_sync.schema}.{environments_table}"
12
13    alter_table_exp = exp.AlterTable(
14        this=exp.to_table(environments_table),
15        actions=[
16            exp.ColumnDef(
17                this=exp.to_column("catalog_name_override"),
18                kind=exp.DataType.build("text"),
19            )
20        ],
21    )
22    engine_adapter.execute(alter_table_exp)