Coverage for mindsdb / migrations / versions / 2023-02-27_ef04cdbe51ed_jobs_user_class.py: 69%

1"""jobs_user_class 

2 

3Revision ID: ef04cdbe51ed 

4Revises: 3154382dab17 

5Create Date: 2023-02-27 11:32:35.298375 

6 

7""" 

8from alembic import op 

9import sqlalchemy as sa 

10import mindsdb.interfaces.storage.db # noqa 

11 

12 

13# revision identifiers, used by Alembic. 

14revision = 'ef04cdbe51ed' 

15down_revision = '3154382dab17' 

16branch_labels = None 

17depends_on = None 

18 

19 

20def upgrade(): 

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

22 with op.batch_alter_table('jobs', schema=None) as batch_op: 

23 batch_op.add_column(sa.Column('user_class', sa.Integer(), nullable=True)) 

24 

25 # ### end Alembic commands ### 

26 

27 

28def downgrade(): 

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

30 with op.batch_alter_table('jobs', schema=None) as batch_op: 

31 batch_op.drop_column('user_class') 

32 

33 # ### end Alembic commands ###