From ab2542f9bb38133806982df77bf1fc25fa537f7f Mon Sep 17 00:00:00 2001 From: Karel van Klink <karel.vanklink@geant.org> Date: Fri, 2 May 2025 14:10:24 +0200 Subject: [PATCH] Update IAS flavor labels --- ...02_465008ed496e_update_ias_flavor_names.py | 44 +++++++++++++++++++ gso/products/product_blocks/ias.py | 4 +- 2 files changed, 46 insertions(+), 2 deletions(-) create mode 100644 gso/migrations/versions/2025-05-02_465008ed496e_update_ias_flavor_names.py 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 000000000..7838e269e --- /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 f57d76fcb..3584420aa 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" -- GitLab