"""add_fsm_field

Revision ID: a2016e2b7112
Revises: 8582618f445e
Create Date: 2022-08-17 13:12:52.920147

"""
from alembic import op
import sqlalchemy as sa
import sqlalchemy_fsm
# revision identifiers, used by Alembic.
revision = 'a2016e2b7112'
down_revision = '402e79ec853d'
branch_labels = None
depends_on = None


def upgrade():
    op.add_column('Campaign', sa.Column('approval_status', sqlalchemy_fsm.sqltypes.FSMField(), server_default='Draft',
                                        nullable=False))
    bind = op.get_bind()
    bind.execute(
        '''UPDATE "Campaign" SET approval_status=(
            SELECT name FROM "CampaignStatus" WHERE "Campaign".campaign_status_id = "CampaignStatus".id
        )'''
    )


def downgrade():
    op.drop_column('Campaign', 'approval_status')