1 """add_created_by
2
3 Revision ID: fa319f214160
4 Revises: ad4b1b4d1e9d
5 Create Date: 2022-10-19 09:35:42.371899
6
7 """
8
9 import sqlalchemy as sa 1 ctx 1a
10 from alembic import op 1 ctx 1a
11
12 import prefect 1 ctx 1a
13
14 # revision identifiers, used by Alembic.
15 revision = "fa319f214160" 1 ctx 1a
16 down_revision = "ad4b1b4d1e9d" 1 ctx 1a
17 branch_labels = None 1 ctx 1a
18 depends_on = None 1 ctx 1a
19
20
21 def upgrade ( ) : 1 ctx 1a
22 # ### commands auto generated by Alembic - please adjust! ###
23 with op . batch_alter_table ( "deployment" , schema = None ) as batch_op : 1 ctx 1a
24 batch_op . add_column ( 1 ctx 1a
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 batch_op . add_column ( 1 ctx 1a
34 sa . Column (
35 "updated_by" ,
36 prefect . server . utilities . database . Pydantic (
37 prefect . server . schemas . core . UpdatedBy
38 ) ,
39 nullable = True ,
40 )
41 )
42
43 with op . batch_alter_table ( "flow_run" , schema = None ) as batch_op : 1 ctx 1a
44 batch_op . add_column ( 1 ctx 1a
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
54 # ### end Alembic commands ###
55
56
57 def downgrade ( ) : 1 ctx 1a
58 # ### commands auto generated by Alembic - please adjust! ###
59 with op . batch_alter_table ( "flow_run" , schema = None ) as batch_op :
60 batch_op . drop_column ( "created_by" )
61
62 with op . batch_alter_table ( "deployment" , schema = None ) as batch_op :
63 batch_op . drop_column ( "updated_by" )
64 batch_op . drop_column ( "created_by" )
65
66 # ### end Alembic commands ###