feat: use new setting to set ce_source

CHANGELOG.rst

@@ -14,6 +14,12 @@ Change Log
 Unreleased
 ----------
 
+[9.3.0] - 2024-01-24
+--------------------
+Changed
+~~~~~~~
+* Allow new EVENTS_SERVICE_NAME setting to override SERVICE_VARIANT for data source.
+
 [9.2.0] - 2023-11-16
 --------------------
 Added

openedx_events/__init__.py

@@ -5,4 +5,4 @@
 more information about the project.
 """
 
-__version__ = "9.2.0"
+__version__ = "9.3.0"

openedx_events/data.py

@@ -27,6 +27,19 @@ def _ensure_utc_time(_, attribute, value):
     raise ValueError(f"'{attribute.name}' must have timezone.utc")
 
 
+def get_source_name():
+    """
+    Get the value that should be used for the source of an event.
+    """
+    # .. setting_name: EVENTS_SERVICE_NAME
+    # .. setting_default: None
+    # .. setting_description: Identifier for the producing/consuming service of an event. Used in setting the source in
+    #   the EventsMetadata. If not set, the EventsMetadata object will look for a SERVICE_VARIANT setting (usually only
+    #   set for lms and cms). The full source will be set to openedx/<EVENTS_SERVICE_NAME or SERVICE_VARIANT>/web.
+    #   If neither variable is set, the source will be "unidentified."
+    return getattr(settings, "EVENTS_SERVICE_NAME", None) or getattr(settings, "SERVICE_VARIANT", None)
+
+
 @attr.s(frozen=True)
 class EventsMetadata:
     """
@@ -62,7 +75,8 @@ class EventsMetadata:
     source = attr.ib(
         type=str, default=None,
         converter=attr.converters.default_if_none(
-            attr.Factory(lambda: "openedx/{service}/web".format(service=getattr(settings, "SERVICE_VARIANT", "")))
+            attr.Factory(lambda: "openedx/{service}/web".format(service=get_source_name()) if get_source_name()
+                         else "unidentified")
         ),
         validator=attr.validators.instance_of(str),
     )

openedx_events/tests/test_data.py

@@ -2,11 +2,14 @@
 from datetime import datetime, timezone
 from uuid import UUID
 
+import ddt
 from django.test import TestCase
+from django.test.utils import override_settings
 
 from openedx_events.data import EventsMetadata
 
 
+@ddt.ddt
 class TestEventsMetadata(TestCase):
     """
     Tests for the EventsMetadata class.
@@ -26,3 +29,20 @@ def test_events_metadata_to_and_from_json(self):
         as_json = self.metadata.to_json()
         from_json = EventsMetadata.from_json(as_json)
         self.assertEqual(self.metadata, from_json)
+
+    @ddt.data(
+        ('settings_variant', None, 'openedx/settings_variant/web'),
+        (None, 'my_service', 'openedx/my_service/web'),
+        (None, None, 'unidentified'),
+        ('settings_variant', 'my_service', 'openedx/my_service/web')
+    )
+    @ddt.unpack
+    def test_events_metadata_source(self, settings_variant, event_bus_service_name, expected_source):
+        with override_settings(
+                SERVICE_VARIANT=settings_variant,
+                EVENTS_SERVICE_NAME=event_bus_service_name,
+        ):
+            metadata = EventsMetadata(
+                event_type='test_type'
+            )
+            self.assertEqual(metadata.source, expected_source)