"""Add status and headstone_photo_s3_key columns to records

Revision ID: 0004a
Revises: 0004
Create Date: 2024-01-01 00:03:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa

revision: str = "0004a"
down_revision: Union[str, None] = "0004"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.add_column('records', sa.Column('status', sa.String(20), nullable=False, server_default='active'))
    op.add_column('records', sa.Column('headstone_photo_s3_key', sa.Text, nullable=True))


def downgrade() -> None:
    op.drop_column('records', 'headstone_photo_s3_key')
    op.drop_column('records', 'status')