add postgres

enrollments-events/app/enrollment.py

@@ -1,6 +1,4 @@
 from dataclasses import asdict, dataclass
-from datetime import timedelta
-from enum import Enum
 from typing import Self, TypedDict
 
 from layercake.dateutils import now, ttl
@@ -36,35 +34,6 @@ class Slot:
         return LinkedEntity(idx, 'ORDER')
 
 
-class LifecycleEvents(str, Enum):
-    """Lifecycle events related to scheduling actions."""
-
-    # Reminder if the user does not access within 3 days
-    # REMINDER_NO_ACCESS_AFTER_3_DAYS = 'SCHEDULE#REMINDER_NO_ACCESS_AFTER_3_DAYS'
-    DOES_NOT_ACCESS = 'schedules#does_not_access'
-
-    # When there is no activity 7 days after the first access
-    # REMINDER_NO_ACTIVITY_AFTER_7_DAYS = 'SCHEDULE#REMINDER_NO_ACTIVITY_AFTER_7_DAYS'
-    NO_ACTIVITY = 'schedules#no_activity'
-
-    # Reminder 30 days before the access period expires
-    # REMINDER_ACCESS_PERIOD_BEFORE_30_DAYS = 'SCHEDULE#REMINDER_ACCESS_PERIOD_BEFORE_30_DAYS'
-    ACCESS_PERIOD_ENDS = 'schedules#access_period_ends'
-
-    # Reminder for certificate expiration set to 30 days from now
-    REMINDER_CERT_EXPIRATION_BEFORE_30_DAYS = (
-        'SCHEDULE#REMINDER_CERT_EXPIRATION_BEFORE_30_DAYS'
-    )
-
-    # Archive the course after the certificate expires
-    # SET_AS_ARCHIVE = 'SCHEDULE#SET_AS_ARCHIVE'
-    ARCHIVE_IT = 'schedules#archive_it'
-
-    # When the access period ends for a course without a certificate
-    # SET_AS_EXPIRE = 'SCHEDULE#SET_AS_EXPIRE'
-    EXPIRATION = 'schedules#expiration'
-
-
 class DeduplicationConflictError(Exception):
     def __init__(self, *args):
         super().__init__('Enrollment already exists')
@@ -105,53 +74,11 @@ def enroll(
         transact.put(
             item={
                 'id': enrollment.id,
-                'sk': 'metadata#course',
+                'sk': 'METADATA#COURSE',
                 'created_at': now_,
                 **course.model_dump(include={'cert', 'access_period'}),
             }
         )
-        transact.put(
-            item={
-                'id': enrollment.id,
-                # Post-migration: uncomment the following line
-                # 'sk': LifecycleEvents.REMINDER_NO_ACCESS_3_DAYS,
-                'sk': LifecycleEvents.DOES_NOT_ACCESS,
-                'name': user.name,
-                'email': user.email,
-                'course': course.name,
-                'created_at': now_,
-                'ttl': ttl(days=3, start_dt=now_),
-            },
-        )
-        # Enrollment expires by default when the access period ends.
-        # When the course is completed, it is automatically removed,
-        # and the `SCHEDULE#SET_AS_ARCHIVE` event is created.
-        transact.put(
-            item={
-                'id': enrollment.id,
-                'sk': LifecycleEvents.EXPIRATION,
-                # Post-migration: uncomment the following line
-                # 'sk': LifecycleEvents.COURSE_EXPIRED,
-                'name': user.name,
-                'email': user.email,
-                'course': course.name,
-                'created_at': now_,
-                'ttl': ttl(start_dt=now_ + timedelta(days=course.access_period)),
-            },
-        )
-        transact.put(
-            item={
-                'id': enrollment.id,
-                # Post-migration: uncomment the following line
-                # 'sk': LifecycleEvents.ACCESS_PERIOD_REMINDER_30_DAYS,
-                'sk': LifecycleEvents.ACCESS_PERIOD_ENDS,
-                'name': user.name,
-                'email': user.email,
-                'course': course.name,
-                'created_at': now_,
-                'ttl': ttl(start_dt=now_ + timedelta(days=course.access_period - 30)),
-            },
-        )
 
         for entity in linked_entities:
             keyprefix = entity.type.lower()
@@ -184,7 +111,7 @@ def enroll(
             transact.put(
                 item={
                     'id': enrollment.id,
-                    'sk': 'cancel_policy',
+                    'sk': 'CANCEL_POLICY',
                     'created_at': now_,
                 }
             )
@@ -204,16 +131,17 @@ def enroll(
         # the deduplication window expires or is removed.
         if deduplication_window:
             offset_days = deduplication_window['offset_days']
-            ttl_expiration = ttl(
-                start_dt=now_ + timedelta(days=course.access_period - offset_days)
+            ttl_ = ttl(
+                start_dt=now_,
+                days=course.access_period - offset_days,
             )
             transact.put(
                 item={
-                    'id': 'lock',
+                    'id': 'LOCK',
                     'sk': lock_hash,
                     'enrollment_id': enrollment.id,
                     'created_at': now_,
-                    'ttl': ttl_expiration,
+                    'ttl': ttl_,
                 },
                 cond_expr='attribute_not_exists(sk)',
                 exc_cls=DeduplicationConflictError,
@@ -221,24 +149,24 @@ def enroll(
             transact.put(
                 item={
                     'id': enrollment.id,
-                    'sk': 'lock',
+                    'sk': 'LOCK',
                     'hash': lock_hash,
                     'created_at': now_,
-                    'ttl': ttl_expiration,
+                    'ttl': ttl_,
                 },
             )
             # Deduplication window can be recalculated if needed
             transact.put(
                 item={
                     'id': enrollment.id,
-                    'sk': 'metadata#deduplication_window',
+                    'sk': 'METADATA#DEDUPLICATION_WINDOW',
                     'offset_days': offset_days,
                     'created_at': now_,
                 },
             )
         else:
             transact.condition(
-                key=KeyPair('lock', lock_hash),
+                key=KeyPair('LOCK', lock_hash),
                 cond_expr='attribute_not_exists(sk)',
                 exc_cls=DeduplicationConflictError,
             )