Coverage for /usr/local/lib/python3.12/site-packages/prefect/server/database/_migrations/versions/postgresql/2022_10_19_093902_6d548701edef_add_created_by.py: 80%

1"""add_created_by 

2 

3Revision ID: 6d548701edef 

4Revises: ad4b1b4d1e9d 

5Create Date: 2022-10-19 09:39:02.371032 

6 

7""" 

8 

9import sqlalchemy as sa 

10from alembic import op 

11 

12import prefect 

13 

14# revision identifiers, used by Alembic. 

15revision = "6d548701edef" 

16down_revision = "ad4b1b4d1e9d" 

17branch_labels = None 

18depends_on = None 

19 

20 

21def upgrade(): 

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

23 op.add_column( 

24 "deployment", 

25 sa.Column( 

26 "created_by", 

27 prefect.server.utilities.database.Pydantic( 

28 prefect.server.schemas.core.CreatedBy 

29 ), 

30 nullable=True, 

31 ), 

32 ) 

33 op.add_column( 

34 "deployment", 

35 sa.Column( 

36 "updated_by", 

37 prefect.server.utilities.database.Pydantic( 

38 prefect.server.schemas.core.UpdatedBy 

39 ), 

40 nullable=True, 

41 ), 

42 ) 

43 op.add_column( 

44 "flow_run", 

45 sa.Column( 

46 "created_by", 

47 prefect.server.utilities.database.Pydantic( 

48 prefect.server.schemas.core.CreatedBy 

49 ), 

50 nullable=True, 

51 ), 

52 ) 

53 # ### end Alembic commands ### 

54 

55 

56def downgrade(): 

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

58 op.drop_column("flow_run", "created_by") 

59 op.drop_column("deployment", "updated_by") 

60 op.drop_column("deployment", "created_by") 

61 # ### end Alembic commands ###