"""empty message

Revision ID: 8d3967062a0a
Revises: 7a5a058d666d
Create Date: 2020-01-27 19:29:05.261333

"""
import sqlalchemy as sa
from alembic import op

# revision identifiers, used by Alembic.
revision = "8d3967062a0a"
down_revision = "7a5a058d666d"
branch_labels = None
depends_on = None

enum_values = ("SMALL", "BIG", "VERY_BIG")
enum_type = sa.Enum(*enum_values, name="ordersizeenum")


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    enum_type.create(op.get_bind(), checkfirst=False)
    op.add_column(
        "campaign",
        sa.Column(
            "order_size",
            sa.Enum("SMALL", "BIG", "VERY_BIG", name="ordersizeenum"),
            nullable=True,
        ),
    )
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column("campaign", "order_size")
    enum_type.drop(op.get_bind(), checkfirst=False)
    # ### end Alembic commands ###
