You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.7 KiB
51 lines
1.7 KiB
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...') |
|
|
|
# create test users |
|
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!') |
|
|
|
# create an account that all 3 users are attached to |
|
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 test stacks for account |
|
Stack.objects.all().delete() |
|
groceries = Stack(name="Groceries", |
|
details="", |
|
amount=800).save() |
|
|
|
takeout = Stack(name="Date Nights", |
|
details="", |
|
amount=400).save() |
|
|
|
shit = Stack(name="Crap", |
|
details="", |
|
amount=500).save() |
|
|
|
Transaction.objects.all().delete() |
|
# generate transaction history |
|
|
|
self.stdout.write('done!')
|
|
|