mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-15 01:32:20 -05:00
58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
|
"""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'
|
||
|
|
|
||
|
|
from alembic import op
|
||
|
|
import sqlalchemy as sa
|
||
|
|
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')")
|
||
|
|
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')
|