54 lines
2.1 KiB
Python
54 lines
2.1 KiB
Python
from wagtail.core.models import Collection
|
|
import factory
|
|
from django.core.files import File
|
|
import os
|
|
|
|
from vbv_lernwelt.media_library.models import LibraryDocument
|
|
from vbv_lernwelt.media_library.tests.media_library_factories import LibraryDocumentFactory
|
|
from factory.django import DjangoModelFactory
|
|
|
|
|
|
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__)), '../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
|
|
|