from courses.api import CourseProgressApi
from progress.models import ProgressLesson


def add_next_lesson(progress):
    lesson_list = CourseProgressApi.get_next(
        progress.course_token,
        [i.lesson_token for i in progress.progresslesson_set.all()]
    )

    for lesson, is_hw in lesson_list:
        if is_hw:
            ProgressLesson.objects.get(
                progress=progress,
                lesson_token=lesson,
                checker=progress.teacher,
            )
        else:
            ProgressLesson.objects.get(
                progress=progress,
                lesson_token=lesson,
                checker=progress.user,
            )

    return "access update course %s" % progress.course_token