From 61c2e484fe66979947044b8a1551b90966df9c92 Mon Sep 17 00:00:00 2001
From: Prashant Makwana <pmakwana@2u.com>
Date: Thu, 24 Oct 2024 15:19:09 +0000
Subject: [PATCH] feat: updating is_active determination logic for v2 endpoint

---
 enterprise_catalog/apps/api/v2/utils.py       | 53 +++++++++++++++++++
 ...enterprise_catalog_get_content_metadata.py | 20 +++++++
 2 files changed, 73 insertions(+)
 create mode 100644 enterprise_catalog/apps/api/v2/utils.py

diff --git a/enterprise_catalog/apps/api/v2/utils.py b/enterprise_catalog/apps/api/v2/utils.py
new file mode 100644
index 00000000..1e1e2601
--- /dev/null
+++ b/enterprise_catalog/apps/api/v2/utils.py
@@ -0,0 +1,53 @@
+import logging
+
+from enterprise_catalog.apps.catalog.constants import (
+    COURSE_RUN_RESTRICTION_TYPE_KEY,
+    RESTRICTION_FOR_B2B,
+)
+
+
+logger = logging.getLogger(__name__)
+
+
+def is_course_run_active(course_run):
+    """
+    Checks whether a course run is active. That is, whether the course run is published,
+    enrollable, and either marketable, or has a b2b restriction type. To ellaborate on the latter:
+
+    Restricted course run records will be set with `is_marketable: false` from the
+    upstream source-of-truth (course-discovery).  But because our discovery <-> catalog
+    synchronization has business logic that filters course run json metadata (inside of courses)
+    to only the *allowed* restricted runs for a catalog, we can safely assume
+    when looking at a course run metadata record in the context of a catalog,
+    if that run has a non-null, B2B restriction type, then it is permitted to be
+    part of the catalog and should be considered active (as long as it is published and enrollable).
+
+    Arguments:
+        course_run (dict): The metadata about a course run.
+
+    Returns:
+        bool: True if the course run is "active"
+    """
+    course_run_status = course_run.get('status') or ''
+    is_published = course_run_status.lower() == 'published'
+    is_enrollable = course_run.get('is_enrollable', False)
+    is_marketable = course_run.get('is_marketable', False)
+    is_restricted = course_run.get(COURSE_RUN_RESTRICTION_TYPE_KEY) == RESTRICTION_FOR_B2B
+
+    return is_published and is_enrollable and (is_marketable or is_restricted)
+
+
+def is_any_course_run_active(course_runs):
+    """
+    Iterates over all course runs to check if there's any course run that is available for enrollment.
+
+    Arguments:
+        course_runs (list): list of course runs
+
+    Returns:
+        bool: True if active course run is found, else False
+    """
+    for course_run in course_runs:
+        if is_course_run_active(course_run):
+            return True
+    return False
diff --git a/enterprise_catalog/apps/api/v2/views/enterprise_catalog_get_content_metadata.py b/enterprise_catalog/apps/api/v2/views/enterprise_catalog_get_content_metadata.py
index 8b05d306..31cb6fb7 100644
--- a/enterprise_catalog/apps/api/v2/views/enterprise_catalog_get_content_metadata.py
+++ b/enterprise_catalog/apps/api/v2/views/enterprise_catalog_get_content_metadata.py
@@ -1,6 +1,8 @@
+from asyncio.log import logger
 from enterprise_catalog.apps.api.v1.views.enterprise_catalog_get_content_metadata import (
     EnterpriseCatalogGetContentMetadata,
 )
+from enterprise_catalog.apps.api.v2.utils import is_any_course_run_active
 
 
 class EnterpriseCatalogGetContentMetadataV2(EnterpriseCatalogGetContentMetadata):
@@ -21,3 +23,21 @@ def get_queryset(self, **kwargs):
             )
 
         return queryset.order_by('catalog_queries')
+
+    def is_active(self, item):
+        """
+        Determines if a content item is active.
+        Args:
+            item (ContentMetadata): The content metadata item to check.
+        Returns:
+            bool: True if the item is active, False otherwise.
+                For courses, checks if any course run is active.
+                For other content types, always returns True.
+        """
+        if item.content_type == 'course':
+            active = is_any_course_run_active(
+                item.json_metadata.get('course_runs', []))
+            if not active:
+                logger.debug(f'[get_content_metadata]: Content item {item.content_key} is not active.')
+            return active
+        return True