"""

Revision ID: 0162_remove_org
Revises: 0161_email_branding
Create Date: 2018-02-06 17:08:11.879844

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

revision = '0162_remove_org'
down_revision = '0161_email_branding'


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('services', 'organisation_id')
    op.drop_column('services_history', 'organisation_id')

    op.drop_table('organisation')

    op.alter_column('service_email_branding', 'email_branding_id', nullable=False)
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('services_history', sa.Column('organisation_id', postgresql.UUID(), autoincrement=False, nullable=True))  # noqa
    op.add_column('services', sa.Column('organisation_id', postgresql.UUID(), autoincrement=False, nullable=True))

    op.create_table(
        'organisation',
        sa.Column('id', postgresql.UUID(), autoincrement=False, nullable=False),
        sa.Column('colour', sa.VARCHAR(length=7), autoincrement=False, nullable=True),
        sa.Column('logo', sa.VARCHAR(length=255), autoincrement=False, nullable=True),
        sa.Column('name', sa.VARCHAR(length=255), autoincrement=False, nullable=True),
        sa.PrimaryKeyConstraint('id', name='organisation_pkey')
    )

    op.create_index('ix_services_history_organisation_id', 'services_history', ['organisation_id'], unique=False)
    op.create_foreign_key('services_organisation_id_fkey', 'services', 'organisation', ['organisation_id'], ['id'])
    op.create_index('ix_services_organisation_id', 'services', ['organisation_id'], unique=False)

    op.alter_column('service_email_branding', 'email_branding_id', nullable=True)

remove organisation from db