feat(test): Add data to populate basic db

This commit is contained in:
moanos [he/him] 2024-03-18 13:57:56 +01:00
parent 1b9d3056a1
commit 12907a601a
2 changed files with 59 additions and 0 deletions

View File

@ -0,0 +1,26 @@
from django.contrib.auth.models import User
from model_bakery.recipe import Recipe, seq
from fellchensammlung.models import *
location = Recipe(
Location,
name=seq('Ort_'),
description=seq('Detaillierte Beschreibung_'),
postcode=seq("7322"),
)
rescue_org = Recipe(
RescueOrganization,
name=seq('Rattennothilfe_'),
location=location.make()
)
rat = Recipe(
Animal,
name=seq('Ratte_'),
)
cat = Recipe(
Animal,
name=seq('Katze_'),
)

View File

@ -0,0 +1,33 @@
from django.core.management.base import BaseCommand, CommandError
from django.contrib.auth.models import User
from fellchensammlung.models import *
from fellchensammlung import baker_recipes
from model_bakery import baker
class Command(BaseCommand):
help = "Populates the database with test data"
@staticmethod
def populate_db():
rat1 = baker.make_recipe(
'fellchensammlung.rat'
)
rat2 = baker.make_recipe(
'fellchensammlung.rat'
)
rescue1 = baker.make_recipe(
'fellchensammlung.rescue_org'
)
rescue2 = baker.make_recipe(
'fellchensammlung.rescue_org'
)
baker.make(AdoptionNotice, name="Vermittung1", animals=[rat1, rat2], organization=rescue1)
User.objects.create_user('test', password='foobar')
User.objects.create_superuser(username="admin", password="admin")
def handle(self, *args, **options):
self.populate_db()