Feature/vrf

gso/migrations/versions/2024-11-27_28c1723c6a00_add_vrf_workflows.py

+"""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"
+    },
+{
+        "name": "terminate_vrf",
+        "target": "TERMINATE",
+        "description": "Terminate a VRF subscription",
+        "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"])