message send: Introduce dataclass to wrap user-notifications variables.

We will in later commits, extend this class to contain methods
to determine if a message is notifiable or not, but for now
we only turn it into a dict and pass it on.

zerver/lib/actions.py

@@ -110,6 +110,7 @@ from zerver.lib.message import (
     update_first_visible_message_id,
     wildcard_mention_allowed,
 )
+from zerver.lib.notification_data import UserMessageNotificationsData
 from zerver.lib.pysa import mark_sanitized
 from zerver.lib.queue import queue_json_publish
 from zerver.lib.realm_icon import realm_icon_url
@@ -1979,15 +1980,17 @@ def do_send_messages(
                 user_id in send_request.wildcard_mention_user_ids and "wildcard_mentioned" in flags
             )
             users.append(
-                dict(
+                asdict(
-                    id=user_id,
+                    UserMessageNotificationsData(
-                    flags=flags,
+                        id=user_id,
-                    mentioned=("mentioned" in flags),
+                        flags=flags,
-                    online_push_enabled=(user_id in send_request.online_push_user_ids),
+                        mentioned=("mentioned" in flags),
-                    stream_push_notify=(user_id in send_request.stream_push_user_ids),
+                        online_push_enabled=(user_id in send_request.online_push_user_ids),
-                    stream_email_notify=(user_id in send_request.stream_email_user_ids),
+                        stream_push_notify=(user_id in send_request.stream_push_user_ids),
-                    wildcard_mention_notify=wildcard_mention_notify,
+                        stream_email_notify=(user_id in send_request.stream_email_user_ids),
-                    sender_is_muted=(user_id in send_request.muted_sender_user_ids),
+                        wildcard_mention_notify=wildcard_mention_notify,
+                        sender_is_muted=(user_id in send_request.muted_sender_user_ids),
+                    )
                 )
             )
 

zerver/lib/notification_data.py

@@ -0,0 +1,20 @@
+from dataclasses import dataclass
+from typing import List
+
+
+@dataclass
+class UserMessageNotificationsData:
+    id: int
+    flags: List[str]
+    mentioned: bool
+    online_push_enabled: bool
+    stream_push_notify: bool
+    stream_email_notify: bool
+    wildcard_mention_notify: bool
+    sender_is_muted: bool
+
+    def __post_init__(self) -> None:
+        if self.mentioned:
+            assert "mentioned" in self.flags
+        if self.wildcard_mention_notify:
+            assert "wildcard_mentioned" in self.flags