import wagtail_factories
from wagtail.core.models import Site

from vbv_lernwelt.learnpath.models import LearningPath, Topic, Circle, LearningSequence, LearningUnit
from vbv_lernwelt.learnpath.tests.learningpath_factories import LearningPathFactory, TopicFactory, CircleFactory, LearningSequenceFactory, LearningUnitFactory


def create_default_learning_path():
    site = Site.objects.filter(is_default_site=True).first()

    if not site:
        site = wagtail_factories.SiteFactory.get(is_default_site=True)

    lp = LearningPathFactory.create(title="Versicherungsvermittler/in", parent=site.root_page)

    tp = TopicFactory.create(title="Basis", is_visible=False, parent=lp)

    circle_1 = CircleFactory.create(title="Basis", parent=tp, description="""In diesem Circle erklären wir dir, wie der Lehrgang
    Versicherungsvermittler / in " aufgebaut ist. Zudem vermitteln wir dir die wichtigsten Grundlagen,
     damit erfolgreich mit deinem Lernpfad starten kannst.""")

    ls_1 = LearningSequenceFactory.create(title='Einleitung', parent=circle_1)
    lu_1 = LearningUnitFactory.create(title="Herzlich Willkommmen", parent=ls_1)

    ls_2 = LearningSequenceFactory.create(title='Grundlagen', parent=circle_1)
    lu_1 = LearningUnitFactory.create(title="Herzlich Willkommmen", parent=ls_2)

    tp = TopicFactory.create(title="Gewinnen von Kunden", parent=lp)
    circle_2 = CircleFactory.create(title="Gewinnen", parent=tp)

    tp = TopicFactory.create(title="Beraten der Kunden", parent=lp)
    circle_3 = CircleFactory.create(title="Einstieg", parent=tp)
    circle_4 = CircleFactory.create(title="Analyse", parent=tp)
    circle_5 = CircleFactory.create(title="Lösung", parent=tp)
    circle_6 = CircleFactory.create(title="Abschluss", parent=tp)

    tp = TopicFactory.create(title="Betreuen und Ausbauen des Kundenstamms", parent=lp)
    circle_7 = CircleFactory.create(title="Betreuen", parent=tp)

    tp = TopicFactory.create(title="Prüfung", is_visible=False, parent=lp)
    circle_7 = CircleFactory.create(title="Prüfungsvorbereitung", parent=tp)


def delete_default_learning_path():
    LearningUnit.objects.all().delete()
    LearningSequence.objects.all().delete()
    Circle.objects.all().delete()
    Topic.objects.all().delete()
    LearningPath.objects.all().delete()