feat: adds get_containers and get_units to the authoring API

Author: pomegranited

openedx_learning/apps/authoring/publishing/api.py

@@ -71,6 +71,7 @@
     "create_container_version",
     "create_next_container_version",
     "get_container",
+    "get_containers",
     "ContainerEntityListEntry",
     "get_entities_in_container",
     "contains_unpublished_changes",
@@ -837,6 +838,25 @@ def get_container(pk: int) -> Container:
     return Container.objects.get(pk=pk)
 
 
+def get_containers(
+    learning_package_id: int,
+    container_cls: type[ContainerModel] = Container,  # type: ignore[assignment]
+) -> QuerySet[ContainerModel]:
+    """
+    [ 🛑 UNSTABLE ]
+    Get all containers in the given learning package.
+
+    Args:
+        learning_package_id: The primary key of the learning package
+        container_cls: The subclass of Container to use, if applicable
+
+    Returns:
+        A queryset containing the container associated with the given learning package.
+    """
+    assert issubclass(container_cls, Container)
+    return container_cls.objects.filter(publishable_entity__learning_package=learning_package_id)
+
+
 @dataclass(frozen=True)
 class ContainerEntityListEntry:
     """

openedx_learning/apps/authoring/units/api.py

@@ -5,6 +5,7 @@
 from dataclasses import dataclass
 from datetime import datetime
 
+from django.db.models import QuerySet
 from django.db.transaction import atomic
 
 from openedx_learning.apps.authoring.components.models import Component, ComponentVersion
@@ -21,6 +22,7 @@
     "create_unit_and_version",
     "get_unit",
     "get_unit_version",
+    "get_units",
     "get_latest_unit_version",
     "UnitListEntry",
     "get_components_in_unit",
@@ -211,6 +213,16 @@ def get_latest_unit_version(unit_pk: int) -> UnitVersion:
     return Unit.objects.get(pk=unit_pk).versioning.latest
 
 
+def get_units(learning_package_id: int) -> QuerySet[Unit]:
+    """
+    [ 🛑 UNSTABLE ] Get all units in a given learning package.
+
+    Args:
+        learning_package_id: The learning package ID.
+    """
+    return publishing_api.get_containers(learning_package_id, container_cls=Unit)
+
+
 @dataclass(frozen=True)
 class UnitListEntry:
     """

tests/openedx_learning/apps/authoring/units/test_api.py

@@ -90,6 +90,18 @@ def test_get_unit(self):
         with self.assertNumQueries(0):
             assert result.versioning.has_unpublished_changes
 
+    def test_get_units(self):
+        """
+        Test get_units()
+        """
+        unit = self.create_unit_with_components([])
+        with self.assertNumQueries(1):
+            result = list(authoring_api.get_units(self.learning_package.id))
+        assert result == [unit]
+        # Versioning data should be pre-loaded via select_related()
+        with self.assertNumQueries(0):
+            assert result[0].versioning.has_unpublished_changes
+
     def test_get_container(self):
         """
         Test get_container()