mirror of https://github.com/zulip/zulip.git
82 lines
3.3 KiB
Python
82 lines
3.3 KiB
Python
import logging
|
|
import time
|
|
from datetime import timedelta
|
|
from typing import Any
|
|
|
|
from django.conf import settings
|
|
from django.core.management.base import BaseCommand
|
|
from django.utils.timezone import now as timezone_now
|
|
|
|
from zerver.lib.actions import build_message_send_dict, do_send_messages
|
|
from zerver.lib.logging_util import log_to_file
|
|
from zerver.lib.management import sleep_forever
|
|
from zerver.lib.message import SendMessageRequest
|
|
from zerver.models import Message, ScheduledMessage, get_user_by_delivery_email
|
|
|
|
## Setup ##
|
|
logger = logging.getLogger(__name__)
|
|
log_to_file(logger, settings.SCHEDULED_MESSAGE_DELIVERER_LOG_PATH)
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = """Deliver scheduled messages from the ScheduledMessage table.
|
|
Run this command under supervisor.
|
|
|
|
This management command is run via supervisor. Do not run on multiple
|
|
machines, as you may encounter multiple sends in a specific race
|
|
condition. (Alternatively, you can set `EMAIL_DELIVERER_DISABLED=True`
|
|
on all but one machine to make the command have no effect.)
|
|
|
|
Usage: ./manage.py deliver_scheduled_messages
|
|
"""
|
|
|
|
def construct_message(self, scheduled_message: ScheduledMessage) -> SendMessageRequest:
|
|
message = Message()
|
|
original_sender = scheduled_message.sender
|
|
message.content = scheduled_message.content
|
|
message.recipient = scheduled_message.recipient
|
|
message.subject = scheduled_message.subject
|
|
message.date_sent = timezone_now()
|
|
message.sending_client = scheduled_message.sending_client
|
|
|
|
delivery_type = scheduled_message.delivery_type
|
|
if delivery_type == ScheduledMessage.SEND_LATER:
|
|
message.sender = original_sender
|
|
elif delivery_type == ScheduledMessage.REMIND:
|
|
message.sender = get_user_by_delivery_email(
|
|
settings.NOTIFICATION_BOT, original_sender.realm
|
|
)
|
|
|
|
message_dict = {
|
|
"message": message,
|
|
"stream": scheduled_message.stream,
|
|
"realm": scheduled_message.realm,
|
|
}
|
|
return build_message_send_dict(message_dict)
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
try:
|
|
if settings.EMAIL_DELIVERER_DISABLED:
|
|
# Here doing a check and sleeping indefinitely on this setting might
|
|
# not sound right. Actually we do this check to avoid running this
|
|
# process on every server that might be in service to a realm. See
|
|
# the comment in zproject/default_settings.py file about renaming this
|
|
# setting.
|
|
sleep_forever()
|
|
|
|
while True:
|
|
messages_to_deliver = ScheduledMessage.objects.filter(
|
|
scheduled_timestamp__lte=timezone_now(), delivered=False
|
|
)
|
|
for message in messages_to_deliver:
|
|
do_send_messages([self.construct_message(message)])
|
|
message.delivered = True
|
|
message.save(update_fields=["delivered"])
|
|
|
|
cur_time = timezone_now()
|
|
time_next_min = (cur_time + timedelta(minutes=1)).replace(second=0, microsecond=0)
|
|
sleep_time = (time_next_min - cur_time).total_seconds()
|
|
time.sleep(sleep_time)
|
|
except KeyboardInterrupt:
|
|
pass
|