set code_owner for celery tasks

ARCHBOM-1260

Co-authored-by: Tim McCormack <tmccormack@edx.org>

openedx/core/djangoapps/bookmarks/tasks.py

@@ -8,6 +8,7 @@ import logging
 import six
 from celery.task import task
 from django.db import transaction
+from edx_django_utils.monitoring import set_code_owner_attribute
 from opaque_keys.edx.keys import CourseKey
 
 from xmodule.modulestore.django import modulestore
@@ -145,6 +146,7 @@ def _update_xblocks_cache(course_key):
 
 
 @task(name=u'openedx.core.djangoapps.bookmarks.tasks.update_xblocks_cache')
+@set_code_owner_attribute
 def update_xblocks_cache(course_id):
     """
     Update the XBlocks cache for a course.

openedx/core/djangoapps/ccxcon/tasks.py

@@ -5,6 +5,7 @@ This file contains celery tasks for ccxcon
 
 from celery.task import task
 from celery.utils.log import get_task_logger
+from edx_django_utils.monitoring import set_code_owner_attribute
 from opaque_keys.edx.keys import CourseKey
 from requests.exceptions import ConnectionError, HTTPError, RequestException, TooManyRedirects
 
@@ -14,6 +15,7 @@ log = get_task_logger(__name__)
 
 
 @task(name='openedx.core.djangoapps.ccxcon.tasks.update_ccxcon')
+@set_code_owner_attribute
 def update_ccxcon(course_id, cur_retry=0):
     """
     Pass through function to update course information on CCXCon.

openedx/core/djangoapps/content/course_overviews/tasks.py

@@ -6,6 +6,7 @@ import six
 from celery import task
 from celery_utils.persist_on_failure import LoggedPersistOnFailureTask
 from django.conf import settings
+from edx_django_utils.monitoring import set_code_owner_attribute
 from opaque_keys.edx.keys import CourseKey
 from six.moves import range  # pylint: disable=ungrouped-imports
 
@@ -59,6 +60,7 @@ def enqueue_async_course_overview_update_tasks(
 
 
 @task(base=LoggedPersistOnFailureTask)
+@set_code_owner_attribute
 def async_course_overview_update(*args, **kwargs):
     course_keys = [CourseKey.from_string(arg) for arg in args]
     CourseOverview.update_select_courses(course_keys, force_update=kwargs['force_update'])

openedx/core/djangoapps/coursegraph/tasks.py

@@ -10,6 +10,7 @@ from celery import task
 from django.conf import settings
 from django.utils import six, timezone
 from edx_django_utils.cache import RequestCache
+from edx_django_utils.monitoring import set_code_owner_attribute
 from opaque_keys.edx.keys import CourseKey
 from py2neo import Graph, Node, Relationship, authenticate, NodeSelector
 from py2neo.compat import integer, string
@@ -247,6 +248,7 @@ def should_dump_course(course_key, graph):
 
 
 @task(routing_key=settings.COURSEGRAPH_JOB_QUEUE)
+@set_code_owner_attribute
 def dump_course_to_neo4j(course_key_string, credentials):
     """
     Serializes a course and writes it to neo4j.

openedx/core/djangoapps/credentials/tasks/v1/tasks.py

@@ -7,6 +7,7 @@ from celery import task
 from celery.utils.log import get_task_logger
 from django.conf import settings
 from django.contrib.auth.models import User
+from edx_django_utils.monitoring import set_code_owner_attribute
 from opaque_keys.edx.keys import CourseKey
 
 from openedx.core.djangoapps.credentials.utils import get_credentials_api_client
@@ -26,6 +27,7 @@ MAX_RETRIES = 11
 
 
 @task(bind=True, ignore_result=True, routing_key=ROUTING_KEY)
+@set_code_owner_attribute
 def send_grade_to_credentials(self, username, course_run_key, verified, letter_grade, percent_grade):
     """ Celery task to notify the Credentials IDA of a grade change via POST. """
     logger.info(u'Running task send_grade_to_credentials for username %s and course %s', username, course_run_key)

openedx/core/djangoapps/credit/tasks.py

@@ -7,6 +7,7 @@ import six
 from celery import task
 from celery.utils.log import get_task_logger
 from django.conf import settings
+from edx_django_utils.monitoring import set_code_owner_attribute
 from opaque_keys import InvalidKeyError
 from opaque_keys.edx.keys import CourseKey, UsageKey
 
@@ -20,6 +21,7 @@ LOGGER = get_task_logger(__name__)
 
 
 @task(default_retry_delay=settings.CREDIT_TASK_DEFAULT_RETRY_DELAY, max_retries=settings.CREDIT_TASK_MAX_RETRIES)
+@set_code_owner_attribute
 def update_credit_course_requirements(course_id):
     """
     Updates course requirements table for a course.

openedx/core/djangoapps/heartbeat/tasks.py

@@ -5,8 +5,10 @@ A trivial task for health checks
 
 from celery.task import task
 from django.conf import settings
+from edx_django_utils.monitoring import set_code_owner_attribute
 
 
 @task(routing_key=settings.HEARTBEAT_CELERY_ROUTING_KEY)
+@set_code_owner_attribute
 def sample_task():
     return True

openedx/core/djangoapps/programs/tasks.py

@@ -9,6 +9,7 @@ from celery.utils.log import get_task_logger
 from django.conf import settings
 from django.contrib.auth.models import User
 from django.contrib.sites.models import Site
+from edx_django_utils.monitoring import set_code_owner_attribute
 from edx_rest_api_client import exceptions
 from opaque_keys.edx.keys import CourseKey
 
@@ -124,6 +125,7 @@ def award_program_certificate(client, username, program_uuid, visible_date):
 
 
 @task(bind=True, ignore_result=True, routing_key=PROGRAM_CERTIFICATES_ROUTING_KEY)
+@set_code_owner_attribute
 def award_program_certificates(self, username):
     """
     This task is designed to be called whenever a student's completion status
@@ -286,6 +288,7 @@ def post_course_certificate(client, username, certificate, visible_date):
 
 
 @task(bind=True, ignore_result=True, routing_key=ROUTING_KEY)
+@set_code_owner_attribute
 def award_course_certificate(self, username, course_run_key):
     """
     This task is designed to be called whenever a student GeneratedCertificate is updated.
@@ -400,6 +403,7 @@ def revoke_program_certificate(client, username, program_uuid):
 
 
 @task(bind=True, ignore_result=True, routing_key=PROGRAM_CERTIFICATES_ROUTING_KEY)
+@set_code_owner_attribute
 def revoke_program_certificates(self, username, course_key):
     """
     This task is designed to be called whenever a student's course certificate is
@@ -523,6 +527,7 @@ def revoke_program_certificates(self, username, course_key):
 
 
 @task(bind=True, ignore_result=True, routing_key=PROGRAM_CERTIFICATES_ROUTING_KEY)
+@set_code_owner_attribute
 def update_certificate_visible_date_on_course_update(self, course_key):
     """
     This task is designed to be called whenever a course is updated with

openedx/core/djangoapps/schedules/tasks.py

@@ -16,7 +16,7 @@ from django.db.utils import DatabaseError
 from edx_ace import ace
 from edx_ace.message import Message
 from edx_ace.utils.date import deserialize, serialize
-from edx_django_utils.monitoring import set_custom_attribute
+from edx_django_utils.monitoring import set_code_owner_attribute, set_custom_attribute
 from eventtracking import tracker
 from opaque_keys.edx.keys import CourseKey
 
@@ -43,6 +43,7 @@ COURSE_NEXT_SECTION_UPDATE_LOG_PREFIX = 'Course Next Section Update'
 
 
 @task(base=LoggedPersistOnFailureTask, bind=True, default_retry_delay=30)
+@set_code_owner_attribute
 def update_course_schedules(self, **kwargs):
     course_key = CourseKey.from_string(kwargs['course_id'])
     new_start_date = deserialize(kwargs['new_start_date_str'])
@@ -144,6 +145,7 @@ class BinnedScheduleMessageBaseTask(ScheduleMessageBaseTask):
 
 
 @task(base=LoggedTask, ignore_result=True, routing_key=ROUTING_KEY)
+@set_code_owner_attribute
 def _recurring_nudge_schedule_send(site_id, msg_str):
     _schedule_send(
         msg_str,
@@ -154,6 +156,7 @@ def _recurring_nudge_schedule_send(site_id, msg_str):
 
 
 @task(base=LoggedTask, ignore_result=True, routing_key=ROUTING_KEY)
+@set_code_owner_attribute
 def _upgrade_reminder_schedule_send(site_id, msg_str):
     _schedule_send(
         msg_str,
@@ -164,6 +167,7 @@ def _upgrade_reminder_schedule_send(site_id, msg_str):
 
 
 @task(base=LoggedTask, ignore_result=True, routing_key=ROUTING_KEY)
+@set_code_owner_attribute
 def _course_update_schedule_send(site_id, msg_str):
     _schedule_send(
         msg_str,

openedx/core/djangoapps/verified_track_content/tasks.py

@@ -8,6 +8,7 @@ import six
 from celery.task import task
 from celery.utils.log import get_task_logger
 from django.contrib.auth.models import User
+from edx_django_utils.monitoring import set_code_owner_attribute
 from opaque_keys.edx.keys import CourseKey
 
 from openedx.core.djangoapps.course_groups.cohorts import add_user_to_cohort, get_cohort, get_cohort_by_name
@@ -17,6 +18,7 @@ LOGGER = get_task_logger(__name__)
 
 
 @task(bind=True, default_retry_delay=60, max_retries=2)
+@set_code_owner_attribute
 def sync_cohort_with_mode(self, course_id, user_id, verified_cohort_name, default_cohort_name):
     """
     If the learner's mode does not match their assigned cohort, move the learner into the correct cohort.