From 4817665e8e735f41e534aebaf676d4227c8e6fc3 Mon Sep 17 00:00:00 2001
From: Karel van Klink <karel.vanklink@geant.org>
Date: Wed, 27 Dec 2023 14:57:43 +0100
Subject: [PATCH] add database migration

---
 ...64c6f392c_add_twamp_deployment_workflow.py | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 gso/migrations/versions/2023-12-27_f0764c6f392c_add_twamp_deployment_workflow.py

diff --git a/gso/migrations/versions/2023-12-27_f0764c6f392c_add_twamp_deployment_workflow.py b/gso/migrations/versions/2023-12-27_f0764c6f392c_add_twamp_deployment_workflow.py
new file mode 100644
index 00000000..119cae59
--- /dev/null
+++ b/gso/migrations/versions/2023-12-27_f0764c6f392c_add_twamp_deployment_workflow.py
@@ -0,0 +1,39 @@
+"""Add TWAMP deployment workflow.
+
+Revision ID: f0764c6f392c
+Revises: 815033570ad7
+Create Date: 2023-12-27 14:31:42.285180
+
+"""
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = 'f0764c6f392c'
+down_revision = '815033570ad7'
+branch_labels = None
+depends_on = None
+
+
+from orchestrator.migrations.helpers import create_workflow, delete_workflow
+
+new_workflows = [
+    {
+        "name": "deploy_twamp",
+        "target": "MODIFY",
+        "description": "Deploy TWAMP",
+        "product_type": "Iptrunk"
+    }
+]
+
+
+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