Coverage for mindsdb / migrations / versions / 2025-03-21_fda503400e43_queries.py: 60%

1"""queries 

2 

3Revision ID: fda503400e43 

4Revises: 11347c213b36 

5Create Date: 2025-03-21 18:50:20.795930 

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 = 'fda503400e43' 

15down_revision = '11347c213b36' 

16branch_labels = None 

17depends_on = None 

18 

19 

20def upgrade(): 

21 op.create_table( 

22 'queries', 

23 sa.Column('id', sa.Integer(), nullable=False), 

24 sa.Column('company_id', sa.Integer(), nullable=True), 

25 sa.Column('sql', sa.String(), nullable=False), 

26 sa.Column('started_at', sa.DateTime(), nullable=True), 

27 sa.Column('finished_at', sa.DateTime(), nullable=True), 

28 sa.Column('parameters', sa.JSON(), nullable=True), 

29 sa.Column('context', sa.JSON(), nullable=True), 

30 sa.Column('processed_rows', sa.Integer(), nullable=True), 

31 sa.Column('error', sa.String(), nullable=True), 

32 sa.Column('updated_at', sa.DateTime(), nullable=True), 

33 sa.Column('created_at', sa.DateTime(), nullable=True), 

34 sa.PrimaryKeyConstraint('id') 

35 ) 

36 

37 with op.batch_alter_table('knowledge_base', schema=None) as batch_op: 

38 batch_op.add_column(sa.Column('query_id', sa.INTEGER(), nullable=True)) 

39 

40 

41def downgrade(): 

42 with op.batch_alter_table('knowledge_base', schema=None) as batch_op: 

43 batch_op.drop_column('query_id') 

44 

45 op.drop_table('queries')