Coverage for opt/mealie/lib/python3.12/site-packages/mealie/alembic/versions/2023-02-22-21.45.52_38514b39a824_add_auth_method_to_user_table.py: 65%

1"""add auth_method to user table 

2 

3Revision ID: 38514b39a824 

4Revises: b04a08da2108 

5Create Date: 2023-02-22 21:45:52.900964 

6 

7""" 

8 

9import sqlalchemy as sa 1 ctx1a

10 

11from alembic import op 1 ctx1a

12 

13# revision identifiers, used by Alembic. 

14revision = "38514b39a824" 1 ctx1a

15down_revision = "b04a08da2108" 1 ctx1a

16branch_labels: str | tuple[str, ...] | None = None 1 ctx1a

17depends_on: str | tuple[str, ...] | None = None 1 ctx1a

18 

19 

20def is_postgres(): 1 ctx1a

21 return op.get_context().dialect.name == "postgresql" 1 ctx1a

22 

23 

24authMethod = sa.Enum("MEALIE", "LDAP", name="authmethod") 1 ctx1a

25 

26 

27def upgrade(): 1 ctx1a

28 if is_postgres(): 28 ↛ 29line 28 didn't jump to line 29 because the condition on line 28 was never true1 ctx1a

29 authMethod.create(op.get_bind()) 

30 

31 op.add_column( 1 ctx1a

32 "users", 

33 sa.Column("auth_method", authMethod, nullable=False, server_default="MEALIE"), 

34 ) 

35 op.execute("UPDATE users SET auth_method = 'LDAP' WHERE password = 'LDAP'") 1 ctx1a

36 

37 

38def downgrade(): 1 ctx1a

39 with op.batch_alter_table("users", schema=None) as batch_op: 

40 batch_op.drop_column("auth_method") 

41 

42 if is_postgres(): 

43 authMethod.drop(op.get_bind())