notifications-api/migrations/versions/0288_add_go_live_user.py

"""

Revision ID: 0288_add_go_live_user
Revises: 0287_drop_branding_domains
Create Date: 2019-04-15 16:50:22.275673

"""

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

revision = "0288_add_go_live_user"
down_revision = "0287_drop_branding_domains"


def upgrade():
    op.add_column("services", sa.Column("go_live_at", sa.DateTime(), nullable=True))
    op.add_column(
        "services",
        sa.Column("go_live_user_id", postgresql.UUID(as_uuid=True), nullable=True),
    )
    op.create_foreign_key(
        "fk_services_go_live_user", "services", "users", ["go_live_user_id"], ["id"]
    )
    op.add_column(
        "services_history", sa.Column("go_live_at", sa.DateTime(), nullable=True)
    )
    op.add_column(
        "services_history",
        sa.Column("go_live_user_id", postgresql.UUID(as_uuid=True), nullable=True),
    )


def downgrade():
    op.drop_column("services_history", "go_live_user_id")
    op.drop_column("services_history", "go_live_at")
    op.drop_constraint("fk_services_go_live_user", "services", type_="foreignkey")
    op.drop_column("services", "go_live_user_id")
    op.drop_column("services", "go_live_at")