zulip/zerver/lib/user_status.py

from django.utils.timezone import now as timezone_now

from zerver.models import (
    UserStatus,
)

from typing import Set

def get_away_user_ids(realm_id: int) -> Set[int]:
    user_ids = UserStatus.objects.filter(
        status=UserStatus.AWAY,
        user_profile__realm_id=realm_id,
        user_profile__is_active=True,
    ).values_list('user_profile_id', flat=True)

    return set(user_ids)

def update_user_status(user_profile_id: int,
                       status: int,
                       client_id: int) -> None:

    timestamp = timezone_now()

    defaults = dict(
        client_id=client_id,
        timestamp=timestamp,
    )

    defaults['status'] = status

    UserStatus.objects.update_or_create(
        user_profile_id=user_profile_id,
        defaults=defaults,
    )