mirror of
https://github.com/zulip/zulip.git
synced 2025-10-24 16:43:57 +00:00
f0a88d51cc
This commit adds the ability for users to include notes with their message reminders. Fixes #35070. Co-Authored-By: Aman Agrawal <amanagr@zulip.com>
61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
import datetime
|
|
|
|
from django.db import transaction
|
|
|
|
from zerver.actions.message_send import check_message
|
|
from zerver.actions.scheduled_messages import do_schedule_messages
|
|
from zerver.lib.addressee import Addressee
|
|
from zerver.lib.message import access_message
|
|
from zerver.lib.reminders import get_reminder_formatted_content
|
|
from zerver.models import Client, ScheduledMessage, UserProfile
|
|
from zerver.tornado.django_api import send_event_on_commit
|
|
|
|
|
|
def schedule_reminder_for_message(
|
|
current_user: UserProfile,
|
|
client: Client,
|
|
message_id: int,
|
|
deliver_at: datetime.datetime,
|
|
note: str,
|
|
) -> int:
|
|
message = access_message(current_user, message_id, is_modifying_message=False)
|
|
# Even though reminder will be sent from NOTIFICATION_BOT, we still
|
|
# set current_user as the sender here to help us make the permission checks easier.
|
|
addressee = Addressee.for_user_profile(current_user)
|
|
# This can raise an exception in the unlikely event that the current user cannot DM themself.
|
|
send_request = check_message(
|
|
current_user,
|
|
client,
|
|
addressee,
|
|
get_reminder_formatted_content(message, current_user, note),
|
|
current_user.realm,
|
|
forwarder_user_profile=current_user,
|
|
)
|
|
send_request.deliver_at = deliver_at
|
|
send_request.reminder_target_message_id = message_id
|
|
send_request.reminder_note = note
|
|
|
|
return do_schedule_messages(
|
|
[send_request],
|
|
current_user,
|
|
read_by_sender=False,
|
|
delivery_type=ScheduledMessage.REMIND,
|
|
)[0]
|
|
|
|
|
|
def notify_remove_reminder(user_profile: UserProfile, reminder_id: int) -> None:
|
|
event = {
|
|
"type": "reminders",
|
|
"op": "remove",
|
|
"reminder_id": reminder_id,
|
|
}
|
|
send_event_on_commit(user_profile.realm, event, [user_profile.id])
|
|
|
|
|
|
@transaction.atomic(durable=True)
|
|
def do_delete_reminder(user_profile: UserProfile, reminder: ScheduledMessage) -> None:
|
|
assert reminder.delivery_type == ScheduledMessage.REMIND
|
|
reminder_id = reminder.id
|
|
reminder.delete()
|
|
notify_remove_reminder(user_profile, reminder_id)
|