import os

import factory
from wagtail.core.models import Collection

from vbv_lernwelt.media_library.models import LibraryDocument
from vbv_lernwelt.media_library.tests.media_library_factories import LibraryDocumentFactory


def create_default_collections():
    c = Collection.objects.all().delete()

    root, created = Collection.objects.get_or_create(name='Root', depth=0)
    versicherungsvermittler = root.add_child(name='Versicherungsvermittler/in')
    handlungsfelder = versicherungsvermittler.add_child(name='Handlungsfelder')

    handlungsfelder_names = ['Fahrzeug', 'Reisen', 'Einkommensicherung', 'Gesundheit', 'Haushalt', 'Sparen',
                             'Pensionierung', 'KMU', 'Wohneigentum', 'Rechtsstreitigkeiten', 'Erben / Vererben',
                             'Selbständigkeit']

    for handlungsfeld in handlungsfelder_names:
        versicherungsvermittler = handlungsfelder.add_child(name=handlungsfeld)


def create_default_documents():
    LibraryDocument.objects.all().delete()

    path = os.path.join(os.path.dirname(os.path.abspath(__file__)),  '../static/media/documents/')

    collection = Collection.objects.get(name='Fahrzeug')

    filename = 'SchweizerischesZivilgesetzbuch.pdf'
    document = LibraryDocumentFactory(
        title='V1 C25 ZGB CH',
        display_text='Schweizerisches Zivilgesetzbuch',
        description='Ein wundervolles Dokument, Bachblüten für Leseratten und metaphysisches Wolbefinden für Handyvekäufer.',
        link_display_text='Dokument laden',
        file=factory.django.FileField(from_path=os.path.join(path, filename), filename=filename),
        collection=collection
    )

    filename = 'SmallPDF.pdf'
    document = LibraryDocumentFactory(
        title='V1 C25 ',
        display_text='Pdf showcase ',
        description='Ein wundervolles Dokument, Bachblüten für Leseratten und metaphysisches Wolbefinden für Handyvekäufer.',
        link_display_text='Dokument laden',
        file=factory.django.FileField(from_path=os.path.join(path, filename), filename=filename),
        collection=collection
    )
    pass