From 681ddabaa6688373d3f486d2ce5dbf2d2f2fc711 Mon Sep 17 00:00:00 2001 From: Robert Raposa Date: Wed, 25 Nov 2020 10:19:21 -0500 Subject: [PATCH] add code_owner for schedules tasks Code owner details for monitoring some schedule task transactions were missing. ARCHBOM-1260 --- openedx/core/djangoapps/schedules/tasks.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/openedx/core/djangoapps/schedules/tasks.py b/openedx/core/djangoapps/schedules/tasks.py index 33c5cd9419..3693ea75ca 100644 --- a/openedx/core/djangoapps/schedules/tasks.py +++ b/openedx/core/djangoapps/schedules/tasks.py @@ -16,7 +16,11 @@ 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_code_owner_attribute, set_custom_attribute +from edx_django_utils.monitoring import ( + set_code_owner_attribute, + set_code_owner_attribute_from_module, + set_custom_attribute +) from eventtracking import tracker from opaque_keys.edx.keys import CourseKey @@ -102,6 +106,7 @@ class BinnedScheduleMessageBaseTask(ScheduleMessageBaseTask): @classmethod def enqueue(cls, site, current_date, day_offset, override_recipient_email=None): + set_code_owner_attribute_from_module(__name__) current_date = resolvers._get_datetime_beginning_of_day(current_date) if not cls.is_enqueue_enabled(site): @@ -127,6 +132,7 @@ class BinnedScheduleMessageBaseTask(ScheduleMessageBaseTask): def run( self, site_id, target_day_str, day_offset, bin_num, override_recipient_email=None, ): + set_code_owner_attribute_from_module(__name__) site = Site.objects.select_related('configuration').get(id=site_id) with emulate_http_request(site=site): msg_type = self.make_message_type(day_offset) @@ -228,6 +234,7 @@ class ScheduleCourseNextSectionUpdate(ScheduleMessageBaseTask): @classmethod def enqueue(cls, site, current_date, day_offset, override_recipient_email=None): + set_code_owner_attribute_from_module(__name__) target_datetime = (current_date - datetime.timedelta(days=day_offset)) if not cls.is_enqueue_enabled(site): @@ -249,6 +256,7 @@ class ScheduleCourseNextSectionUpdate(ScheduleMessageBaseTask): ) def run(self, site_id, target_day_str, course_key, override_recipient_email=None): + set_code_owner_attribute_from_module(__name__) site = Site.objects.select_related('configuration').get(id=site_id) with emulate_http_request(site=site): _annotate_for_monitoring(message_types.CourseUpdate(), site, 0, target_day_str, -1)