@@ -42,15 +42,17 @@ def upgrade() -> None:
42
42
op .add_column (
43
43
"item" ,
44
44
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 ,
46
48
),
47
49
)
48
50
49
51
# Populate the new columns with UUIDs
50
52
op .execute ('UPDATE "user" SET new_id = uuid_generate_v4()' )
51
53
op .execute ("UPDATE item SET new_id = uuid_generate_v4()" )
52
54
op .execute (
53
- ' UPDATE item SET new_owner_id = '
55
+ " UPDATE item SET new_owner_id = "
54
56
'(SELECT new_id FROM "user" WHERE "user".id = item.owner_id)' ,
55
57
)
56
58
@@ -75,7 +77,11 @@ def upgrade() -> None:
75
77
76
78
# Recreate foreign key constraint
77
79
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" ],
79
85
)
80
86
81
87
@@ -89,12 +95,10 @@ def downgrade() -> None:
89
95
# Populate the old columns with default values
90
96
# Generate sequences for the integer IDs if not exist
91
97
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' ,
94
99
)
95
100
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" ,
98
102
)
99
103
100
104
op .execute (
@@ -108,7 +112,7 @@ def downgrade() -> None:
108
112
109
113
op .execute ("UPDATE \" user\" SET old_id = nextval('user_id_seq')" )
110
114
op .execute (
111
- ' UPDATE item SET old_id = nextval(\ ' item_id_seq\ ' ), '
115
+ " UPDATE item SET old_id = nextval('item_id_seq'), "
112
116
'old_owner_id = (SELECT old_id FROM "user" '
113
117
'WHERE "user".id = item.owner_id)' ,
114
118
)
@@ -130,5 +134,9 @@ def downgrade() -> None:
130
134
131
135
# Recreate foreign key constraint
132
136
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" ],
134
142
)
0 commit comments