|
5 | 5 | Create Date: 2026-03-10 18:29:54.555074 |
6 | 6 |
|
7 | 7 | """ |
| 8 | + |
8 | 9 | from typing import Sequence, Union |
9 | 10 |
|
10 | 11 | import sqlalchemy as sa |
|
13 | 14 | from alembic import op |
14 | 15 |
|
15 | 16 | # revision identifiers, used by Alembic. |
16 | | -revision: str = '67608331ee8a' |
17 | | -down_revision: Union[str, Sequence[str], None] = 'add_modelled_policies' |
| 17 | +revision: str = "67608331ee8a" |
| 18 | +down_revision: Union[str, Sequence[str], None] = "add_modelled_policies" |
18 | 19 | branch_labels: Union[str, Sequence[str], None] = None |
19 | 20 | depends_on: Union[str, Sequence[str], None] = None |
20 | 21 |
|
21 | 22 |
|
22 | 23 | def upgrade() -> None: |
23 | 24 | """Upgrade schema.""" |
24 | | - op.create_table('parameter_nodes', |
25 | | - sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False), |
26 | | - sa.Column('label', sqlmodel.sql.sqltypes.AutoString(), nullable=True), |
27 | | - sa.Column('description', sqlmodel.sql.sqltypes.AutoString(), nullable=True), |
28 | | - sa.Column('tax_benefit_model_version_id', sa.Uuid(), nullable=False), |
29 | | - sa.Column('id', sa.Uuid(), nullable=False), |
30 | | - sa.Column('created_at', sa.DateTime(), nullable=False), |
31 | | - sa.ForeignKeyConstraint(['tax_benefit_model_version_id'], ['tax_benefit_model_versions.id'], ), |
32 | | - sa.PrimaryKeyConstraint('id') |
| 25 | + op.create_table( |
| 26 | + "parameter_nodes", |
| 27 | + sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False), |
| 28 | + sa.Column("label", sqlmodel.sql.sqltypes.AutoString(), nullable=True), |
| 29 | + sa.Column("description", sqlmodel.sql.sqltypes.AutoString(), nullable=True), |
| 30 | + sa.Column("tax_benefit_model_version_id", sa.Uuid(), nullable=False), |
| 31 | + sa.Column("id", sa.Uuid(), nullable=False), |
| 32 | + sa.Column("created_at", sa.DateTime(), nullable=False), |
| 33 | + sa.ForeignKeyConstraint( |
| 34 | + ["tax_benefit_model_version_id"], |
| 35 | + ["tax_benefit_model_versions.id"], |
| 36 | + ), |
| 37 | + sa.PrimaryKeyConstraint("id"), |
33 | 38 | ) |
34 | 39 |
|
35 | 40 |
|
36 | 41 | def downgrade() -> None: |
37 | 42 | """Downgrade schema.""" |
38 | | - op.drop_table('parameter_nodes') |
| 43 | + op.drop_table("parameter_nodes") |
0 commit comments