CodeCow/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2026-07-12 21:04:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

message: Use RawUnreadMessageResult typing more.

Browse Source 
This is a cleaner implementation given we have the TypedDict defined.
This commit is contained in:
Tim Abbott 2019-08-02 17:33:42 -07:00
parent 8f7a86f5ec
commit ba67a9b76a
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
zerver/lib/message.py
View File

@ -895,7 +895,7 @@ def aggregate_unread_data(raw_data: RawUnreadMessagesResult) -> UnreadMessagesRe
return result
def apply_unread_message_event(user_profile: UserProfile,
state: Dict[str, Any],
state: RawUnreadMessagesResult,
message: Dict[str, Any],
flags: List[str]) -> None:
message_id = message['id']
@ -950,13 +950,13 @@ def apply_unread_message_event(user_profile: UserProfile,
if 'mentioned' in flags:
state['mentions'].add(message_id)
def remove_message_id_from_unread_mgs(state: Dict[str, Any],
def remove_message_id_from_unread_mgs(state: RawUnreadMessagesResult,
message_id: int) -> None:
# The opposite of apply_unread_message_event; removes a read or
# deleted message from a raw_unread_msgs data structure.
for key in ['pm_dict', 'stream_dict', 'huddle_dict']:
state[key].pop(message_id, None)
state['pm_dict'].pop(message_id, None)
state['stream_dict'].pop(message_id, None)
state['huddle_dict'].pop(message_id, None)
state['unmuted_stream_msgs'].discard(message_id)
state['mentions'].discard(message_id)