mirror of https://github.com/zulip/zulip.git
63 lines
1.4 KiB
Python
63 lines
1.4 KiB
Python
from typing import Any, Dict, Optional
|
|
|
|
from django.db.models import Q
|
|
from django.utils.timezone import now as timezone_now
|
|
|
|
from zerver.models import UserStatus
|
|
|
|
|
|
def get_user_info_dict(realm_id: int) -> Dict[str, Dict[str, Any]]:
|
|
rows = (
|
|
UserStatus.objects.filter(
|
|
user_profile__realm_id=realm_id,
|
|
user_profile__is_active=True,
|
|
)
|
|
.exclude(
|
|
Q(status=UserStatus.NORMAL) & Q(status_text=""),
|
|
)
|
|
.values(
|
|
"user_profile_id",
|
|
"status",
|
|
"status_text",
|
|
)
|
|
)
|
|
|
|
user_dict: Dict[str, Dict[str, Any]] = {}
|
|
for row in rows:
|
|
away = row["status"] == UserStatus.AWAY
|
|
status_text = row["status_text"]
|
|
user_id = row["user_profile_id"]
|
|
|
|
dct = {}
|
|
if away:
|
|
dct["away"] = away
|
|
if status_text:
|
|
dct["status_text"] = status_text
|
|
|
|
user_dict[str(user_id)] = dct
|
|
|
|
return user_dict
|
|
|
|
|
|
def update_user_status(
|
|
user_profile_id: int, status: Optional[int], status_text: Optional[str], client_id: int
|
|
) -> None:
|
|
|
|
timestamp = timezone_now()
|
|
|
|
defaults = dict(
|
|
client_id=client_id,
|
|
timestamp=timestamp,
|
|
)
|
|
|
|
if status is not None:
|
|
defaults["status"] = status
|
|
|
|
if status_text is not None:
|
|
defaults["status_text"] = status_text
|
|
|
|
UserStatus.objects.update_or_create(
|
|
user_profile_id=user_profile_id,
|
|
defaults=defaults,
|
|
)
|