From 9684b0c456a32baa8abc07f2548c37e0f80a80fe Mon Sep 17 00:00:00 2001
From: Karel van Klink <karel.vanklink@geant.org>
Date: Wed, 27 Mar 2024 10:32:19 +0100
Subject: [PATCH] add db migration
---
...7a_modify_connection_streategy_workflow.py | 6 +--
...289c0_remove_subscription_cancellation_.py | 51 +++++++++++++++++++
2 files changed, 54 insertions(+), 3 deletions(-)
create mode 100644 gso/migrations/versions/2024-03-27_4ec89ab289c0_remove_subscription_cancellation_.py
diff --git a/gso/migrations/versions/2024-03-21_a2cd3f2e6d7a_modify_connection_streategy_workflow.py b/gso/migrations/versions/2024-03-21_a2cd3f2e6d7a_modify_connection_streategy_workflow.py
index b401e07b..d15621b8 100644
--- a/gso/migrations/versions/2024-03-21_a2cd3f2e6d7a_modify_connection_streategy_workflow.py
+++ b/gso/migrations/versions/2024-03-21_a2cd3f2e6d7a_modify_connection_streategy_workflow.py
@@ -1,4 +1,4 @@
-"""Modify connection streategy workflow..
+"""Modify connection strategy workflow.
Revision ID: a2cd3f2e6d7a
Revises:
@@ -10,9 +10,9 @@ from alembic import op
# revision identifiers, used by Alembic.
revision = 'a2cd3f2e6d7a'
-down_revision = None
+down_revision = 'd61c0f92da1e'
branch_labels = None
-depends_on = 'd61c0f92da1e'
+depends_on = None
from orchestrator.migrations.helpers import create_workflow, delete_workflow
diff --git a/gso/migrations/versions/2024-03-27_4ec89ab289c0_remove_subscription_cancellation_.py b/gso/migrations/versions/2024-03-27_4ec89ab289c0_remove_subscription_cancellation_.py
new file mode 100644
index 00000000..d9ee3856
--- /dev/null
+++ b/gso/migrations/versions/2024-03-27_4ec89ab289c0_remove_subscription_cancellation_.py
@@ -0,0 +1,51 @@
+"""remove subscription cancellation workflow.
+
+Revision ID: 4ec89ab289c0
+Revises:
+Create Date: 2024-03-27 10:21:08.539591
+
+"""
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = '4ec89ab289c0'
+down_revision = '734e36a3e70b'
+branch_labels = None
+depends_on = None
+
+
+from orchestrator.migrations.helpers import create_workflow, delete_workflow
+
+old_workflows = [
+ {
+ "name": "cancel_subscription",
+ "target": "TERMINATE",
+ "description": "Cancel a subscription",
+ "product_type": "Iptrunk"
+ },
+ {
+ "name": "cancel_subscription",
+ "target": "TERMINATE",
+ "description": "Cancel a subscription",
+ "product_type": "Router"
+ },
+ {
+ "name": "cancel_subscription",
+ "target": "TERMINATE",
+ "description": "Cancel a subscription",
+ "product_type": "Site"
+ }
+]
+
+
+def upgrade() -> None:
+ conn = op.get_bind()
+ for workflow in old_workflows:
+ delete_workflow(conn, workflow["name"])
+
+
+def downgrade() -> None:
+ conn = op.get_bind()
+ for workflow in old_workflows:
+ create_workflow(conn, workflow)
--
GitLab