From cc9fcb42d95faccfdefacf85d53a881c3733a3b0 Mon Sep 17 00:00:00 2001 From: Rebecca Law Date: Thu, 12 May 2016 09:22:35 +0100 Subject: [PATCH] Migration script to add template version to jobs and notifications. --- .../versions/0013_add_template_version.py | 29 +++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 migrations/versions/0013_add_template_version.py diff --git a/migrations/versions/0013_add_template_version.py b/migrations/versions/0013_add_template_version.py new file mode 100644 index 000000000..69f5a6694 --- /dev/null +++ b/migrations/versions/0013_add_template_version.py @@ -0,0 +1,29 @@ +"""empty message + +Revision ID: 0013_add_template_version +Revises: 0010_events_table +Create Date: 2016-05-11 16:00:51.478012 + +""" + +# revision identifiers, used by Alembic. +revision = '0013_add_template_version' +down_revision = '0010_events_table' + +from alembic import op +import sqlalchemy as sa +from sqlalchemy.dialects import postgresql + +def upgrade(): + op.add_column('jobs', sa.Column('template_version', sa.Integer(), nullable=True)) + op.get_bind() + op.execute('update jobs set template_version = (select version from templates where id = template_id)') + op.add_column('notifications', sa.Column('template_version', sa.Integer(), nullable=True)) + op.execute('update notifications set template_version = (select version from templates where id = template_id)') + op.alter_column('jobs', 'template_version', nullable=False) + op.alter_column('notifications', 'template_version', nullable=False) + + +def downgrade(): + op.drop_column('notifications', 'template_version') + op.drop_column('jobs', 'template_version')