"""

Revision ID: 0315_document_download_count
Revises: 0314_populate_email_access
Create Date: 2020-02-12 14:19:18.066425

"""

import sqlalchemy as sa
from alembic import op

revision = "0315_document_download_count"
down_revision = "0314_populate_email_access"


def upgrade():
    op.add_column(
        "notifications",
        sa.Column("document_download_count", sa.Integer(), nullable=True),
    )
    op.add_column(
        "notification_history",
        sa.Column("document_download_count", sa.Integer(), nullable=True),
    )


def downgrade():
    op.drop_column("notifications", "document_download_count")
    op.drop_column("notification_history", "document_download_count")