from wagtail_factories import StreamFieldFactory

from vbv_lernwelt.assignment.tests.assignment_factories import (
    PerformanceObjectiveBlockFactory,
    AssignmentListPageFactory,
    AssignmentFactory,
    TaskBlockFactory,
)
from vbv_lernwelt.course.consts import COURSE_UK
from vbv_lernwelt.course.models import CoursePage


def create_assignments():
    course_page = CoursePage.objects.get(course_id=COURSE_UK)
    assignment_page = AssignmentListPageFactory(
        parent=course_page,
    )
    AssignmentFactory(
        parent=assignment_page,
        title="Auftrag 1",
        performance_objectives=StreamFieldFactory(
            {
                "performance_objective": PerformanceObjectiveBlockFactory(),
            }
        ),
        effort_required="1 - 2 Stunden",
        assessment_document_url="https://www.vbv.ch",
        tasks=StreamFieldFactory({"task": TaskBlockFactory()}),
    )