diff --git a/gso/migrations/versions/2025-06-20_24858fd1d805_add_modify_note_workflow_to_existing_.py b/gso/migrations/versions/2025-06-20_24858fd1d805_add_modify_note_workflow_to_existing_.py new file mode 100644 index 0000000000000000000000000000000000000000..7414647ef21f79530bef8db6e0f022605f218c74 --- /dev/null +++ b/gso/migrations/versions/2025-06-20_24858fd1d805_add_modify_note_workflow_to_existing_.py @@ -0,0 +1,70 @@ +"""Add modify note workflow to existing products. + +Revision ID: 24858fd1d805 +Revises: 550e3aebc1c5 +Create Date: 2025-06-20 10:51:57.321841 + +""" +import sqlalchemy as sa +from alembic import op + +# revision identifiers, used by Alembic. +revision = '24858fd1d805' +down_revision = '550e3aebc1c5' +branch_labels = None +depends_on = None + + +from orchestrator.migrations.helpers import create_workflow, delete_workflow, add_products_to_workflow_by_product_tag, \ + remove_products_from_workflow_by_product_tag + +product_tags = [ + "ER", + "COP", + "EP", + "G_IP", + "G_PLUS", + "IAS", + "IMP_ER", + "IMP_COP", + "IMP_EP", + "IMP_G_IP", + "IMP_G_PLUS", + "IMP_IAS", + "IMP_IP_TRUNK", + "IMP_LSI", + "IMP_LHC", + "IMP_OFFICE_RTR", + "IMPORTED_OPENGEAR", + "IMP_RE_LHCONE", + "IMP_RE_PEER", + "IMP_RTR", + "IMP_SITE", + "IMP_SPOP_SWITCH", + "IMP_SWITCH", + "IPTRUNK", + "LSI", + "LHC", + "OFFICE_ROUTER", + "OPENGEAR", + "POP_VLAN", + "RE_LHCONE", + "RE_PEER", + "RTR", + "SITE", + "Super_POP_SWITCH", + "SWITCH", + "VRF", +] + + +def upgrade() -> None: + conn = op.get_bind() + for product in product_tags: + add_products_to_workflow_by_product_tag(conn, "modify_note", product) + + +def downgrade() -> None: + conn = op.get_bind() + for product in product_tags: + remove_products_from_workflow_by_product_tag(conn, "modify_note", product)