"""add legal_hold_data to users for encrypted PII retention

Revision ID: add_legal_hold_data_to_users
Revises: add_user_soft_delete_fields
Create Date: 2025-12-08 00:00:00.000000

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import JSON


# revision identifiers, used by Alembic.
revision = 'add_legal_hold_data_to_users'
down_revision = '76c7521bc046'  # Revises: create_cookie_consents_table
branch_labels = None
depends_on = None


def upgrade():
    """Add legal_hold_data JSON column for encrypted PII storage"""
    op.add_column('users', sa.Column('legal_hold_data', JSON, nullable=True))


def downgrade():
    """Remove legal_hold_data column"""
    op.drop_column('users', 'legal_hold_data')