message: Use more accurate `recipient_type` when applying unread event.

This commit is contained in:
Lauryn Menard 2023-04-18 17:43:35 +02:00 committed by Tim Abbott
parent 3ad5e7dfc0
commit c87fbacaed
1 changed files with 5 additions and 5 deletions

View File

@ -1273,17 +1273,17 @@ def apply_unread_message_event(
) -> None: ) -> None:
message_id = message["id"] message_id = message["id"]
if message["type"] == "stream": if message["type"] == "stream":
message_type = "stream" recipient_type = "stream"
elif message["type"] == "private": elif message["type"] == "private":
others = [recip for recip in message["display_recipient"] if recip["id"] != user_profile.id] others = [recip for recip in message["display_recipient"] if recip["id"] != user_profile.id]
if len(others) <= 1: if len(others) <= 1:
message_type = "private" recipient_type = "private"
else: else:
message_type = "huddle" recipient_type = "huddle"
else: else:
raise AssertionError("Invalid message type {}".format(message["type"])) raise AssertionError("Invalid message type {}".format(message["type"]))
if message_type == "stream": if recipient_type == "stream":
stream_id = message["stream_id"] stream_id = message["stream_id"]
topic = message[TOPIC_NAME] topic = message[TOPIC_NAME]
state["stream_dict"][message_id] = RawUnreadStreamDict( state["stream_dict"][message_id] = RawUnreadStreamDict(
@ -1300,7 +1300,7 @@ def apply_unread_message_event(
): ):
state["unmuted_stream_msgs"].add(message_id) state["unmuted_stream_msgs"].add(message_id)
elif message_type == "private": elif recipient_type == "private":
if len(others) == 1: if len(others) == 1:
other_user_id = others[0]["id"] other_user_id = others[0]["id"]
else: else: