261 lines
9.1 KiB
Python
261 lines
9.1 KiB
Python
import datetime
|
|
import logging
|
|
|
|
import analytics
|
|
from celery import task
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.sites.models import Site
|
|
from django.core.exceptions import ValidationError
|
|
|
|
from django.db.utils import DatabaseError
|
|
|
|
from celery_utils.logged_task import LoggedTask
|
|
from celery_utils.persist_on_failure import LoggedPersistOnFailureTask
|
|
from edx_ace import ace
|
|
from edx_ace.message import Message
|
|
from edx_ace.utils.date import deserialize, serialize
|
|
from opaque_keys.edx.keys import CourseKey
|
|
|
|
from openedx.core.djangoapps.monitoring_utils import set_custom_metric
|
|
from openedx.core.djangoapps.schedules import message_types
|
|
from openedx.core.djangoapps.schedules.models import Schedule, ScheduleConfig
|
|
from openedx.core.djangoapps.schedules import resolvers
|
|
from openedx.core.lib.celery.task_utils import emulate_http_request
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
ROUTING_KEY = getattr(settings, 'ACE_ROUTING_KEY', None)
|
|
KNOWN_RETRY_ERRORS = ( # Errors we expect occasionally that could resolve on retry
|
|
DatabaseError,
|
|
ValidationError,
|
|
)
|
|
|
|
|
|
RECURRING_NUDGE_LOG_PREFIX = 'Recurring Nudge'
|
|
UPGRADE_REMINDER_LOG_PREFIX = 'Upgrade Reminder'
|
|
COURSE_UPDATE_LOG_PREFIX = 'Course Update'
|
|
|
|
|
|
@task(base=LoggedPersistOnFailureTask, bind=True, default_retry_delay=30, routing_key=ROUTING_KEY)
|
|
def update_course_schedules(self, **kwargs):
|
|
course_key = CourseKey.from_string(kwargs['course_id'])
|
|
new_start_date = deserialize(kwargs['new_start_date_str'])
|
|
new_upgrade_deadline = deserialize(kwargs['new_upgrade_deadline_str'])
|
|
|
|
try:
|
|
Schedule.objects.filter(enrollment__course_id=course_key).update(
|
|
start=new_start_date,
|
|
upgrade_deadline=new_upgrade_deadline
|
|
)
|
|
except Exception as exc: # pylint: disable=broad-except
|
|
if not isinstance(exc, KNOWN_RETRY_ERRORS):
|
|
LOG.exception("Unexpected failure: task id: %s, kwargs=%s".format(self.request.id, kwargs))
|
|
raise self.retry(kwargs=kwargs, exc=exc)
|
|
|
|
|
|
class ScheduleMessageBaseTask(LoggedTask):
|
|
"""
|
|
Base class for top-level Schedule tasks that create subtasks
|
|
for each Bin.
|
|
"""
|
|
ignore_result = True
|
|
routing_key = ROUTING_KEY
|
|
num_bins = resolvers.DEFAULT_NUM_BINS
|
|
enqueue_config_var = None # define in subclass
|
|
log_prefix = None
|
|
resolver = None # define in subclass
|
|
async_send_task = None # define in subclass
|
|
|
|
@classmethod
|
|
def log_debug(cls, message, *args, **kwargs):
|
|
"""
|
|
Wrapper around LOG.debug that prefixes the message.
|
|
"""
|
|
LOG.debug(cls.log_prefix + ': ' + message, *args, **kwargs)
|
|
|
|
@classmethod
|
|
def log_info(cls, message, *args, **kwargs):
|
|
"""
|
|
Wrapper around LOG.info that prefixes the message.
|
|
"""
|
|
LOG.info(cls.log_prefix + ': ' + message, *args, **kwargs)
|
|
|
|
@classmethod
|
|
def enqueue(cls, site, current_date, day_offset, override_recipient_email=None):
|
|
current_date = resolvers._get_datetime_beginning_of_day(current_date)
|
|
|
|
if not cls.is_enqueue_enabled(site):
|
|
cls.log_info('Message queuing disabled for site %s', site.domain)
|
|
return
|
|
|
|
target_date = current_date + datetime.timedelta(days=day_offset)
|
|
cls.log_info('Target date = %s', target_date.isoformat())
|
|
for bin in range(cls.num_bins):
|
|
task_args = (
|
|
site.id,
|
|
serialize(target_date),
|
|
day_offset,
|
|
bin,
|
|
override_recipient_email,
|
|
)
|
|
cls.log_info('Launching task with args = %r', task_args)
|
|
cls().apply_async(
|
|
task_args,
|
|
retry=False,
|
|
)
|
|
|
|
@classmethod
|
|
def is_enqueue_enabled(cls, site):
|
|
if cls.enqueue_config_var:
|
|
return getattr(ScheduleConfig.current(site), cls.enqueue_config_var)
|
|
return False
|
|
|
|
def run(
|
|
self, site_id, target_day_str, day_offset, bin_num, override_recipient_email=None,
|
|
):
|
|
site = Site.objects.select_related('configuration').get(id=site_id)
|
|
with emulate_http_request(site=site):
|
|
msg_type = self.make_message_type(day_offset)
|
|
_annotate_for_monitoring(msg_type, site, bin_num, target_day_str, day_offset)
|
|
return self.resolver(
|
|
self.async_send_task,
|
|
site,
|
|
deserialize(target_day_str),
|
|
day_offset,
|
|
bin_num,
|
|
override_recipient_email=override_recipient_email,
|
|
).send(msg_type)
|
|
|
|
def make_message_type(self, day_offset):
|
|
raise NotImplementedError
|
|
|
|
|
|
@task(base=LoggedTask, ignore_result=True, routing_key=ROUTING_KEY)
|
|
def _recurring_nudge_schedule_send(site_id, msg_str):
|
|
_schedule_send(
|
|
msg_str,
|
|
site_id,
|
|
'deliver_recurring_nudge',
|
|
RECURRING_NUDGE_LOG_PREFIX,
|
|
)
|
|
|
|
|
|
@task(base=LoggedTask, ignore_result=True, routing_key=ROUTING_KEY)
|
|
def _upgrade_reminder_schedule_send(site_id, msg_str):
|
|
_schedule_send(
|
|
msg_str,
|
|
site_id,
|
|
'deliver_upgrade_reminder',
|
|
UPGRADE_REMINDER_LOG_PREFIX,
|
|
)
|
|
|
|
|
|
@task(base=LoggedTask, ignore_result=True, routing_key=ROUTING_KEY)
|
|
def _course_update_schedule_send(site_id, msg_str):
|
|
_schedule_send(
|
|
msg_str,
|
|
site_id,
|
|
'deliver_course_update',
|
|
COURSE_UPDATE_LOG_PREFIX,
|
|
)
|
|
|
|
|
|
class ScheduleRecurringNudge(ScheduleMessageBaseTask):
|
|
num_bins = resolvers.RECURRING_NUDGE_NUM_BINS
|
|
enqueue_config_var = 'enqueue_recurring_nudge'
|
|
log_prefix = RECURRING_NUDGE_LOG_PREFIX
|
|
resolver = resolvers.RecurringNudgeResolver
|
|
async_send_task = _recurring_nudge_schedule_send
|
|
|
|
def make_message_type(self, day_offset):
|
|
return message_types.RecurringNudge(abs(day_offset))
|
|
|
|
|
|
class ScheduleUpgradeReminder(ScheduleMessageBaseTask):
|
|
num_bins = resolvers.UPGRADE_REMINDER_NUM_BINS
|
|
enqueue_config_var = 'enqueue_upgrade_reminder'
|
|
log_prefix = UPGRADE_REMINDER_LOG_PREFIX
|
|
resolver = resolvers.UpgradeReminderResolver
|
|
async_send_task = _upgrade_reminder_schedule_send
|
|
|
|
def make_message_type(self, day_offset):
|
|
return message_types.UpgradeReminder()
|
|
|
|
|
|
class ScheduleCourseUpdate(ScheduleMessageBaseTask):
|
|
num_bins = resolvers.COURSE_UPDATE_NUM_BINS
|
|
enqueue_config_var = 'enqueue_course_update'
|
|
log_prefix = COURSE_UPDATE_LOG_PREFIX
|
|
resolver = resolvers.CourseUpdateResolver
|
|
async_send_task = _course_update_schedule_send
|
|
|
|
def make_message_type(self, day_offset):
|
|
return message_types.CourseUpdate()
|
|
|
|
|
|
def _schedule_send(msg_str, site_id, delivery_config_var, log_prefix):
|
|
site = Site.objects.select_related('configuration').get(pk=site_id)
|
|
if _is_delivery_enabled(site, delivery_config_var, log_prefix):
|
|
msg = Message.from_string(msg_str)
|
|
|
|
user = User.objects.get(username=msg.recipient.username)
|
|
with emulate_http_request(site=site, user=user):
|
|
_annonate_send_task_for_monitoring(msg)
|
|
LOG.debug('%s: Sending message = %s', log_prefix, msg_str)
|
|
ace.send(msg)
|
|
_track_message_sent(site, user, msg)
|
|
|
|
|
|
def _track_message_sent(site, user, msg):
|
|
properties = {
|
|
'site': site.domain,
|
|
'app_label': msg.app_label,
|
|
'name': msg.name,
|
|
'language': msg.language,
|
|
'uuid': unicode(msg.uuid),
|
|
'send_uuid': unicode(msg.send_uuid),
|
|
}
|
|
course_ids = msg.context.get('course_ids', [])
|
|
properties['num_courses'] = len(course_ids)
|
|
if len(course_ids) > 0:
|
|
properties['course_ids'] = course_ids[:10]
|
|
properties['primary_course_id'] = course_ids[0]
|
|
|
|
analytics.track(
|
|
user_id=user.id,
|
|
event='edx.bi.email.sent',
|
|
properties=properties
|
|
)
|
|
|
|
|
|
def _is_delivery_enabled(site, delivery_config_var, log_prefix):
|
|
if getattr(ScheduleConfig.current(site), delivery_config_var, False):
|
|
return True
|
|
else:
|
|
LOG.info('%s: Message delivery disabled for site %s', log_prefix, site.domain)
|
|
|
|
|
|
def _annotate_for_monitoring(message_type, site, bin_num, target_day_str, day_offset):
|
|
# This identifies the type of message being sent, for example: schedules.recurring_nudge3.
|
|
set_custom_metric('message_name', '{0}.{1}'.format(message_type.app_label, message_type.name))
|
|
# The domain name of the site we are sending the message for.
|
|
set_custom_metric('site', site.domain)
|
|
# This is the "bin" of data being processed. We divide up the work into chunks so that we don't tie up celery
|
|
# workers for too long. This could help us identify particular bins that are problematic.
|
|
set_custom_metric('bin', bin_num)
|
|
# The date we are processing data for.
|
|
set_custom_metric('target_day', target_day_str)
|
|
# The number of days relative to the current date to process data for.
|
|
set_custom_metric('day_offset', day_offset)
|
|
# A unique identifier for this batch of messages being sent.
|
|
set_custom_metric('send_uuid', message_type.uuid)
|
|
|
|
|
|
def _annonate_send_task_for_monitoring(msg):
|
|
# A unique identifier for this batch of messages being sent.
|
|
set_custom_metric('send_uuid', msg.send_uuid)
|
|
# A unique identifier for this particular message.
|
|
set_custom_metric('uuid', msg.uuid)
|