From addd73609b56ad39a28f902c0c486f73edcb273b Mon Sep 17 00:00:00 2001 From: Guy Davis Date: Mon, 4 Mar 2024 10:52:32 +1300 Subject: [PATCH] [guivis/KeepPolling#17] - fix question mark for questions - create some blank users - added run all function to do all the setup required --- populate.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) 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()