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