Coverage for mindsdb / migrations / versions / 2024-08-12_8e17ff6b75e9_agents_deleted_at.py: 43%

1"""agents-deleted-at 

2 

3Revision ID: 8e17ff6b75e9 

4Revises: 45eb2eb61f70 

5Create Date: 2024-08-12 19:13:44.327111 

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 = '8e17ff6b75e9' 

15down_revision = '45eb2eb61f70' 

16branch_labels = None 

17depends_on = None 

18 

19 

20def upgrade(): 

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

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

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

24 

25 with op.batch_alter_table('skills', schema=None) as batch_op: 

26 batch_op.add_column(sa.Column('created_at', sa.DateTime(), nullable=True)) 

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

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

29 

30 # ### end Alembic commands ### 

31 

32 

33def downgrade(): 

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

35 with op.batch_alter_table('skills', schema=None) as batch_op: 

36 batch_op.drop_column('deleted_at') 

37 batch_op.drop_column('updated_at') 

38 batch_op.drop_column('created_at') 

39 

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

41 batch_op.drop_column('deleted_at') 

42 # ### end Alembic commands ###