feat: add management command for course prompts

CHANGELOG.rst

@@ -11,6 +11,10 @@ Change Log
 
 .. There should always be an "Unreleased" section for changes pending release.
 
+1.5.0 - 2023-10-18
+******************
+* Add management command to generate course prompts
+
 1.4.0 - 2023-09-11
 ******************
 * Send reduced message list if needed to avoid going over token limit

learning_assistant/__init__.py

@@ -2,6 +2,6 @@
 Plugin for a learning assistant backend, intended for use within edx-platform.
 """
 
-__version__ = '1.4.0'
+__version__ = '1.5.0'
 
 default_app_config = 'learning_assistant.apps.LearningAssistantConfig'  # pylint: disable=invalid-name

learning_assistant/management/commands/set_course_prompts.py

@@ -0,0 +1,109 @@
+"""
+Django management command to generate course prompts.
+"""
+import json
+import logging
+from posixpath import join as urljoin
+
+from django.conf import settings
+from django.core.management.base import BaseCommand
+from edx_rest_api_client.client import OAuthAPIClient
+from opaque_keys.edx.keys import CourseKey
+
+try:
+    from openedx.core.djangoapps.oauth_dispatch.jwt import create_jwt_for_user
+except ImportError:
+    create_jwt_for_user = None
+
+from learning_assistant.models import CoursePrompt
+
+logger = logging.getLogger(__name__)
+
+
+class Command(BaseCommand):
+    """
+    Django Management command to create a set of course prompts
+    """
+
+    def add_arguments(self, parser):
+
+        # list of course ids
+        parser.add_argument(
+            '--course_ids',
+            dest='course_ids',
+            help='Comma separated list of course_ids to generate. Only newer style course ids can be supplied.',
+        )
+
+        # pre-message
+        parser.add_argument(
+            '--pre_message',
+            dest='pre_message',
+            help='Message to prepend to course topics',
+        )
+
+        parser.add_argument(
+            '--skills_descriptor',
+            dest='skills_descriptor',
+            help='Message that describes skill structure'
+        )
+
+        # post-message
+        parser.add_argument(
+            '--post_message',
+            dest='post_message',
+            help='Message to append to course topics',
+        )
+
+    @staticmethod
+    def _get_discovery_api_client():
+        """
+        Returns an API client which can be used to make Catalog API requests.
+        """
+        return OAuthAPIClient(
+            base_url=settings.DISCOVERY_BACKEND_SERVICE_EDX_OAUTH2_PROVIDER_URL,
+            client_id=settings.DISCOVERY_BACKEND_SERVICE_EDX_OAUTH2_KEY,
+            client_secret=settings.DISCOVERY_BACKEND_SERVICE_EDX_OAUTH2_SECRET,
+        )
+
+    def handle(self, *args, **options):
+        """
+        Management command entry point.
+
+        This command is meant to generate a small (<500) set of course prompts. If a larger number of prompts
+        should be created, consider adding batching to this command.
+        """
+        course_ids = options['course_ids']
+        pre_message = options['pre_message']
+        skills_descriptor = options['skills_descriptor']
+        post_message = options['post_message']
+
+        client = self._get_discovery_api_client()
+
+        course_ids_list = course_ids.split(',')
+        for course_run_id in course_ids_list:
+            course_key = CourseKey.from_string(course_run_id)
+
+            # discovery API requires course keys, not course run keys
+            course_id = f'{course_key.org}+{course_key.course}'
+
+            url = urljoin(
+                settings.DISCOVERY_BASE_URL,
+                'api/v1/courses/{course_id}'.format(course_id=course_id)
+            )
+            response_data = client.get(url).json()
+            title = response_data['title']
+            skill_names = response_data['skill_names']
+
+            # create restructured dictionary with data
+            course_dict = {'title': title, 'topics': skill_names}
+
+            # append descriptor message and decode json dict into a string
+            skills_message = skills_descriptor + json.dumps(course_dict)
+
+            # finally, create list of prompt messages and save
+            prompt_messages = [pre_message, skills_message, post_message]
+            CoursePrompt.objects.update_or_create(
+                course_id=course_run_id, json_prompt_content=prompt_messages
+            )
+
+            logger.info('Updated course prompt for course_run_id=%s', course_run_id)

learning_assistant/management/commands/tests/test_set_course_prompts.py

@@ -0,0 +1,73 @@
+"""
+Tests for the set_course_prompts management command.
+"""
+import json
+from posixpath import join as urljoin
+from unittest.mock import MagicMock, patch
+from requests import Response
+
+from django.conf import settings
+from django.core.management import call_command
+from django.test import TestCase
+
+from learning_assistant.models import CoursePrompt
+
+
+class SetCoursePromptsTests(TestCase):
+    """Test set_course_prompts command"""
+    command = 'set_course_prompts'
+
+    def setUp(self):
+        self.pre_message = 'This is the first message'
+        self.skills_descriptor = 'These are the skills: '
+        self.post_message = 'This message comes after'
+        self.course_ids = 'course-v1:edx+test+23,course-v1:edx+test+24'
+        self.course_title = 'Intro to Testing'
+        self.skill_names = ['Testing', 'Computers', 'Coding']
+
+    def get_mock_discovery_response(self):
+        """
+        Create scaled down mock of discovery response
+        """
+        response_data = {
+            'title': self.course_title,
+            'skill_names': self.skill_names
+        }
+        return response_data
+
+    @patch('learning_assistant.management.commands.set_course_prompts.Command._get_discovery_api_client')
+    def test_course_prompts_created(self, mock_get_discovery_client):
+        """
+        Assert that course prompts are created by calling management command.
+        """
+        mock_client = MagicMock()
+        mock_get_discovery_client.return_value = mock_client
+        mock_client.get.return_value = MagicMock(status_code=200, json=lambda : self.get_mock_discovery_response())
+
+        call_command(
+            self.command,
+            course_ids=self.course_ids,
+            pre_message=self.pre_message,
+            skills_descriptor=self.skills_descriptor,
+            post_message=self.post_message,
+        )
+
+        # assert that discovery api was called with course id, not course run id
+        expected_url = urljoin(
+            settings.DISCOVERY_BASE_URL,
+            'api/v1/courses/{course_id}'.format(course_id='edx+test')
+        )
+        mock_client.get.assert_any_call(expected_url)
+        mock_client.get.assert_called()
+
+        # assert that number of prompts created is equivalent to number of courses passed in to command
+        prompts = CoursePrompt.objects.filter()
+        self.assertEqual(len(prompts), len(self.course_ids.split(',')))
+
+        # assert structure of prompt
+        course_prompt = prompts[0].json_prompt_content
+        self.assertEqual(len(course_prompt), 3)
+
+        skills_message = self.skills_descriptor + json.dumps({'title': self.course_title, 'topics': self.skill_names})
+        expected_response = [self.pre_message, skills_message, self.post_message]
+        self.assertEqual(course_prompt, expected_response)

requirements/base.in

@@ -5,4 +5,5 @@ Django             # Web application framework
 django-model-utils
 djangorestframework
 edx-drf-extensions
+edx-rest-api-client
 edx-opaque-keys

requirements/base.txt

@@ -18,7 +18,7 @@ click==8.1.7
     # via edx-django-utils
 cryptography==41.0.4
     # via pyjwt
-django==3.2.21
+django==3.2.22
     # via
     #   -c https://raw.githubusercontent.com/edx/edx-lint/master/edx_lint/files/common_constraints.txt
     #   -r requirements/base.in
@@ -45,27 +45,33 @@ djangorestframework==3.14.0
 drf-jwt==1.19.2
     # via edx-drf-extensions
 edx-django-utils==5.7.0
-    # via edx-drf-extensions
-edx-drf-extensions==8.10.0
+    # via
+    #   edx-drf-extensions
+    #   edx-rest-api-client
+edx-drf-extensions==8.12.0
     # via -r requirements/base.in
 edx-opaque-keys==2.5.1
     # via
     #   -r requirements/base.in
     #   edx-drf-extensions
+edx-rest-api-client==5.6.1
+    # via -r requirements/base.in
 idna==3.4
     # via requests
 newrelic==9.1.0
     # via edx-django-utils
 pbr==5.11.1
     # via stevedore
-psutil==5.9.5
+psutil==5.9.6
     # via edx-django-utils
 pycparser==2.21
     # via cffi
 pyjwt[crypto]==2.8.0
     # via
     #   drf-jwt
     #   edx-drf-extensions
+    #   edx-rest-api-client
+    #   pyjwt
 pymongo==3.13.0
     # via edx-opaque-keys
 pynacl==1.5.0
@@ -75,9 +81,14 @@ pytz==2023.3.post1
     #   django
     #   djangorestframework
 requests==2.31.0
-    # via edx-drf-extensions
+    # via
+    #   edx-drf-extensions
+    #   edx-rest-api-client
+    #   slumber
 semantic-version==2.10.0
     # via edx-drf-extensions
+slumber==0.7.1
+    # via edx-rest-api-client
 sqlparse==0.4.4
     # via django
 stevedore==5.1.0
@@ -88,5 +99,5 @@ typing-extensions==4.8.0
     # via
     #   asgiref
     #   edx-opaque-keys
-urllib3==2.0.6
+urllib3==2.0.7
     # via requests

requirements/ci.txt

@@ -41,7 +41,7 @@ tox==3.28.0
     #   tox-battery
 tox-battery==0.6.2
     # via -r requirements/ci.in
-urllib3==2.0.6
+urllib3==2.0.7
     # via requests
 virtualenv==20.24.5
     # via tox