mirror of
https://github.com/zulip/zulip.git
synced 2025-11-02 21:13:36 +00:00
60762a750b
This commit refactors 'set_user_topic_visibility_policy_in_database' to perform bulk database operations and the related changes. There is an increase in database query count because requests to delete user_topic rows now take two queries instead of one. This is required for logging the info for a request to delete a non-existent user_topic row while performing bulk operations at the same time. The overall query count will be lower while performing bulk operations (multiple user_profiles instead of one).
83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
import datetime
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from django.utils.timezone import now as timezone_now
|
|
|
|
from zerver.lib.timestamp import datetime_to_timestamp
|
|
from zerver.lib.user_topics import (
|
|
bulk_set_user_topic_visibility_policy_in_database,
|
|
get_topic_mutes,
|
|
)
|
|
from zerver.models import Stream, UserProfile
|
|
from zerver.tornado.django_api import send_event
|
|
|
|
|
|
def bulk_do_set_user_topic_visibility_policy(
|
|
user_profiles: List[UserProfile],
|
|
stream: Stream,
|
|
topic: str,
|
|
*,
|
|
visibility_policy: int,
|
|
last_updated: Optional[datetime.datetime] = None,
|
|
skip_muted_topics_event: bool = False,
|
|
) -> None:
|
|
if last_updated is None:
|
|
last_updated = timezone_now()
|
|
|
|
user_profiles_with_changed_user_topic_rows = bulk_set_user_topic_visibility_policy_in_database(
|
|
user_profiles,
|
|
stream.id,
|
|
topic,
|
|
visibility_policy=visibility_policy,
|
|
recipient_id=stream.recipient_id,
|
|
last_updated=last_updated,
|
|
)
|
|
|
|
# Users with requests to set the visibility_policy to its current value
|
|
# or to delete a UserTopic row that doesn't exist shouldn't
|
|
# send an unnecessary event.
|
|
if len(user_profiles_with_changed_user_topic_rows) == 0:
|
|
return
|
|
|
|
for user_profile in user_profiles_with_changed_user_topic_rows:
|
|
# This first muted_topics event is deprecated and will be removed
|
|
# once clients are migrated to handle the user_topic event type
|
|
# instead.
|
|
if not skip_muted_topics_event:
|
|
muted_topics_event = dict(
|
|
type="muted_topics", muted_topics=get_topic_mutes(user_profile)
|
|
)
|
|
send_event(user_profile.realm, muted_topics_event, [user_profile.id])
|
|
|
|
user_topic_event: Dict[str, Any] = {
|
|
"type": "user_topic",
|
|
"stream_id": stream.id,
|
|
"topic_name": topic,
|
|
"last_updated": datetime_to_timestamp(last_updated),
|
|
"visibility_policy": visibility_policy,
|
|
}
|
|
|
|
send_event(user_profile.realm, user_topic_event, [user_profile.id])
|
|
|
|
|
|
def do_set_user_topic_visibility_policy(
|
|
user_profile: UserProfile,
|
|
stream: Stream,
|
|
topic: str,
|
|
*,
|
|
visibility_policy: int,
|
|
last_updated: Optional[datetime.datetime] = None,
|
|
skip_muted_topics_event: bool = False,
|
|
) -> None:
|
|
# For conciseness, this function should be used when a single
|
|
# user_profile is involved. In the case of multiple user profiles,
|
|
# call 'bulk_do_set_user_topic_visibility_policy' directly.
|
|
bulk_do_set_user_topic_visibility_policy(
|
|
[user_profile],
|
|
stream,
|
|
topic,
|
|
visibility_policy=visibility_policy,
|
|
last_updated=last_updated,
|
|
skip_muted_topics_event=skip_muted_topics_event,
|
|
)
|