b2b40e2770
The course now supports reading and writing the verification deadline attribute. XCOM-536
64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
""" API v1 serializers. """
|
|
from datetime import datetime
|
|
|
|
import pytz
|
|
from rest_framework import serializers
|
|
|
|
from commerce.api.v1.models import Course
|
|
from course_modes.models import CourseMode
|
|
|
|
|
|
class CourseModeSerializer(serializers.ModelSerializer):
|
|
""" CourseMode serializer. """
|
|
name = serializers.CharField(source='mode_slug')
|
|
price = serializers.IntegerField(source='min_price')
|
|
expires = serializers.DateTimeField(source='expiration_datetime', required=False, blank=True)
|
|
|
|
def get_identity(self, data):
|
|
try:
|
|
return data.get('name', None)
|
|
except AttributeError:
|
|
return None
|
|
|
|
class Meta(object): # pylint: disable=missing-docstring
|
|
model = CourseMode
|
|
fields = ('name', 'currency', 'price', 'sku', 'expires')
|
|
|
|
|
|
class CourseSerializer(serializers.Serializer):
|
|
""" Course serializer. """
|
|
id = serializers.CharField() # pylint: disable=invalid-name
|
|
name = serializers.CharField(read_only=True)
|
|
verification_deadline = serializers.DateTimeField(blank=True)
|
|
modes = CourseModeSerializer(many=True, allow_add_remove=True)
|
|
|
|
def validate(self, attrs):
|
|
""" Ensure the verification deadline occurs AFTER the course mode enrollment deadlines. """
|
|
verification_deadline = attrs.get('verification_deadline', None)
|
|
|
|
if verification_deadline:
|
|
upgrade_deadline = None
|
|
|
|
# Find the earliest upgrade deadline
|
|
for mode in attrs['modes']:
|
|
expires = mode.expiration_datetime
|
|
if expires:
|
|
# If we don't already have an upgrade_deadline value, use datetime.max so that we can actually
|
|
# complete the comparison.
|
|
upgrade_deadline = min(expires, upgrade_deadline or datetime.max.replace(tzinfo=pytz.utc))
|
|
|
|
# In cases where upgrade_deadline is None (e.g. the verified professional mode), allow a verification
|
|
# deadline to be set anyway.
|
|
if upgrade_deadline is not None and verification_deadline < upgrade_deadline:
|
|
raise serializers.ValidationError(
|
|
'Verification deadline must be after the course mode upgrade deadlines.')
|
|
|
|
return attrs
|
|
|
|
def restore_object(self, attrs, instance=None):
|
|
if instance is None:
|
|
return Course(attrs['id'], attrs['modes'], attrs['verification_deadline'])
|
|
|
|
instance.update(attrs)
|
|
return instance
|