From aa70a732cc34929114a3a0b476ec3248dd402e1a Mon Sep 17 00:00:00 2001 From: Neda Moeini <neda.moeini@geant.org> Date: Wed, 27 Nov 2024 15:14:26 +0100 Subject: [PATCH] Make migration for modification WF --- ...24-11-27_28c1723c6a00_add_vrf_workflows.py | 45 +++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 gso/migrations/versions/2024-11-27_28c1723c6a00_add_vrf_workflows.py diff --git a/gso/migrations/versions/2024-11-27_28c1723c6a00_add_vrf_workflows.py b/gso/migrations/versions/2024-11-27_28c1723c6a00_add_vrf_workflows.py new file mode 100644 index 00000000..1457aecf --- /dev/null +++ b/gso/migrations/versions/2024-11-27_28c1723c6a00_add_vrf_workflows.py @@ -0,0 +1,45 @@ +"""Add VRF workflows.. + +Revision ID: 28c1723c6a00 +Revises: 6103f83731bb +Create Date: 2024-11-27 14:44:57.271584 + +""" +import sqlalchemy as sa +from alembic import op + +# revision identifiers, used by Alembic. +revision = '28c1723c6a00' +down_revision = '6103f83731bb' +branch_labels = None +depends_on = None + + +from orchestrator.migrations.helpers import create_workflow, delete_workflow + +new_workflows = [ + { + "name": "create_vrf", + "target": "CREATE", + "description": "Create VRF", + "product_type": "VRF" + }, + { + "name": "modify_vrf_router_list", + "target": "MODIFY", + "description": "Modify VRF router list", + "product_type": "VRF" + } +] + + +def upgrade() -> None: + conn = op.get_bind() + for workflow in new_workflows: + create_workflow(conn, workflow) + + +def downgrade() -> None: + conn = op.get_bind() + for workflow in new_workflows: + delete_workflow(conn, workflow["name"]) -- GitLab