@@ -42,15 +42,17 @@ def upgrade() -> None:
4242 op .add_column (
4343 "item" ,
4444 sa .Column (
45- "new_owner_id" , postgresql .UUID (as_uuid = True ), nullable = True ,
45+ "new_owner_id" ,
46+ postgresql .UUID (as_uuid = True ),
47+ nullable = True ,
4648 ),
4749 )
4850
4951 # Populate the new columns with UUIDs
5052 op .execute ('UPDATE "user" SET new_id = uuid_generate_v4()' )
5153 op .execute ("UPDATE item SET new_id = uuid_generate_v4()" )
5254 op .execute (
53- ' UPDATE item SET new_owner_id = '
55+ " UPDATE item SET new_owner_id = "
5456 '(SELECT new_id FROM "user" WHERE "user".id = item.owner_id)' ,
5557 )
5658
@@ -75,7 +77,11 @@ def upgrade() -> None:
7577
7678 # Recreate foreign key constraint
7779 op .create_foreign_key (
78- "item_owner_id_fkey" , "item" , "user" , ["owner_id" ], ["id" ],
80+ "item_owner_id_fkey" ,
81+ "item" ,
82+ "user" ,
83+ ["owner_id" ],
84+ ["id" ],
7985 )
8086
8187
@@ -89,12 +95,10 @@ def downgrade() -> None:
8995 # Populate the old columns with default values
9096 # Generate sequences for the integer IDs if not exist
9197 op .execute (
92- 'CREATE SEQUENCE IF NOT EXISTS user_id_seq AS INTEGER '
93- 'OWNED BY "user".old_id' ,
98+ 'CREATE SEQUENCE IF NOT EXISTS user_id_seq AS INTEGER OWNED BY "user".old_id' ,
9499 )
95100 op .execute (
96- "CREATE SEQUENCE IF NOT EXISTS item_id_seq AS INTEGER "
97- "OWNED BY item.old_id" ,
101+ "CREATE SEQUENCE IF NOT EXISTS item_id_seq AS INTEGER OWNED BY item.old_id" ,
98102 )
99103
100104 op .execute (
@@ -108,7 +112,7 @@ def downgrade() -> None:
108112
109113 op .execute ("UPDATE \" user\" SET old_id = nextval('user_id_seq')" )
110114 op .execute (
111- ' UPDATE item SET old_id = nextval(\ ' item_id_seq\ ' ), '
115+ " UPDATE item SET old_id = nextval('item_id_seq'), "
112116 'old_owner_id = (SELECT old_id FROM "user" '
113117 'WHERE "user".id = item.owner_id)' ,
114118 )
@@ -130,5 +134,9 @@ def downgrade() -> None:
130134
131135 # Recreate foreign key constraint
132136 op .create_foreign_key (
133- "item_owner_id_fkey" , "item" , "user" , ["owner_id" ], ["id" ],
137+ "item_owner_id_fkey" ,
138+ "item" ,
139+ "user" ,
140+ ["owner_id" ],
141+ ["id" ],
134142 )
0 commit comments