diff --git a/gso/migrations/versions/2025-05-02_465008ed496e_update_ias_flavor_names.py b/gso/migrations/versions/2025-05-02_465008ed496e_update_ias_flavor_names.py new file mode 100644 index 0000000000000000000000000000000000000000..7838e269e5f0e0eee1320ffdeb4e39044c6c077e --- /dev/null +++ b/gso/migrations/versions/2025-05-02_465008ed496e_update_ias_flavor_names.py @@ -0,0 +1,44 @@ +"""Update IAS flavor names. + +Revision ID: 465008ed496e +Revises: a3177c5f9641 +Create Date: 2025-05-02 13:54:40.520906 + +""" +import sqlalchemy as sa +from alembic import op + +# revision identifiers, used by Alembic. +revision = '465008ed496e' +down_revision = 'a3177c5f9641' +branch_labels = None +depends_on = None + + +def upgrade() -> None: + conn = op.get_bind() + conn.execute(sa.text(""" +UPDATE subscription_instance_values +SET value = 'IASPS_OPTOUT' +WHERE value = 'IASPS Opt-OUT' + """)) + conn.execute(sa.text(""" +UPDATE subscription_instance_values +SET value = 'IASPS_OPTIN' +WHERE value = 'IASPS Opt-IN' + """)) + + +def downgrade() -> None: + conn = op.get_bind() + conn.execute(sa.text(""" +UPDATE subscription_instance_values +SET value = 'IASPS Opt-IN' +WHERE value = 'IASPS_OPTIN' + """)) + conn.execute(sa.text(""" +UPDATE subscription_instance_values +SET value = 'IASPS Opt-OUT' +WHERE value = 'IASPS_OPTOUT' + """)) + diff --git a/gso/products/product_blocks/ias.py b/gso/products/product_blocks/ias.py index f57d76fcb8541369b0381e24f09f0caaa2790090..3584420aa78f728a88de2b20c01b64f10b2e2224 100644 --- a/gso/products/product_blocks/ias.py +++ b/gso/products/product_blocks/ias.py @@ -14,8 +14,8 @@ from gso.products.product_blocks.l3_core_service import ( class IASFlavor(strEnum): """IAS flavors.""" - IAS_PS_OPT_OUT = "IASPS Opt-OUT" - IAS_PS_OPT_IN = "IASPS Opt-IN" + IAS_PS_OPT_OUT = "IASPS_OPTOUT" + IAS_PS_OPT_IN = "IASPS_OPTIN" IASGWS = "IASGWS"