"""
Revision ID: 0223_add_domain_constraint
Revises: 0222_drop_service_branding
Create Date: 2018-08-24 13:36:49.346156
 """
from alembic import op


revision = '0223_add_domain_constraint'
down_revision = '0222_drop_service_branding'


def upgrade():

    op.execute("""
        update
            email_branding
        set
            domain = null
        where
            domain = ''
    """)
    op.create_unique_constraint('uq_email_branding_domain', 'email_branding', ['domain'])


def downgrade():

    op.drop_constraint('uq_email_branding_domain', 'email_branding')