# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

"""db: migration step after 95c01895c006 failed to add usk_public_key_idx in alembic step b74907136bc1

Revision ID: 4851d15bc437
Revises: 151b4a4e8c48
Create Date: 2019-11-24 02:51:14.029583

"""

# The following opaque hexadecimal identifiers ("revisions") are used
# by Alembic to track this migration script and its relations to others.
revision = '4851d15bc437'
down_revision = '151b4a4e8c48'
branch_labels = None
depends_on = None

import sqlalchemy as sa
from alembic import op


def upgrade():
    meta = sa.MetaData()
    meta.reflect(bind=op.get_bind())

    if not any(i.name == 'usk_public_key_idx' for i in meta.tables['user_ssh_keys'].indexes):
        with op.batch_alter_table('user_ssh_keys', schema=None) as batch_op:
            batch_op.create_index('usk_public_key_idx', ['public_key'], unique=False)


def downgrade():
    with op.batch_alter_table('user_ssh_keys', schema=None) as batch_op:
        batch_op.drop_index('usk_public_key_idx')
