Add migration to add user.pubid column

hypothesis · Jan 14, 2025 · 39a83dc · 39a83dc
1 parent dac0c3f
commit 39a83dc
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/h/migrations/versions/3bae642f2b36_add_pubid_to_user.py b/h/migrations/versions/3bae642f2b36_add_pubid_to_user.py
@@ -0,0 +1,17 @@
+"""Add pubid to user."""
+
+import sqlalchemy as sa
+from alembic import op
+
+revision = "3bae642f2b36"
+down_revision = "9c0efdf762a6"
+
+
+def upgrade():
+    op.add_column("user", sa.Column("pubid", sa.Text(), nullable=True))
+    op.create_unique_constraint(op.f("uq__user__pubid"), "user", ["pubid"])
+
+
+def downgrade():
+    op.drop_constraint(op.f("uq__user__pubid"), "user", type_="unique")
+    op.drop_column("user", "pubid")