"""store canonical waterbody detail facts

Revision ID: 0020
Revises: 0019
"""

from alembic import op
import sqlalchemy as sa


revision = "0020"
down_revision = "0019"
branch_labels = None
depends_on = None


def upgrade() -> None:
    for name in ("source_aliases", "source_fish_species", "source_image_urls", "source_point_urls"):
        op.add_column("waterbody", sa.Column(name, sa.JSON(), nullable=True))


def downgrade() -> None:
    for name in ("source_point_urls", "source_image_urls", "source_fish_species", "source_aliases"):
        op.drop_column("waterbody", name)