mirror of
https://github.com/zulip/zulip.git
synced 2025-11-16 11:52:01 +00:00
365fe0b3d5
Fixes #2665. Regenerated by tabbott with `lint --fix` after a rebase and change in parameters. Note from tabbott: In a few cases, this converts technical debt in the form of unsorted imports into different technical debt in the form of our largest files having very long, ugly import sequences at the start. I expect this change will increase pressure for us to split those files, which isn't a bad thing. Signed-off-by: Anders Kaseorg <anders@zulip.com>
28 lines
737 B
Python
28 lines
737 B
Python
from typing import Any, Dict, List
|
|
|
|
|
|
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]] = dict()
|
|
|
|
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
|