Coverage for mindsdb / migrations / versions / 2022-08-19_976f15a37e6a_predictors_versioning.py: 50%

1"""predictors-versioning 

2 

3Revision ID: 976f15a37e6a 

4Revises: 6e834843e7e9 

5Create Date: 2022-08-19 11:22:52.085339 

6 

7""" 

8from alembic import op 

9import sqlalchemy as sa 

10from sqlalchemy.sql import text 

11 

12import mindsdb.interfaces.storage.db # noqa 

13 

14 

15# revision identifiers, used by Alembic. 

16revision = '976f15a37e6a' 

17down_revision = '6e834843e7e9' 

18branch_labels = None 

19depends_on = None 

20 

21 

22def upgrade(): 

23 conn = op.get_bind() 

24 session = sa.orm.Session(bind=conn) 

25 

26 with op.batch_alter_table('predictor', schema=None) as batch_op: 

27 batch_op.add_column(sa.Column('deleted_at', sa.DateTime(), nullable=True)) 

28 batch_op.add_column(sa.Column('active', sa.Boolean(), nullable=True)) 

29 

30 conn.execute(text(''' 

31 update predictor set active = :val; 

32 '''), { 

33 'val': True 

34 }) 

35 

36 session.commit() 

37 

38 

39def downgrade(): 

40 with op.batch_alter_table('predictor', schema=None) as batch_op: 

41 batch_op.drop_column('active') 

42 batch_op.drop_column('deleted_at')