diff --git a/lms/djangoapps/experiments/audit_expiry_urgency.py b/lms/djangoapps/experiments/audit_expiry_urgency.py index bd34ad65f07b..0c662648d6ad 100644 --- a/lms/djangoapps/experiments/audit_expiry_urgency.py +++ b/lms/djangoapps/experiments/audit_expiry_urgency.py @@ -31,7 +31,8 @@ VARIANT_CONTROL = 'control_5_7_weeks' VARIANT_EXPIRY_7_DAYS = 'expiry_7_days' -VALID_VARIANTS = {VARIANT_CONTROL, VARIANT_EXPIRY_7_DAYS} +VARIANT_EXPIRY_14_DAYS = 'expiry_14_days' +VALID_VARIANTS = {VARIANT_CONTROL, VARIANT_EXPIRY_7_DAYS, VARIANT_EXPIRY_14_DAYS} ATTRIBUTE_NAMESPACE = 'audit_expiry_experiment' ATTRIBUTE_NAME_EXPERIMENT_KEY = 'experiment_key' @@ -204,6 +205,9 @@ def compute_audit_expiry_at(enrollment, variant, access_duration): if variant == VARIANT_EXPIRY_7_DAYS: expiry_at = content_availability_date + timedelta(days=7) return expiry_at, 7 + elif variant == VARIANT_EXPIRY_14_DAYS: + expiry_at = content_availability_date + timedelta(days=14) + return expiry_at, 14 expiry_at = content_availability_date + access_duration return expiry_at, access_duration.days