from marshmallow import Schema, fields

from ..validators import validate_datetime_greater_than_now


class EnrollmentCreateSchema(Schema):
    start_date = fields.DateTime(validate=validate_datetime_greater_than_now, required=True)
    end_date = fields.DateTime(validate=validate_datetime_greater_than_now, required=True)