From 327a5f2ed469fc359eea34f410d8a441809debd7 Mon Sep 17 00:00:00 2001 From: Anthony LC Date: Thu, 30 May 2024 15:57:11 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=A4=A1(demo)=20generate=20dummy=20user?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit To search users we need to have some dummy data users. We add some dummy users when we run the `make demo` command. --- src/backend/demo/defaults.py | 5 +++++ .../demo/management/commands/create_demo.py | 18 ++++++++++++++++++ 2 files changed, 23 insertions(+) create mode 100644 src/backend/demo/defaults.py diff --git a/src/backend/demo/defaults.py b/src/backend/demo/defaults.py new file mode 100644 index 00000000..5c4a4b45 --- /dev/null +++ b/src/backend/demo/defaults.py @@ -0,0 +1,5 @@ +"""Parameters that define how the demo site will be built.""" + +NB_OBJECTS = { + "users": 100, +} diff --git a/src/backend/demo/management/commands/create_demo.py b/src/backend/demo/management/commands/create_demo.py index a8d7e598..3d7b7c71 100644 --- a/src/backend/demo/management/commands/create_demo.py +++ b/src/backend/demo/management/commands/create_demo.py @@ -14,6 +14,8 @@ from faker import Faker from core import models +from demo import defaults + fake = Faker() logger = logging.getLogger("impress.commands.demo.create_demo") @@ -108,6 +110,21 @@ def create_demo(stdout): queue = BulkQueue(stdout) + with Timeit(stdout, "Creating users"): + for i in range(defaults.NB_OBJECTS["users"]): + queue.push( + models.User( + admin_email=f"user{i:d}@example.com", + email=f"user{i:d}@example.com", + password="!", + is_superuser=False, + is_active=True, + is_staff=False, + language=random.choice(settings.LANGUAGES)[0], + ) + ) + queue.flush() + with Timeit(stdout, "Creating Template"): with open( file="demo/data/template/code.txt", mode="r", encoding="utf-8" @@ -121,6 +138,7 @@ def create_demo(stdout): queue.push( models.Template( + id="baca9e2a-59fb-42ef-b5c6-6f6b05637111", title="Demo Template", description="This is the demo template", code=code_data,