feat(test): Add data to populate basic db
This commit is contained in:
parent
1b9d3056a1
commit
12907a601a
26
src/fellchensammlung/baker_recipes.py
Normal file
26
src/fellchensammlung/baker_recipes.py
Normal 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_'),
|
||||
)
|
33
src/fellchensammlung/management/commands/populate_db.py
Normal file
33
src/fellchensammlung/management/commands/populate_db.py
Normal 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()
|
Loading…
Reference in New Issue
Block a user