feat: Add REST endpoints to update Components in a Collections (temp)

docs/hooks/events.rst

@@ -247,3 +247,15 @@ Content Authoring Events
    * - `CONTENT_OBJECT_TAGS_CHANGED <https://github.com/openedx/openedx-events/blob/c0eb4ba1a3d7d066d58e5c87920b8ccb0645f769/openedx_events/content_authoring/signals.py#L207>`_
      - org.openedx.content_authoring.content.object.tags.changed.v1
      - 2024-03-31
+
+   * - `LIBRARY_COLLECTION_CREATED <https://github.com/openedx/openedx-events/blob/main/openedx_events/content_authoring/signals.py#L219>`_
+     - org.openedx.content_authoring.content.library.collection.created.v1
+     - 2024-08-23
+
+   * - `LIBRARY_COLLECTION_UPDATED <https://github.com/openedx/openedx-events/blob/main/openedx_events/content_authoring/signals.py#L230>`_
+     - org.openedx.content_authoring.content.library.collection.updated.v1
+     - 2024-08-23
+
+   * - `LIBRARY_COLLECTION_DELETED <https://github.com/openedx/openedx-events/blob/main/openedx_events/content_authoring/signals.py#L241>`_
+     - org.openedx.content_authoring.content.library.collection.deleted.v1
+     - 2024-08-23

openedx/core/djangoapps/content_libraries/api.py

@@ -69,15 +69,20 @@
 from django.utils.translation import gettext as _
 from edx_rest_api_client.client import OAuthAPIClient
 from lxml import etree
-from opaque_keys.edx.keys import UsageKey, UsageKeyV2
+from opaque_keys.edx.keys import BlockTypeKey, UsageKey, UsageKeyV2
 from opaque_keys.edx.locator import (
     LibraryLocatorV2,
     LibraryUsageLocatorV2,
     LibraryLocator as LibraryLocatorV1
 )
 from opaque_keys import InvalidKeyError
-from openedx_events.content_authoring.data import ContentLibraryData, LibraryBlockData
+from openedx_events.content_authoring.data import (
+    ContentLibraryData,
+    ContentObjectData,
+    LibraryBlockData,
+)
 from openedx_events.content_authoring.signals import (
+    CONTENT_OBJECT_TAGS_CHANGED,
     CONTENT_LIBRARY_CREATED,
     CONTENT_LIBRARY_DELETED,
     CONTENT_LIBRARY_UPDATED,
@@ -86,7 +91,7 @@
     LIBRARY_BLOCK_UPDATED,
 )
 from openedx_learning.api import authoring as authoring_api
-from openedx_learning.api.authoring_models import Component, MediaType
+from openedx_learning.api.authoring_models import Collection, Component, MediaType, LearningPackage
 from organizations.models import Organization
 from xblock.core import XBlock
 from xblock.exceptions import XBlockNotFoundError
@@ -150,6 +155,7 @@ class ContentLibraryMetadata:
     Class that represents the metadata about a content library.
     """
     key = attr.ib(type=LibraryLocatorV2)
+    learning_package = attr.ib(type=LearningPackage)
     title = attr.ib("")
     description = attr.ib("")
     num_blocks = attr.ib(0)
@@ -323,13 +329,14 @@ def get_metadata(queryset, text_search=None):
             has_unpublished_changes=False,
             has_unpublished_deletes=False,
             license=lib.license,
+            learning_package=lib.learning_package,
         )
         for lib in queryset
     ]
     return libraries
 
 
-def require_permission_for_library_key(library_key, user, permission):
+def require_permission_for_library_key(library_key, user, permission) -> ContentLibrary:
     """
     Given any of the content library permission strings defined in
     openedx.core.djangoapps.content_libraries.permissions,
@@ -339,10 +346,12 @@ def require_permission_for_library_key(library_key, user, permission):
     Raises django.core.exceptions.PermissionDenied if the user doesn't have
     permission.
     """
-    library_obj = ContentLibrary.objects.get_by_key(library_key)
+    library_obj = ContentLibrary.objects.get_by_key(library_key)  # type: ignore[attr-defined]
     if not user.has_perm(permission, obj=library_obj):
         raise PermissionDenied
 
+    return library_obj
+
 
 def get_library(library_key):
     """
@@ -408,6 +417,7 @@ def get_library(library_key):
         license=ref.license,
         created=learning_package.created,
         updated=learning_package.updated,
+        learning_package=learning_package
     )
 
 
@@ -479,6 +489,7 @@ def create_library(
         allow_public_learning=ref.allow_public_learning,
         allow_public_read=ref.allow_public_read,
         license=library_license,
+        learning_package=ref.learning_package
     )
 
 
@@ -1056,6 +1067,69 @@ def revert_changes(library_key):
     )
 
 
+def update_collection_contents(
+    library: ContentLibraryMetadata,
+    collection_pk: int,
+    usage_keys: list[UsageKeyV2],
+    remove=False,
+) -> int:
+    """
+    Associates the Collection with Components for the given UsageKeys.
+
+    By default the Components are added to the Collection.
+    If remove=True, the Components are removed from the Collection.
+
+    Raises:
+    * Collection.DoesNotExist if no Collection with the given pk is found in the given library.
+    * Component.DoesNotExist if any of the given usage_keys don't correspond to a Component in the
+      library/learning_package.
+    """
+    learning_package = library.learning_package
+    collections_qset = authoring_api.get_learning_package_collections(
+        learning_package.id,
+    ).filter(pk=collection_pk)
+
+    collection = collections_qset.first()
+    if not collection:
+        raise Collection.DoesNotExist()
+
+    # Fetch the Component.key values for the provided UsageKeys.
+    component_keys = []
+    for usage_key in usage_keys:
+        # Parse the block_family from the key to use as namespace.
+        block_type = BlockTypeKey.from_string(str(usage_key))
+
+        # Can raise Component.DoesNotExist
+        component = authoring_api.get_component_by_key(
+            learning_package.id,
+            namespace=block_type.block_family,
+            type_name=usage_key.block_type,
+            local_key=usage_key.block_id,
+        )
+        component_keys.append(component.key)
+
+    # Note: Component.key matches its PublishableEntity.key
+    contents_qset = learning_package.publishable_entities.filter(
+        key__in=component_keys,
+    )
+    if not contents_qset.count():
+        raise Component.DoesNotExist()
+
+    if remove:
+        count = authoring_api.remove_from_collections(collections_qset, contents_qset)
+    else:
+        count = authoring_api.add_to_collections(collections_qset, contents_qset)
+
+    # Emit a CONTENT_OBJECT_TAGS_CHANGED event for each of the objects added/removed
+    object_keys = contents_qset.values_list("key", flat=True)
+    for object_key in object_keys:
+        CONTENT_OBJECT_TAGS_CHANGED.send_event(
+            content_object=ContentObjectData(object_id=object_key),
+        )
+
+    return count
+
+
 # V1/V2 Compatibility Helpers
 # (Should be removed as part of
 #  https://github.com/openedx/edx-platform/issues/32457)

openedx/core/djangoapps/content_libraries/apps.py

@@ -37,3 +37,4 @@ def ready(self):
         Import signal handler's module to ensure they are registered.
         """
         from . import signal_handlers  # pylint: disable=unused-import
+        from .collections import handlers  # pylint: disable=unused-import

openedx/core/djangoapps/content_libraries/collections/handlers.py

@@ -0,0 +1,57 @@
+"""
+Signal handlers for Content Library Collections.
+"""
+
+import logging
+
+from django.dispatch import receiver
+from openedx_events.content_authoring.data import LibraryCollectionData
+from openedx_events.content_authoring.signals import (
+    LIBRARY_COLLECTION_CREATED,
+    LIBRARY_COLLECTION_UPDATED,
+    LIBRARY_COLLECTION_DELETED,
+)
+
+
+log = logging.getLogger(__name__)
+
+
+@receiver(LIBRARY_COLLECTION_CREATED)
+def library_collection_created_handler(**kwargs):
+    """
+    Content Library Collection Created signal handler
+    """
+    library_collection_data = kwargs.get("library_collection", None)
+    if not library_collection_data or not isinstance(library_collection_data, LibraryCollectionData):
+        log.error("Received null or incorrect data for event")
+        return
+
+    log.info("Received Collection Created Signal")
+
+    # TODO: Implement handler logic
+
+
+@receiver(LIBRARY_COLLECTION_UPDATED)
+def library_collection_updated_handler(**kwargs):
+    """
+    Content Library Collection Updated signal handler
+    """
+    library_collection_data = kwargs.get("library_collection", None)
+    if not library_collection_data or not isinstance(library_collection_data, LibraryCollectionData):
+        log.error("Received null or incorrect data for event")
+        return
+
+    log.info("Received Collection Updated Signal")
+
+
+@receiver(LIBRARY_COLLECTION_DELETED)
+def library_collection_deleted_handler(**kwargs):
+    """
+    Content Library Collection Deleted signal handler
+    """
+    library_collection_data = kwargs.get("library_collection", None)
+    if not library_collection_data or not isinstance(library_collection_data, LibraryCollectionData):
+        log.error("Received null or incorrect data for event")
+        return
+
+    log.info("Received Collection Deleted Signal")