diff --git a/server/core/management/commands/dummy_data.py b/server/core/management/commands/dummy_data.py index fee625f8..60513856 100644 --- a/server/core/management/commands/dummy_data.py +++ b/server/core/management/commands/dummy_data.py @@ -112,4 +112,5 @@ class Command(BaseCommand): # now create all and rooms management.call_command('dummy_rooms', verbosity=0) + management.call_command('dummy_news', verbosity=0) diff --git a/server/core/management/commands/dummy_rooms.py b/server/core/management/commands/dummy_rooms.py index da5bf02f..2c2a7175 100644 --- a/server/core/management/commands/dummy_rooms.py +++ b/server/core/management/commands/dummy_rooms.py @@ -3,13 +3,10 @@ import random import shutil from django.conf import settings -from django.contrib.auth import get_user_model from django.core.management import BaseCommand -from wagtail.core.models import Site from rooms.factories import RoomFactory, RoomEntryFactory from rooms.models import Room -from users.factories import SchoolClassFactory data = [ { diff --git a/server/newsteaser/management/__init__.py b/server/newsteaser/management/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/server/newsteaser/management/commands/__init__.py b/server/newsteaser/management/commands/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/server/newsteaser/management/commands/dummy_news.py b/server/newsteaser/management/commands/dummy_news.py new file mode 100644 index 00000000..d614e26d --- /dev/null +++ b/server/newsteaser/management/commands/dummy_news.py @@ -0,0 +1,14 @@ +from django.core.management import BaseCommand + +from newsteaser.factories import NewsTeaserFactory +from newsteaser.models import NewsTeaser + + +class Command(BaseCommand): + + def handle(self, *args, **options): + NewsTeaser.objects.all().delete() + + for index in range(9): + NewsTeaserFactory() +