"""Add an index on ssh_search_key

Revision ID: 0a8f99c161e2
Revises: ba538b2648b7
Create Date: 2018-10-04 10:49:44.739141

"""

# revision identifiers, used by Alembic.
revision = "0a8f99c161e2"
down_revision = "ba538b2648b7"

from alembic import op
import sqlalchemy as sa


def upgrade():
    """ Creates an index on ssh_search_key in the deploykeys table.
    """
    op.alter_column(
        "deploykeys",
        "ssh_search_key",
        nullable=False,
        type_=sa.String(length=60),
    )
    op.create_index(
        op.f("ix_deploykeys_deploykeys_ssh_search_key"),
        "deploykeys",
        ["ssh_search_key"],
        unique=True,
    )


def downgrade():
    """ Drop index on ssh_search_key in the deploykeys table.
    """
    op.drop_index(
        op.f("ix_deploykeys_deploykeys_ssh_search_key"),
        table_name="deploykeys",
    )
    op.alter_column(
        "deploykeys", "ssh_search_key", nullable=False, type_=sa.Text
    )