"""add_table_form_result

Revision ID: bb12537357fe
Revises: 
Create Date: 2023-10-15 21:41:49.174995

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa
from sqlalchemy import TIMESTAMP, func, text

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


def upgrade():
    op.create_table('form',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('base_prompt', sa.String(length=1024), nullable=True),
                    sa.Column('p_choice', sa.String(length=255), nullable=True),
                    sa.Column('role', sa.String(length=255), nullable=True),
                    sa.Column('uuid', sa.String(length=255), nullable=True),
                    sa.Column('name', sa.String(length=1024), nullable=True),
                    sa.Column('desc', sa.String(length=2048), nullable=True),
                    sa.Column('price', sa.Float(), nullable=True),
                    sa.Column('favorable', sa.String(length=1024), nullable=True),
                    sa.Column('remark', sa.String(length=1024), nullable=True),
                    sa.Column('otherPrompt', sa.String(length=1024), nullable=True),
                    sa.Column('lang', sa.String(length=1024), nullable=True),
                    sa.Column('type', sa.String(length=1024), nullable=True),
                    sa.Column('created_at', TIMESTAMP, server_default=func.now()),
                    sa.Column('updated_at', TIMESTAMP, server_default=text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP')),
                    sa.PrimaryKeyConstraint('id')
                    )
    op.create_table('result',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('prompt', sa.String(length=2048), nullable=True),
                    sa.Column('uuid', sa.String(length=255), nullable=True),
                    sa.Column('name', sa.String(length=255), nullable=True),
                    sa.Column('value', sa.String(length=4096), nullable=True),
                    sa.Column('lang', sa.String(length=1024), nullable=True),
                    sa.Column('created_at', TIMESTAMP, server_default=func.now()),
                    sa.Column('updated_at', TIMESTAMP, server_default=text('CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP')),
                    sa.PrimaryKeyConstraint('id')
                    )


def downgrade():
    op.drop_table('form')
    op.drop_table('result')