From cc5fe4e27d006b6d8af7cb8e82a5133883f990d0 Mon Sep 17 00:00:00 2001
From: Mathieu Agopian <mathieu@agopian.info>
Date: Wed, 7 Oct 2026 11:11:23 +0200
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8(migration)=20import=5Fmigration=20man?=
 =?UTF-8?q?agement=20command=20imports=20mentions?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Straightfoward import of mentions

Signed-off-by: Mathieu Agopian <mathieu@agopian.info>
---
 .../management/commands/import_migration.py   | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/src/backend/core/management/commands/import_migration.py b/src/backend/core/management/commands/import_migration.py
index 7a51f1aa9..8bd9589b7 100644
--- a/src/backend/core/management/commands/import_migration.py
+++ b/src/backend/core/management/commands/import_migration.py
@@ -31,6 +31,7 @@ from core.models import (
     DocumentFavorite,
     Invitation,
     LinkTrace,
+    Mention,
     Reaction,
     Thread,
     User,
@@ -133,6 +134,8 @@ class Command(BaseCommand):
                     created = self._import_reaction(pk, fields)
                 case "core.reaction_users":
                     created = self._import_reaction_users(pk, fields, user_uuid_remap)
+                case "core.mention":
+                    created = self._import_mention(pk, fields, user_uuid_remap)
                 case _:
                     if model_label not in skipped_models:
                         self.stderr.write(
@@ -505,3 +508,25 @@ class Command(BaseCommand):
             [UserReconciliationCsvImport(**fields, id=pk)]
         )
         return True
+
+    def _import_mention(self, pk, fields, user_uuid_remap):
+        """
+        Import one mention record.
+
+        Returns True if a new row was created, False if it already existed.
+
+        A Mention is created whenever a user is @-mentioned in a document body
+        or in a comment thread. It carries two user FKs: mentioned_user_id
+        (nullable — the mentioned person) and mentioned_by_user_id (the author
+        of the mention, non-nullable). Both may need remapping.
+        """
+        for field in ("mentioned_user_id", "mentioned_by_user_id"):
+            user_id = fields.get(field)
+            if user_id and user_id in user_uuid_remap:
+                fields[field] = user_uuid_remap[user_id]
+
+        if Mention.objects.filter(id=pk).exists():
+            return False
+
+        Mention.objects.bulk_create([Mention(**fields, id=pk)])
+        return True