Coverage for mindsdb / migrations / versions / 2025-04-22_53502b6d63bf_query_database.py: 69%

1"""query_database 

2 

3Revision ID: 53502b6d63bf 

4Revises: fda503400e43 

5Create Date: 2025-04-22 16:30:15.139978 

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 = '53502b6d63bf' 

15down_revision = 'fda503400e43' 

16branch_labels = None 

17depends_on = None 

18 

19 

20def upgrade(): 

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

22 batch_op.add_column(sa.Column('database', sa.String(), nullable=True)) 

23 

24 

25def downgrade(): 

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

27 batch_op.drop_column('database')