notifications-api/migrations/versions/0136_user_mobile_nullable.py

"""

Revision ID: 0136_user_mobile_nullable
Revises: 0135_stats_template_usage
Create Date: 2017-11-08 11:49:05.773974

"""

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

revision = "0136_user_mobile_nullable"
down_revision = "0135_stats_template_usage"


def upgrade():
    op.alter_column("users", "mobile_number", nullable=True)

    op.create_check_constraint(
        "ck_users_mobile_or_email_auth",
        "users",
        "auth_type = 'email_auth' or mobile_number is not null",
    )


def downgrade():
    op.alter_column("users", "mobile_number", nullable=False)
    op.drop_constraint("ck_users_mobile_or_email_auth", "users")