"""Create edit date user

Revision ID: 1dc57c110497
Revises: 4df3ee92e98f
Create Date: 2019-06-03 20:09:09.746009

"""
from datetime import datetime

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = '1dc57c110497'
down_revision = '4df3ee92e98f'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    bind = op.get_bind()
    session = sa.orm.Session(bind=bind)

    bind.execute('''
        INSERT INTO "HistoryUser" 
        (id, email, edit_date) 
        VALUES ('system', 'system@example.com', {}) 
    '''.format(datetime.max))

    op.add_column(
        'Campaign',
        sa.Column(
            'create_date',
            sa.DateTime(),
            nullable=False,
            server_default='1970-01-01 00:00:00'
        )
    )
    op.alter_column('Campaign', 'create_date', server_default=None)
    op.add_column(
        'Campaign',
        sa.Column(
            'create_user_id',
            sa.String(),
            nullable=False,
            server_default='system'
        )
    )
    op.alter_column('Campaign', 'create_user_id', server_default=None)
    op.add_column(
        'Campaign',
        sa.Column(
            'edit_date',
            sa.DateTime(),
            nullable=False,
            server_default='1970-01-01 00:00:00'
        )
    )
    op.alter_column('Campaign', 'edit_date', server_default=None)
    op.add_column(
        'Campaign',
        sa.Column(
            'edit_user_id',
            sa.String(),
            nullable=False,
            server_default='system'
        )
    )
    op.alter_column('Campaign', 'edit_user_id', server_default=None)
    op.create_foreign_key(
        op.f('fk_Campaign_create_user_id_HistoryUser'),
        'Campaign',
        'HistoryUser',
        ['create_user_id'],
        ['id']
    )
    op.create_foreign_key(
        op.f('fk_Campaign_edit_user_id_HistoryUser'),
        'Campaign',
        'HistoryUser',
        ['edit_user_id'],
        ['id']
    )
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_constraint(
        op.f('fk_Campaign_edit_user_id_HistoryUser'),
        'Campaign',
        type_='foreignkey'
    )
    op.drop_constraint(
        op.f('fk_Campaign_create_user_id_HistoryUser'),
        'Campaign',
        type_='foreignkey'
    )
    op.drop_column('Campaign', 'edit_user_id')
    op.drop_column('Campaign', 'edit_date')
    op.drop_column('Campaign', 'create_user_id')
    op.drop_column('Campaign', 'create_date')
    # ### end Alembic commands ###