add postgres

enrollments-events/app/events/stopgap/patch_course_metadata.py

@@ -8,11 +8,12 @@ from aws_lambda_powertools.utilities.data_classes import (
 )
 from aws_lambda_powertools.utilities.typing import LambdaContext
 from layercake.dateutils import now
-from layercake.dynamodb import DynamoDBPersistenceLayer
+from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair
 from sqlite_utils import Database
 
 from boto3clients import dynamodb_client
 from config import (
+    COURSE_TABLE,
     ENROLLMENT_TABLE,
     SQLITE_DATABASE,
     SQLITE_TABLE,
@@ -22,6 +23,7 @@ sqlite3.register_converter('json', json.loads)
 
 logger = Logger(__name__)
 enrollment_layer = DynamoDBPersistenceLayer(ENROLLMENT_TABLE, dynamodb_client)
+course_layer = DynamoDBPersistenceLayer(COURSE_TABLE, dynamodb_client)
 
 
 @event_source(data_class=EventBridgeEvent)
@@ -37,7 +39,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
             transact.put(
                 item={
                     'id': new_image['id'],
-                    'sk': 'metadata#deduplication_window',
+                    'sk': 'METADATA#DEDUPLICATION_WINDOW',
                     'offset_days': 90,
                     'created_at': now_,
                 }
@@ -45,7 +47,7 @@ def lambda_handler(event: EventBridgeEvent, context: LambdaContext) -> bool:
             transact.put(
                 item={
                     'id': new_image['id'],
-                    'sk': 'metadata#course',
+                    'sk': 'METADATA#COURSE',
                     'created_at': now_,
                     'access_period': int(course['access_period']),
                     'cert': {
@@ -66,6 +68,10 @@ class CourseNotFoundError(Exception):
 
 
 def _get_course(course_id: str) -> dict:
+    course = course_layer.get_item(KeyPair(pk=course_id, sk='0'))
+    if course:
+        return course
+
     with sqlite3.connect(
         database=SQLITE_DATABASE, detect_types=sqlite3.PARSE_DECLTYPES
     ) as conn: