"""add citation_status and date_completed to global_literature

Revision ID: c6d7e8f9g0h1
Revises: b5c9d3e4f5g6
Create Date: 2026-07-07 20:00:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa


revision: str = 'c6d7e8f9g0h1'
down_revision: Union[str, None] = 'b5c9d3e4f5g6'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.add_column('global_literature', sa.Column('citation_status', sa.String(20), nullable=True))
    op.add_column('global_literature', sa.Column('date_completed', sa.DateTime(timezone=True), nullable=True))


def downgrade() -> None:
    op.drop_column('global_literature', 'date_completed')
    op.drop_column('global_literature', 'citation_status')
