"""change_edit_date_in_models

Revision ID: 12f2a62e9787
Revises: 33b81e576e3e
Create Date: 2020-04-03 09:35:07.846351

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql


# revision identifiers, used by Alembic.
revision = '12f2a62e9787'
down_revision = '33b81e576e3e'
branch_labels = None
depends_on = None


def upgrade():

    op.add_column('HistoryUser',
                  sa.Column('created_at',
                            postgresql.TIMESTAMP(timezone=True),
                            server_default=sa.func.current_timestamp(),
                            nullable=False))

    op.add_column('CampaignHistory',
                  sa.Column('created_at',
                            postgresql.TIMESTAMP(timezone=True),
                            server_default=sa.func.current_timestamp(),
                            nullable=False))

    bind = op.get_bind()
    bind.execute('''
            UPDATE "HistoryUser" 
            SET created_at = edit_date;
            ALTER TABLE "HistoryUser" 
            DROP COLUMN edit_date;
        ''')
    bind.execute('''
            UPDATE "CampaignHistory" 
            SET created_at = edit_date;
            ALTER TABLE "CampaignHistory" 
            DROP COLUMN edit_date;
        ''')


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    pass
    # ### end Alembic commands ###