"""add_is_confidential_flag_to_projects_table

Revision ID: aa7ad7644aa2
Revises: f1c89330a35d
Create Date: 2021-06-14 12:58:57.973805

"""
from alembic import op
# import sqlalchemy as sa
# from sqlalchemy.dialects import postgresql
from migrations.fixtures.delphi_tables.views import DelphiViews
from migrations.fixtures.delphi_tables.v1.triggers.project_trigger import ProjectTrigger
from migrations.fixtures.delphi_tables.v1.data_migrator import DataMigrator
from migrations.fixtures.delphi_tables.v1.tables import DelphiTables
from migrations.fixtures.delphi_tables.v1.trigger_migrator import TriggerMigrator
# revision identifiers, used by Alembic.
revision = 'aa7ad7644aa2'
down_revision = 'f1c89330a35d'
branch_labels = None
depends_on = None


def upgrade():
    bind = op.get_bind()
    delphi_views = DelphiViews(bind)
    delphi_views.drop()
    delphi_views.create()

    delphi_tables = DelphiTables(bind)

    delphi_tables.drop()
    delphi_tables.create()
    data_migrator = DataMigrator(bind)
    data_migrator.execute()
    delphi_tables.reindex()

    TriggerMigrator(bind, ProjectTrigger()).rollback()
    TriggerMigrator(bind, ProjectTrigger()).upgrade()