95 lines
3.0 KiB
Python
95 lines
3.0 KiB
Python
import datetime
|
|
|
|
from celery.task import task
|
|
from django.conf import settings
|
|
from django.contrib.sites.models import Site
|
|
from django.core.urlresolvers import reverse
|
|
from django.utils.http import urlquote
|
|
|
|
from edx_ace import ace
|
|
from edx_ace.message import MessageType, Message
|
|
from edx_ace.recipient import Recipient
|
|
from edx_ace.utils.date import deserialize
|
|
from openedx.core.djangoapps.schedules.models import Schedule, ScheduleConfig
|
|
|
|
|
|
ROUTING_KEY = getattr(settings, 'ACE_ROUTING_KEY', None)
|
|
|
|
|
|
class RecurringNudge(MessageType):
|
|
def __init__(self, day, *args, **kwargs):
|
|
super(RecurringNudge, self).__init__(*args, **kwargs)
|
|
self.name = "recurringnudge_day{}".format(day)
|
|
|
|
|
|
@task(ignore_result=True, routing_key=ROUTING_KEY)
|
|
def recurring_nudge_schedule_hour(
|
|
site_id, day, target_hour_str, org_list, exclude_orgs=False, override_recipient_email=None,
|
|
):
|
|
target_hour = deserialize(target_hour_str)
|
|
msg_type = RecurringNudge(day)
|
|
|
|
for (user, language, context) in _recurring_nudge_schedules_for_hour(target_hour, org_list, exclude_orgs):
|
|
msg = msg_type.personalize(
|
|
Recipient(
|
|
user.username,
|
|
override_recipient_email or user.email,
|
|
),
|
|
language,
|
|
context,
|
|
)
|
|
_recurring_nudge_schedule_send.apply_async((site_id, str(msg)), retry=False)
|
|
|
|
|
|
@task(ignore_result=True, routing_key=ROUTING_KEY)
|
|
def _recurring_nudge_schedule_send(site_id, msg_str):
|
|
site = Site.objects.get(pk=site_id)
|
|
if not ScheduleConfig.current(site).deliver_recurring_nudge:
|
|
return
|
|
|
|
msg = Message.from_string(msg_str)
|
|
ace.send(msg)
|
|
|
|
|
|
def _recurring_nudge_schedules_for_hour(target_hour, org_list, exclude_orgs=False):
|
|
schedules = Schedule.objects.select_related(
|
|
'enrollment__user__profile',
|
|
'enrollment__course',
|
|
).filter(
|
|
start__gte=target_hour,
|
|
start__lt=target_hour + datetime.timedelta(minutes=60),
|
|
enrollment__is_active=True,
|
|
)
|
|
|
|
if org_list is not None:
|
|
if exclude_orgs:
|
|
schedules = schedules.exclude(enrollment__course__org__in=org_list)
|
|
else:
|
|
schedules = schedules.filter(enrollment__course__org__in=org_list)
|
|
|
|
if "read_replica" in settings.DATABASES:
|
|
schedules = schedules.using("read_replica")
|
|
|
|
for schedule in schedules:
|
|
enrollment = schedule.enrollment
|
|
user = enrollment.user
|
|
|
|
course_id_str = str(enrollment.course_id)
|
|
course = enrollment.course
|
|
|
|
course_root = reverse('course_root', args=[course_id_str])
|
|
|
|
def absolute_url(relative_path):
|
|
return u'{}{}'.format(settings.LMS_ROOT_URL, urlquote(relative_path))
|
|
|
|
template_context = {
|
|
'student_name': user.profile.name,
|
|
'course_name': course.display_name,
|
|
'course_url': absolute_url(course_root),
|
|
|
|
# This is used by the bulk email optout policy
|
|
'course_id': course_id_str,
|
|
}
|
|
|
|
yield (user, course.language, template_context)
|