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