rag-gateway/alembic/versions/abc6bb9129ed_user_app_update.py

"""user app update

Revision ID: abc6bb9129ed
Revises: d4c8f204280f
Create Date: 2024-12-09 16:57:06.559644

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import mysql

# revision identifiers, used by Alembic.
revision: str = 'abc6bb9129ed'
down_revision: Union[str, None] = 'd4c8f204280f'
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('user_app', sa.Column('access_token', sa.String(length=1000), nullable=True))
    op.add_column('user_app', sa.Column('refresh_token', sa.String(length=1000), nullable=True))
    op.add_column('user_app', sa.Column('token_at', sa.DateTime(), nullable=True))
    op.alter_column('user_app', 'app_type',
               existing_type=mysql.INTEGER(),
               type_=sa.String(length=16),
               existing_nullable=True)
    op.drop_index('ix_user_app_username', table_name='user_app')
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_index('ix_user_app_username', 'user_app', ['username'], unique=True)
    op.alter_column('user_app', 'app_type',
               existing_type=sa.String(length=16),
               type_=mysql.INTEGER(),
               existing_nullable=True)
    op.drop_column('user_app', 'token_at')
    op.drop_column('user_app', 'refresh_token')
    op.drop_column('user_app', 'access_token')
    # ### end Alembic commands ###