Try Mike's Magic Migration Method (pypi#16722)

miketheman · Sep 16, 2024 · ffebdad · ffebdad
1 parent 6eb044f
commit ffebdad
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/warehouse/migrations/versions/0b74ed7d4880_add_verified_emails_columns_to_release.py b/warehouse/migrations/versions/0b74ed7d4880_add_verified_emails_columns_to_release.py
@@ -36,18 +36,35 @@ def upgrade():
  "author_email_verified",
  sa.Boolean(),
  server_default=sa.text("false"),
- nullable=False,
+ nullable=True,
  ),
  )
+ op.execute(
+ """
+ UPDATE releases
+ SET author_email_verified = false
+ WHERE author_email_verified IS NULL
+ """
+ )
+ op.alter_column("releases", "author_email_verified", nullable=False)
+
  op.add_column(
  "releases",
  sa.Column(
  "maintainer_email_verified",
  sa.Boolean(),
  server_default=sa.text("false"),
- nullable=False,
+ nullable=True,
  ),
  )
+ op.execute(
+ """
+ UPDATE releases
+ SET maintainer_email_verified = false
+ WHERE maintainer_email_verified IS NULL
+ """
+ )
+ op.alter_column("releases", "maintainer_email_verified", nullable=False)
 
 
 def downgrade():