moanos/Notfellchen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Don't populate the database again if already done so

Browse Source
This commit is contained in:
moanos [he/him] 2024-04-14 16:20:05 +02:00
parent 7314637520
commit 0bc3c3a764
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/fellchensammlung/management/commands/populate_db.py
View File

@ -1,3 +1,5 @@
import logging
from django.core.management.base import BaseCommand, CommandError from django.core.management.base import BaseCommand, CommandError
from django.core.files import File from django.core.files import File
@ -12,6 +14,15 @@ class Command(BaseCommand):
@staticmethod @staticmethod
def populate_db(): def populate_db():
# Check if there already is and AdoptionNotice named Vermittung1TestSalt9227. If it is, the database
# is already populated and no data will be added again
try:
AdoptionNotice.objects.get(name="Vermittung1TestSalt9227")
except AdoptionNotice.DoesNotExist:
print("Database already populated. No additional data will be created")
return
rescue1 = baker.make_recipe( rescue1 = baker.make_recipe(
'fellchensammlung.rescue_org' 'fellchensammlung.rescue_org'
) )
@ -19,7 +30,7 @@ class Command(BaseCommand):
'fellchensammlung.rescue_org' 'fellchensammlung.rescue_org'
) )
adoption1 = baker.make(AdoptionNotice, name="Vermittung1", organization=rescue1) adoption1 = baker.make(AdoptionNotice, name="Vermittung1TestSalt9227", organization=rescue1)
adoption2 = baker.make(AdoptionNotice, name="Vermittung2", organization=rescue2) adoption2 = baker.make(AdoptionNotice, name="Vermittung2", organization=rescue2)