"""update created_at and updated_at in agente table

Revision ID: 893cd8ed8c4c
Revises: bd97ab8ba7db
Create Date: 2024-03-06 16:30:00.000000

"""
from alembic import op
import sqlalchemy as sa

revision = '893cd8ed8c4c'
down_revision = 'bd97ab8ba7db'
branch_labels = None
depends_on = None

def upgrade():
    op.execute("UPDATE agente SET created_at = NOW() WHERE created_at IS NULL")
    op.execute("UPDATE agente SET updated_at = NOW() WHERE updated_at IS NULL")
    with op.batch_alter_table('agente', schema=None) as batch_op:
        batch_op.alter_column('created_at',
               existing_type=sa.DATETIME(),
               nullable=False,
               server_default=sa.text('CURRENT_TIMESTAMP'))
        batch_op.alter_column('updated_at',
               existing_type=sa.DATETIME(),
               nullable=False,
               server_default=sa.text('CURRENT_TIMESTAMP'))

def downgrade():
    with op.batch_alter_table('agente', schema=None) as batch_op:
        batch_op.alter_column('updated_at',
               existing_type=sa.DATETIME(),
               nullable=True,
               server_default=None)
        batch_op.alter_column('created_at',
               existing_type=sa.DATETIME(),
               nullable=True,
               server_default=None)