"""Remove email from users

Revision ID: 003
Revises: 002
Create Date: 2024-12-19

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa

revision: str = '003'
down_revision: Union[str, None] = '002'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # Drop unique constraint on email
    op.drop_constraint('users_email_key', 'users', type_='unique')
    # Drop email column
    op.drop_column('users', 'email')


def downgrade() -> None:
    # Add email column back
    op.add_column('users', sa.Column('email', sa.String(255), nullable=True))
    # Add unique constraint back
    op.create_unique_constraint('users_email_key', 'users', ['email'])