snapshot of https://github.com/zulip/zulip.git @ 48c69911c7dd4e03b20552eb3bd77ad8151c1357 (2026-09-15)

zerver/data_import/user_handler.py

@@ -0,0 +1,33 @@
+from typing import Any
+
+from zerver.data_import.import_util import validate_user_emails_for_import
+
+
+class UserHandler:
+    """
+    Our UserHandler class is a glorified wrapper
+    around the data that eventually goes into
+    zerver_userprofile.
+
+    The class helps us do things like map ids
+    to names for mentions.
+    """
+
+    def __init__(self) -> None:
+        self.id_to_user_map: dict[int, dict[str, Any]] = {}
+
+    def add_user(self, user: dict[str, Any]) -> None:
+        user_id = user["id"]
+        self.id_to_user_map[user_id] = user
+
+    def get_user(self, user_id: int) -> dict[str, Any]:
+        user = self.id_to_user_map[user_id]
+        return user
+
+    def get_all_users(self) -> list[dict[str, Any]]:
+        users = list(self.id_to_user_map.values())
+        return users
+
+    def validate_user_emails(self) -> None:
+        all_users = self.get_all_users()
+        validate_user_emails_for_import([user["delivery_email"] for user in all_users])