"""enrich_client_users

Revision ID: a2b3c4d5e6f7
Revises: f1a2b3c4d5e6
Create Date: 2026-04-14 00:00:00.000000

"""

from alembic import op
import sqlalchemy as sa

# revision identifiers, used by Alembic.
revision = "a2b3c4d5e6f7"
down_revision = "f1a2b3c4d5e6"
branch_labels = None
depends_on = None


def upgrade():
    with op.batch_alter_table("client_users") as batch_op:
        batch_op.add_column(sa.Column("phone", sa.String(20), nullable=True))
        batch_op.add_column(sa.Column("whatsapp", sa.String(20), nullable=True))
        batch_op.add_column(sa.Column("cpf", sa.String(14), nullable=True))
        batch_op.add_column(sa.Column("birth_date", sa.Date, nullable=True))
        batch_op.add_column(sa.Column("address_street", sa.String(200), nullable=True))
        batch_op.add_column(sa.Column("address_number", sa.String(20), nullable=True))
        batch_op.add_column(
            sa.Column("address_complement", sa.String(100), nullable=True)
        )
        batch_op.add_column(
            sa.Column("address_neighborhood", sa.String(100), nullable=True)
        )
        batch_op.add_column(sa.Column("address_city", sa.String(100), nullable=True))
        batch_op.add_column(sa.Column("address_state", sa.String(2), nullable=True))
        batch_op.add_column(sa.Column("address_zip", sa.String(9), nullable=True))
        batch_op.add_column(sa.Column("notes", sa.Text, nullable=True))


def downgrade():
    with op.batch_alter_table("client_users") as batch_op:
        batch_op.drop_column("notes")
        batch_op.drop_column("address_zip")
        batch_op.drop_column("address_state")
        batch_op.drop_column("address_city")
        batch_op.drop_column("address_neighborhood")
        batch_op.drop_column("address_complement")
        batch_op.drop_column("address_number")
        batch_op.drop_column("address_street")
        batch_op.drop_column("birth_date")
        batch_op.drop_column("cpf")
        batch_op.drop_column("whatsapp")
        batch_op.drop_column("phone")