Coverage for mindsdb / migrations / versions / 2023-05-24_6d748f2c7b0b_remove_streams.py: 82%

1"""remove streams 

2 

3Revision ID: 6d748f2c7b0b 

4Revises: b8be148dbc85 

5Create Date: 2023-05-24 15:22:11.969791 

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 = '6d748f2c7b0b' 

15down_revision = 'b8be148dbc85' 

16branch_labels = None 

17depends_on = None 

18 

19 

20def upgrade(): 

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

22 op.drop_table('stream') 

23 # ### end Alembic commands ### 

24 

25 

26def downgrade(): 

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

28 op.create_table( 

29 'stream', 

30 sa.Column('id', sa.INTEGER(), nullable=False), 

31 sa.Column('name', sa.VARCHAR(), nullable=False), 

32 sa.Column('stream_in', sa.VARCHAR(), nullable=False), 

33 sa.Column('stream_out', sa.VARCHAR(), nullable=False), 

34 sa.Column('anomaly_stream', sa.VARCHAR(), nullable=True), 

35 sa.Column('integration', sa.VARCHAR(), nullable=True), 

36 sa.Column('predictor', sa.VARCHAR(), nullable=False), 

37 sa.Column('company_id', sa.INTEGER(), nullable=True), 

38 sa.Column('updated_at', sa.DATETIME(), nullable=True), 

39 sa.Column('created_at', sa.DATETIME(), nullable=True), 

40 sa.Column('type', sa.VARCHAR(), nullable=True), 

41 sa.Column('connection_info', sa.VARCHAR(), nullable=True), 

42 sa.Column('learning_params', sa.VARCHAR(), nullable=True), 

43 sa.Column('learning_threshold', sa.INTEGER(), nullable=True), 

44 sa.PrimaryKeyConstraint('id') 

45 ) 

46 # ### end Alembic commands ###