Coverage for mindsdb / migrations / versions / 2023-07-19_ad04ee0bd385_chatbot_to_task.py: 47%

1"""chatbot_to_task 

2 

3Revision ID: ad04ee0bd385 

4Revises: a57506731839 

5Create Date: 2023-07-19 16:18:14.791761 

6 

7""" 

8from alembic import op 

9import sqlalchemy as sa 

10import mindsdb.interfaces.storage.db # noqa 

11 

12# revision identifiers, used by Alembic. 

13revision = 'ad04ee0bd385' 

14down_revision = 'a57506731839' 

15branch_labels = None 

16depends_on = None 

17 

18 

19def upgrade(): 

20 # ### commands auto generated by Alembic - please adjust! ### 

21 with op.batch_alter_table('chat_bots', schema=None) as batch_op: 

22 batch_op.drop_column('chat_engine') 

23 batch_op.drop_column('user_class') 

24 batch_op.drop_column('company_id') 

25 batch_op.drop_column('is_running') 

26 

27 # ### end Alembic commands ### 

28 

29 

30def downgrade(): 

31 # ### commands auto generated by Alembic - please adjust! ### 

32 with op.batch_alter_table('chat_bots', schema=None) as batch_op: 

33 batch_op.add_column(sa.Column('is_running', sa.BOOLEAN(), nullable=True)) 

34 batch_op.add_column(sa.Column('company_id', sa.INTEGER(), nullable=True)) 

35 batch_op.add_column(sa.Column('user_class', sa.INTEGER(), nullable=True)) 

36 batch_op.add_column(sa.Column('chat_engine', sa.VARCHAR(), nullable=True)) 

37 

38 # ### end Alembic commands ###