mirror of https://github.com/zulip/zulip.git
38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
from typing import List, Union
|
|
|
|
from django.utils.translation import gettext as _
|
|
|
|
from zerver.lib.exceptions import ResourceNotFoundError
|
|
from zerver.models import (
|
|
APIScheduledDirectMessageDict,
|
|
APIScheduledStreamMessageDict,
|
|
ScheduledMessage,
|
|
UserProfile,
|
|
)
|
|
|
|
|
|
def access_scheduled_message(
|
|
user_profile: UserProfile, scheduled_message_id: int
|
|
) -> ScheduledMessage:
|
|
try:
|
|
return ScheduledMessage.objects.get(id=scheduled_message_id, sender=user_profile)
|
|
except ScheduledMessage.DoesNotExist:
|
|
raise ResourceNotFoundError(_("Scheduled message does not exist"))
|
|
|
|
|
|
def get_undelivered_scheduled_messages(
|
|
user_profile: UserProfile,
|
|
) -> List[Union[APIScheduledDirectMessageDict, APIScheduledStreamMessageDict]]:
|
|
scheduled_messages = ScheduledMessage.objects.filter(
|
|
realm_id=user_profile.realm_id,
|
|
sender=user_profile,
|
|
# Notably, we don't require failed=False, since we will want
|
|
# to display those to users.
|
|
delivered=False,
|
|
delivery_type=ScheduledMessage.SEND_LATER,
|
|
).order_by("scheduled_timestamp")
|
|
scheduled_message_dicts: List[
|
|
Union[APIScheduledDirectMessageDict, APIScheduledStreamMessageDict]
|
|
] = [scheduled_message.to_dict() for scheduled_message in scheduled_messages]
|
|
return scheduled_message_dicts
|