"""Remove reference to subdir

Revision ID: 53e2ac668296
Revises: ed01e313d3cb
Create Date: 2017-03-28 15:01:31.073382

"""

# revision identifiers, used by Alembic.
import json

import logging
from alembic.script.revision import RevisionError
from alembic.util import CommandError
from alembic import op as original_op
from data.migrations.progress import ProgressWrapper
import sqlalchemy as sa
from sqlalchemy.dialects import mysql

revision = '53e2ac668296'
down_revision = 'ed01e313d3cb'

log = logging.getLogger(__name__)


def run_migration(migrate_function, progress_reporter):
  op = ProgressWrapper(original_op, progress_reporter)
  conn = op.get_bind()
  triggers = conn.execute("SELECT id, config FROM repositorybuildtrigger")
  for trigger in triggers:
    config = json.dumps(migrate_function(json.loads(trigger[1])))
    try:
      conn.execute("UPDATE repositorybuildtrigger SET config=%s WHERE id=%s", config, trigger[0])
    except(RevisionError, CommandError) as e:
      log.warning("Failed to update build trigger %s with exception: ", trigger[0], e)


def upgrade(tables, tester, progress_reporter):
  run_migration(delete_subdir, progress_reporter)


def downgrade(tables, tester, progress_reporter):
  run_migration(add_subdir, progress_reporter)


def delete_subdir(config):
    """ Remove subdir from config """
    if not config:
        return config
    if 'subdir' in config:
        del config['subdir']

    return config


def add_subdir(config):
    """ Add subdir back into config """
    if not config:
        return config
    if 'context' in config:
        config['subdir'] = config['context']

    return config