diff --git a/populate.py b/populate.py index ba58daa..979eafb 100644 --- a/populate.py +++ b/populate.py @@ -13,7 +13,7 @@ def populate(): for i in range(number_of_polls): poll = Poll() question = data_maker.sentence() - question.replace(".", "?") + question = question.replace(".", "?") poll.question = question poll.save() @@ -52,3 +52,20 @@ def create_users_and_votes(): vote.option = Option.objects.get(pk=option_id) vote.user = user vote.save() + + +def create_blank_users(): + from django.contrib.auth.models import User + + users_to_create = 16 + for i in range(users_to_create): + User.objects.create_user( + f"user{i}", + password="password", + ) + + +def run_all(): + populate() + create_users_and_votes() + create_blank_users()