Add university data fields to inferred_user table and enumeration

Author: lpi-tn

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "welearn-database"
-version = "1.4.4"
+version = "1.4.5"
 description = "All stuff related to relationnal database from the WeLearn project"
 authors = [
     {name = "Théo",email = "theo.nardin@cri-paris.org"}

welearn_database/alembic/versions/8b780aea403a_add_university_data.py

@@ -0,0 +1,48 @@
+"""add_university_data
+
+Revision ID: 8b780aea403a
+Revises: f1ce0ad2845b
+Create Date: 2026-06-03 11:34:02.441435
+
+"""
+
+from typing import Sequence, Union
+
+import sqlalchemy as sa
+from alembic import op
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision: str = "8b780aea403a"
+down_revision: Union[str, None] = "f1ce0ad2845b"
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    op.add_column(
+        "inferred_user",
+        sa.Column("university_title", sa.String(), nullable=True),
+        schema="user_related",
+    )
+    op.add_column(
+        "inferred_user",
+        sa.Column(
+            "role",
+            postgresql.ENUM(
+                "student",
+                "teacher",
+                "staff",
+                name="university_role",
+                schema="user_related",
+            ),
+            nullable=True,
+        ),
+        schema="user_related",
+    )
+
+
+def downgrade() -> None:
+    op.drop_column("inferred_user", "university_title", schema="user_related")
+    op.drop_column("inferred_user", "role", schema="user_related")
+    op.execute("DROP TYPE user_related.university_role")

welearn_database/data/enumeration.py

@@ -43,3 +43,9 @@ class ExternalIdType(StrEnum):
 class FilterType(StrEnum):
     SDG = auto()
     SOURCE = auto()
+
+
+class UniversityRole(StrEnum):
+    STUDENT = auto()
+    TEACHER = auto()
+    STAFF = auto()

welearn_database/data/models/user_related.py

@@ -5,7 +5,7 @@
 from sqlalchemy.dialects.postgresql import ENUM, TIMESTAMP
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 
-from welearn_database.data.enumeration import DbSchemaEnum, FilterType
+from welearn_database.data.enumeration import DbSchemaEnum, FilterType, UniversityRole
 from welearn_database.data.models.document_related import WeLearnDocument
 
 from . import Base
@@ -216,6 +216,15 @@ class InferredUser(Base):
         types.Uuid, primary_key=True, nullable=False, server_default="gen_random_uuid()"
     )
     origin_referrer: Mapped[str | None]
+    university_title: Mapped[str | None]
+    role: Mapped[str] = mapped_column(
+        ENUM(
+            *(r.value.lower() for r in UniversityRole),
+            name="university_role",
+            schema="user_related",
+        ),
+        nullable=False,
+    )
     created_at: Mapped[datetime] = mapped_column(
         TIMESTAMP(timezone=False),
         nullable=False,