notifications-api/migrations/versions/0046_organisations_and_branding.py

"""empty message

Revision ID: 0046_organisations_and_branding
Revises: 0045_billable_units
Create Date: 2016-08-04 12:00:43.682610

"""

# revision identifiers, used by Alembic.
revision = "0046_organisations_and_branding"
down_revision = "0045_billable_units"

import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects import postgresql


def upgrade():
    op.create_table(
        "branding_type",
        sa.Column("name", sa.String(length=255), nullable=False),
        sa.PrimaryKeyConstraint("name"),
    )
    op.create_table(
        "organisation",
        sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column("colour", sa.String(length=7), nullable=True),
        sa.Column("logo", sa.String(length=255), nullable=True),
        sa.Column("name", sa.String(length=255), nullable=True),
        sa.PrimaryKeyConstraint("id"),
    )

    op.add_column("services", sa.Column("branding", sa.String(length=255)))
    op.add_column(
        "services", sa.Column("organisation_id", postgresql.UUID(as_uuid=True))
    )
    op.add_column("services_history", sa.Column("branding", sa.String(length=255)))
    op.add_column(
        "services_history", sa.Column("organisation_id", postgresql.UUID(as_uuid=True))
    )

    op.execute("INSERT INTO branding_type VALUES ('govuk'), ('org'), ('both')")

    # insert UKVI data as initial test data. hex and crest pulled from alphagov/whitehall
    op.execute(
        """INSERT INTO organisation VALUES (
        '9d25d02d-2915-4e98-874b-974e123e8536',
        '#9325b2',
        'ho_crest_27px_x2.png',
        'UK Visas and Immigration'
    )"""
    )
    op.execute("UPDATE services SET branding='govuk'")
    op.execute("UPDATE services_history SET branding='govuk'")

    op.alter_column("services", "branding", nullable=False)
    op.alter_column("services_history", "branding", nullable=False)

    op.create_index(
        op.f("ix_services_branding"), "services", ["branding"], unique=False
    )
    op.create_index(
        op.f("ix_services_organisation_id"),
        "services",
        ["organisation_id"],
        unique=False,
    )
    op.create_index(
        op.f("ix_services_history_branding"),
        "services_history",
        ["branding"],
        unique=False,
    )
    op.create_index(
        op.f("ix_services_history_organisation_id"),
        "services_history",
        ["organisation_id"],
        unique=False,
    )

    op.create_foreign_key(None, "services", "branding_type", ["branding"], ["name"])
    op.create_foreign_key(None, "services", "organisation", ["organisation_id"], ["id"])


def downgrade():
    op.drop_column("services_history", "organisation_id")
    op.drop_column("services_history", "branding")
    op.drop_column("services", "organisation_id")
    op.drop_column("services", "branding")
    op.drop_table("organisation")
    op.drop_table("branding_type")