rag-gateway/alembic/versions/4b3a7c69ceac_init_table.py

"""init table

Revision ID: 4b3a7c69ceac
Revises: 
Create Date: 2024-12-06 16:12:04.760402

"""
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 = '4b3a7c69ceac'
down_revision: Union[str, None] = None
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.drop_index('ix_user_test_id', table_name='user_test')
    op.drop_index('ix_user_test_username', table_name='user_test')
    op.drop_table('user_test')
    op.alter_column('user', 'ragflow_id',
               existing_type=mysql.VARCHAR(length=32),
               nullable=True)
    op.alter_column('user', 'bisheng_id',
               existing_type=mysql.INTEGER(),
               nullable=True)
    op.alter_column('user', 'status',
               existing_type=mysql.VARCHAR(length=10),
               nullable=True,
               existing_server_default=sa.text("'1'"))
    op.alter_column('user', 'permission',
               existing_type=mysql.VARCHAR(length=16),
               nullable=True,
               existing_server_default=sa.text("'general'"))
    op.drop_column('user', 'updated_at11')
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('user', sa.Column('updated_at11', mysql.INTEGER(), autoincrement=False, nullable=True))
    op.alter_column('user', 'permission',
               existing_type=mysql.VARCHAR(length=16),
               nullable=False,
               existing_server_default=sa.text("'general'"))
    op.alter_column('user', 'status',
               existing_type=mysql.VARCHAR(length=10),
               nullable=False,
               existing_server_default=sa.text("'1'"))
    op.alter_column('user', 'bisheng_id',
               existing_type=mysql.INTEGER(),
               nullable=False)
    op.alter_column('user', 'ragflow_id',
               existing_type=mysql.VARCHAR(length=32),
               nullable=False)
    op.create_table('user_test',
    sa.Column('id', mysql.INTEGER(), autoincrement=True, nullable=False),
    sa.Column('username', mysql.VARCHAR(length=255), nullable=True),
    sa.Column('hashed_password', mysql.VARCHAR(length=255), nullable=True),
    sa.Column('password', mysql.VARCHAR(length=255), nullable=True),
    sa.Column('compellation', mysql.VARCHAR(length=255), nullable=False),
    sa.Column('phone', mysql.VARCHAR(length=255), nullable=False),
    sa.Column('email', mysql.VARCHAR(length=255), nullable=False),
    sa.Column('description', mysql.VARCHAR(length=255), nullable=False),
    sa.Column('ragflow_id', mysql.VARCHAR(length=32), nullable=True),
    sa.Column('bisheng_id', mysql.INTEGER(), autoincrement=False, nullable=True),
    sa.PrimaryKeyConstraint('id'),
    mysql_collate='utf8mb4_0900_ai_ci',
    mysql_default_charset='utf8mb4',
    mysql_engine='InnoDB'
    )
    op.create_index('ix_user_test_username', 'user_test', ['username'], unique=True)
    op.create_index('ix_user_test_id', 'user_test', ['id'], unique=False)
    # ### end Alembic commands ###