mirror of https://github.com/zulip/zulip.git
44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
|
import datetime
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
from django.utils.timezone import now as timezone_now
|
||
|
|
from django.utils.translation import gettext as _
|
||
|
|
|
||
|
|
from zerver.lib.exceptions import JsonableError
|
||
|
|
from zerver.lib.user_topics import add_topic_mute, get_topic_mutes, remove_topic_mute
|
||
|
|
from zerver.models import Stream, UserProfile, UserTopic
|
||
|
|
from zerver.tornado.django_api import send_event
|
||
|
|
|
||
|
|
|
||
|
|
def do_mute_topic(
|
||
|
|
user_profile: UserProfile,
|
||
|
|
stream: Stream,
|
||
|
|
topic: str,
|
||
|
|
date_muted: Optional[datetime.datetime] = None,
|
||
|
|
ignore_duplicate: bool = False,
|
||
|
|
) -> None:
|
||
|
|
if date_muted is None:
|
||
|
|
date_muted = timezone_now()
|
||
|
|
add_topic_mute(
|
||
|
|
user_profile,
|
||
|
|
stream.id,
|
||
|
|
stream.recipient_id,
|
||
|
|
topic,
|
||
|
|
date_muted,
|
||
|
|
ignore_duplicate=ignore_duplicate,
|
||
|
|
)
|
||
|
|
event = dict(type="muted_topics", muted_topics=get_topic_mutes(user_profile))
|
||
|
|
send_event(user_profile.realm, event, [user_profile.id])
|
||
|
|
|
||
|
|
|
||
|
|
def do_unmute_topic(user_profile: UserProfile, stream: Stream, topic: str) -> None:
|
||
|
|
# Note: If you add any new code to this function, the
|
||
|
|
# remove_topic_mute call in do_update_message will need to be
|
||
|
|
# updated for correctness.
|
||
|
|
try:
|
||
|
|
remove_topic_mute(user_profile, stream.id, topic)
|
||
|
|
except UserTopic.DoesNotExist:
|
||
|
|
raise JsonableError(_("Topic is not muted"))
|
||
|
|
event = dict(type="muted_topics", muted_topics=get_topic_mutes(user_profile))
|
||
|
|
send_event(user_profile.realm, event, [user_profile.id])
|