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()]
    )

    pl = None

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

    return pl