refactor!: use LibraryContainerLocator in LibraryContainerData

The library_key can be inferred from the container_key

Author: pomegranited

openedx_events/content_authoring/data.py

@@ -12,7 +12,12 @@
 
 import attr
 from opaque_keys.edx.keys import CourseKey, UsageKey
-from opaque_keys.edx.locator import LibraryCollectionLocator, LibraryLocatorV2, LibraryUsageLocatorV2
+from opaque_keys.edx.locator import (
+    LibraryCollectionLocator,
+    LibraryContainerLocator,
+    LibraryLocatorV2,
+    LibraryUsageLocatorV2,
+)
 
 
 @attr.s(frozen=True)
@@ -233,12 +238,10 @@ class LibraryContainerData:
     Data related to a library container that has changed.
 
     Attributes:
-        library_key (LibraryLocatorV2): a key that represents a content library.
-        container_key (str): identifies the container within the library's learning package (e.g.  unit, section)
+        container_key (LibraryContainerLocator): identifies the container (e.g.  unit, section)
         background (bool): indicate whether the sender doesn't want to wait for handler to finish execution,
            i.e., the handler can run the task in background. By default it is False.
     """
 
-    library_key = attr.ib(type=LibraryLocatorV2)
-    container_key = attr.ib(type=str)
+    container_key = attr.ib(type=LibraryContainerLocator)
     background = attr.ib(type=bool, default=False)

openedx_events/event_bus/avro/custom_serializers.py

@@ -8,7 +8,12 @@
 
 from ccx_keys.locator import CCXLocator
 from opaque_keys.edx.keys import CourseKey, UsageKey
-from opaque_keys.edx.locator import LibraryCollectionLocator, LibraryLocatorV2, LibraryUsageLocatorV2
+from opaque_keys.edx.locator import (
+    LibraryCollectionLocator,
+    LibraryContainerLocator,
+    LibraryLocatorV2,
+    LibraryUsageLocatorV2,
+)
 
 from openedx_events.event_bus.avro.types import PYTHON_TYPE_TO_AVRO_MAPPING
 
@@ -131,6 +136,25 @@ def deserialize(data: str):
         return LibraryCollectionLocator.from_string(data)
 
 
+class LibraryContainerLocatorAvroSerializer(BaseCustomTypeAvroSerializer):
+    """
+    CustomTypeAvroSerializer for LibraryContainerLocator class.
+    """
+
+    cls = LibraryContainerLocator
+    field_type = PYTHON_TYPE_TO_AVRO_MAPPING[str]
+
+    @staticmethod
+    def serialize(obj) -> str:
+        """Serialize obj into string."""
+        return str(obj)
+
+    @staticmethod
+    def deserialize(data: str):
+        """Deserialize string into obj."""
+        return LibraryContainerLocator.from_string(data)
+
+
 class LibraryLocatorV2AvroSerializer(BaseCustomTypeAvroSerializer):
     """
     CustomTypeAvroSerializer for LibraryLocatorV2 class.
@@ -195,6 +219,7 @@ def deserialize(data: str):
     CcxCourseLocatorAvroSerializer,
     DatetimeAvroSerializer,
     LibraryCollectionLocatorAvroSerializer,
+    LibraryContainerLocatorAvroSerializer,
     LibraryLocatorV2AvroSerializer,
     LibraryUsageLocatorV2AvroSerializer,
     UsageKeyAvroSerializer,

openedx_events/event_bus/avro/tests/schemas/org+openedx+content_authoring+content_library+container+created+v1_schema.avsc

@@ -9,10 +9,6 @@
         "name": "LibraryContainerData",
         "type": "record",
         "fields": [
-          {
-            "name": "library_key",
-            "type": "string"
-          },
           {
             "name": "container_key",
             "type": "string"

openedx_events/event_bus/avro/tests/schemas/org+openedx+content_authoring+content_library+container+deleted+v1_schema.avsc

@@ -9,10 +9,6 @@
         "name": "LibraryContainerData",
         "type": "record",
         "fields": [
-          {
-            "name": "library_key",
-            "type": "string"
-          },
           {
             "name": "container_key",
             "type": "string"

openedx_events/event_bus/avro/tests/schemas/org+openedx+content_authoring+content_library+container+updated+v1_schema.avsc

@@ -9,10 +9,6 @@
         "name": "LibraryContainerData",
         "type": "record",
         "fields": [
-          {
-            "name": "library_key",
-            "type": "string"
-          },
           {
             "name": "container_key",
             "type": "string"

openedx_events/event_bus/avro/tests/test_avro.py

@@ -11,7 +11,12 @@
 from fastavro.repository.base import SchemaRepositoryError
 from fastavro.schema import load_schema
 from opaque_keys.edx.keys import CourseKey, UsageKey
-from opaque_keys.edx.locator import LibraryCollectionLocator, LibraryLocatorV2, LibraryUsageLocatorV2
+from opaque_keys.edx.locator import (
+    LibraryCollectionLocator,
+    LibraryContainerLocator,
+    LibraryLocatorV2,
+    LibraryUsageLocatorV2,
+)
 
 from openedx_events.event_bus.avro.deserializer import AvroSignalDeserializer, deserialize_bytes_to_event_data
 from openedx_events.event_bus.avro.serializer import AvroSignalSerializer, serialize_event_data_to_bytes
@@ -110,6 +115,9 @@ def generate_test_event_data_for_data_type(data_type):  # pragma: no cover
             "block-v1:edx+DemoX+Demo_course+type@video+block@UaEBjyMjcLW65gaTXggB93WmvoxGAJa0JeHRrDThk",
         ),
         LibraryCollectionLocator: LibraryCollectionLocator.from_string('lib-collection:MITx:reallyhardproblems:col1'),
+        LibraryContainerLocator: LibraryCollectionLocator.from_string(
+            'lct:MITx:reallyhardproblems:unit:test-container',
+        ),
         LibraryLocatorV2: LibraryLocatorV2.from_string('lib:MITx:reallyhardproblems'),
         LibraryUsageLocatorV2: LibraryUsageLocatorV2.from_string('lb:MITx:reallyhardproblems:problem:problem1'),
         List[int]: [1, 2, 3],
@@ -129,6 +137,9 @@ def generate_test_event_data_for_data_type(data_type):  # pragma: no cover
         dict[str, LibraryCollectionLocator]: {
             'key': LibraryCollectionLocator.from_string('lib-collection:MITx:reallyhardproblems:col1'),
         },
+        dict[str, LibraryContainerLocator]: {
+            'key': LibraryContainerLocator.from_string('lct:MITx:reallyhardproblems:unit:test-container'),
+        },
         dict[str, LibraryUsageLocatorV2]: {
             'key': LibraryUsageLocatorV2.from_string('lb:MITx:reallyhardproblems:problem:problem1'),
         },