76 lines
3.3 KiB
Python
76 lines
3.3 KiB
Python
"""
|
|
Contain celery tasks
|
|
"""
|
|
from celery import shared_task
|
|
from django.contrib.auth import get_user_model
|
|
from edx_django_utils.monitoring import set_code_owner_attribute
|
|
from opaque_keys.edx.locator import CourseKey
|
|
|
|
from lms.djangoapps.courseware.courses import get_course_with_access
|
|
from lms.djangoapps.discussion.rest_api.discussions_notifications import DiscussionNotificationSender
|
|
from openedx.core.djangoapps.django_comment_common.comment_client import Comment
|
|
from openedx.core.djangoapps.django_comment_common.comment_client.thread import Thread
|
|
from openedx.core.djangoapps.notifications.config.waffle import ENABLE_NOTIFICATIONS
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
@shared_task
|
|
@set_code_owner_attribute
|
|
def send_thread_created_notification(thread_id, course_key_str, user_id):
|
|
"""
|
|
Send notification when a new thread is created
|
|
"""
|
|
course_key = CourseKey.from_string(course_key_str)
|
|
if not ENABLE_NOTIFICATIONS.is_enabled(course_key):
|
|
return
|
|
thread = Thread(id=thread_id).retrieve()
|
|
user = User.objects.get(id=user_id)
|
|
course = get_course_with_access(user, 'load', course_key, check_if_enrolled=True)
|
|
notification_sender = DiscussionNotificationSender(thread, course, user)
|
|
notification_sender.send_new_thread_created_notification()
|
|
|
|
|
|
@shared_task
|
|
@set_code_owner_attribute
|
|
def send_response_notifications(thread_id, course_key_str, user_id, comment_id, parent_id=None):
|
|
"""
|
|
Send notifications to users who are subscribed to the thread.
|
|
"""
|
|
course_key = CourseKey.from_string(course_key_str)
|
|
if not ENABLE_NOTIFICATIONS.is_enabled(course_key):
|
|
return
|
|
thread = Thread(id=thread_id).retrieve()
|
|
user = User.objects.get(id=user_id)
|
|
course = get_course_with_access(user, 'load', course_key, check_if_enrolled=True)
|
|
notification_sender = DiscussionNotificationSender(thread, course, user, parent_id, comment_id)
|
|
notification_sender.send_new_comment_notification()
|
|
notification_sender.send_new_response_notification()
|
|
notification_sender.send_new_comment_on_response_notification()
|
|
notification_sender.send_response_on_followed_post_notification()
|
|
|
|
|
|
@shared_task
|
|
@set_code_owner_attribute
|
|
def send_response_endorsed_notifications(thread_id, response_id, course_key_str, endorsed_by):
|
|
"""
|
|
Send notifications when a response is marked answered/ endorsed
|
|
"""
|
|
course_key = CourseKey.from_string(course_key_str)
|
|
if not ENABLE_NOTIFICATIONS.is_enabled(course_key):
|
|
return
|
|
thread = Thread(id=thread_id).retrieve()
|
|
response = Comment(id=response_id).retrieve()
|
|
creator = User.objects.get(id=response.user_id)
|
|
endorser = User.objects.get(id=endorsed_by)
|
|
course = get_course_with_access(creator, 'load', course_key, check_if_enrolled=True)
|
|
notification_sender = DiscussionNotificationSender(thread, course, creator, comment_id=response_id)
|
|
# skip sending notification to author of thread if they are the same as the author of the response
|
|
if response.user_id != thread.user_id:
|
|
# sends notification to author of thread
|
|
notification_sender.send_response_endorsed_on_thread_notification()
|
|
# sends notification to author of response
|
|
if int(response.user_id) != endorser.id:
|
|
notification_sender.creator = User.objects.get(id=response.user_id)
|
|
notification_sender.send_response_endorsed_notification()
|