Coverage for mindsdb / migrations / versions / 2023-06-27_607709e1615b_update_project_names.py: 39%

1"""update_project_names 

2 

3Revision ID: 607709e1615b 

4Revises: 4c26ad04eeaa 

5Create Date: 2023-06-27 18:33:29.436607 

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 = '607709e1615b' 

15down_revision = 'b5bf593ba659' 

16branch_labels = None 

17depends_on = None 

18 

19 

20def upgrade(): 

21 

22 def _rename(table): 

23 conn = op.get_bind() 

24 

25 data = conn.execute( 

26 table 

27 .select() 

28 .where(table.c.name.like("%.%")) 

29 ).fetchall() 

30 

31 for row in data: 

32 name = row[0] 

33 name2 = name.replace('.', '_') 

34 

35 op.execute( 

36 table 

37 .update() 

38 .where(table.c.name == name) 

39 .values({'name': name2}) 

40 ) 

41 

42 projects = sa.Table( 

43 'project', 

44 sa.MetaData(), 

45 sa.Column('name', sa.String()), 

46 ) 

47 

48 _rename(projects) 

49 

50 integrations = sa.Table( 

51 'integration', 

52 sa.MetaData(), 

53 sa.Column('name', sa.String()), 

54 ) 

55 

56 _rename(integrations) 

57 

58 

59def downgrade(): 

60 pass