alembic/versions/fcc0580a4769_add_updated_model_planning.py

"""add updated model planning

Revision ID: fcc0580a4769
Revises: 4dd226fee84e
Create Date: 2024-10-08 10:01:41.964473

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = 'fcc0580a4769'
down_revision: Union[str, None] = '4dd226fee84e'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('bot_meta', sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=False))
    op.add_column('bot_meta', sa.Column('updated_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=False))
    op.add_column('planning', sa.Column('planning_name', sa.String(length=200), nullable=False))
    op.add_column('planning', sa.Column('duration', sa.Integer(), nullable=False))
    op.add_column('planning', sa.Column('token_used', sa.Integer(), nullable=False))
    op.add_column('planning', sa.Column('token_planned', sa.Integer(), nullable=False))
    op.add_column('planning', sa.Column('start_date', sa.DateTime(), nullable=False))
    op.add_column('planning', sa.Column('end_date', sa.DateTime(), nullable=False))
    op.add_column('planning', sa.Column('is_activated', sa.Boolean(), nullable=True))
    op.add_column('planning', sa.Column('created_at', sa.DateTime(), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=False))
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column('planning', 'created_at')
    op.drop_column('planning', 'is_activated')
    op.drop_column('planning', 'end_date')
    op.drop_column('planning', 'start_date')
    op.drop_column('planning', 'token_planned')
    op.drop_column('planning', 'token_used')
    op.drop_column('planning', 'duration')
    op.drop_column('planning', 'planning_name')
    op.drop_column('bot_meta', 'updated_at')
    op.drop_column('bot_meta', 'created_at')
    # ### end Alembic commands ###