from server.api.models import Transaction, Account, User, Stack from django.core.management.base import BaseCommand class Command(BaseCommand): help = "Seed database for dev" def handle(self, *args, **options): self.stdout.write('clearing and seeding database...') User.objects.all().delete() elijah = User(username="elijah", email="elijah@westwinds.io", password="toffee15").save() ievgen = User(username="ievgen", email="ievgen@westwinds.io", password="toffee15").save() tanner = User(username="tanner", email="tanner@westwinds.io", password="toffee15").save() self.stdout.write(str(User.objects.all())) self.stdout.write('Users Created!') Account.objects.all().delete() home = Account(name="home", details="home stacks", income=4000, expenses=2000).save() self.stdout.write(str(Account.objects.all())) self.stdout.write('Accounts Created!') # create seed accounts # Home Account # $3000 # - home expenses Stack.objects.all().delete() # create test stacks # - name: shit Transaction.objects.all().delete() # generate transaction history self.stdout.write('done!')