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

feat: add option to clear the database before populating it again

Browse Source
This commit is contained in:
moanos [he/him] 2024-05-30 16:58:11 +02:00
parent 76c24ba694
commit 8295716f07
1 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -7,14 +7,30 @@ from fellchensammlung import baker_recipes
from model_bakery import baker from model_bakery import baker
from fellchensammlung.models import AdoptionNotice, Species, Animal, Image, ModerationAction, User, Member, Rule, \ from fellchensammlung.models import AdoptionNotice, Species, Animal, Image, ModerationAction, User, Member, Rule, \
Report, Comment Report, Comment, ReportAdoptionNotice
class Command(BaseCommand): class Command(BaseCommand):
help = "Populates the database with test data" help = "Populates the database with test data"
def add_arguments(self, parser):
# Named (optional) arguments
parser.add_argument(
"--clear",
action="store_true",
help="Delete existing data",
)
@staticmethod @staticmethod
def populate_db(): def populate_db(clear=False):
if clear:
ModerationAction.objects.all().delete()
Report.objects.all().delete()
Rule.objects.all().delete()
Comment.objects.all().delete()
AdoptionNotice.objects.all().delete()
Animal.objects.all().delete()
User.objects.all().delete()
# Check if there already is and AdoptionNotice named Vermittung1TestSalt9227. If it is, the database # Check if there already is and AdoptionNotice named Vermittung1TestSalt9227. If it is, the database
# is already populated and no data will be added again # is already populated and no data will be added again
@ -65,8 +81,8 @@ class Command(BaseCommand):
title="Rule three", title="Rule three",
rule_text="Everything needs at least three rules") rule_text="Everything needs at least three rules")
report1 = baker.make(Report, adoption_notice=adoption1, reported_broken_rules=[rule1, rule2], report1 = baker.make(ReportAdoptionNotice, adoption_notice=adoption1, reported_broken_rules=[rule1, rule2],
comment="This seems sketchy") user_comment="This seems sketchy")
moderation_action1 = baker.make(ModerationAction, moderation_action1 = baker.make(ModerationAction,
report=report1, report=report1,
@ -96,4 +112,4 @@ class Command(BaseCommand):
reply_to=comment1) reply_to=comment1)
def handle(self, *args, **options): def handle(self, *args, **options):
self.populate_db() self.populate_db(options["clear"])