Add pubid to User model

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")

h/models/user.py

@@ -1,5 +1,6 @@
 import datetime
 import re
+from functools import partial
 from typing import TYPE_CHECKING
 
 import sqlalchemy as sa
@@ -9,6 +10,7 @@
 from h.db import Base
 from h.exceptions import InvalidUserId
 from h.models import helpers
+from h.pubid import generate
 from h.util.user import format_userid, split_user
 
 if TYPE_CHECKING:
@@ -20,6 +22,7 @@
 USERNAME_PATTERN = "(?i)^[A-Z0-9._]+$"
 EMAIL_MAX_LENGTH = 100
 DISPLAY_NAME_MAX_LENGTH = 30
+USER_PUBID_LENGTH = 12
 
 
 def _normalise_username(username):
@@ -158,6 +161,14 @@ def __table_args__(cls):  # pylint:disable=no-self-argument
 
     id = sa.Column(sa.Integer, autoincrement=True, primary_key=True)
 
+    #: A publicly visible unique identifier for the user
+    pubid = sa.Column(
+        sa.Text(),
+        default=partial(generate, USER_PUBID_LENGTH),
+        unique=True,
+        nullable=True,
+    )
+
     #: Username as chosen by the user on registration
     _username = sa.Column("username", sa.UnicodeText(), nullable=False)