From 332ffa3a5f7105c51cc6a3b5558902f6e6ea1206 Mon Sep 17 00:00:00 2001 From: phzhik Date: Thu, 6 Jul 2023 17:51:14 +0400 Subject: [PATCH] + create_test_data command --- store/management/commands/create_test_data.py | 36 +++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 store/management/commands/create_test_data.py diff --git a/store/management/commands/create_test_data.py b/store/management/commands/create_test_data.py new file mode 100644 index 0000000..95719d5 --- /dev/null +++ b/store/management/commands/create_test_data.py @@ -0,0 +1,36 @@ +from django.contrib.auth.hashers import make_password +from django.core.management import BaseCommand +from tqdm import tqdm + +from store.models import User + +users = [ + { + "email": "poizonstore@mail.ru", + "password": "219404Poizon", + "job_title": User.ADMIN + }, + { + "email": "poizonmanager1@mail.ru", + "password": "poizonm1", + "job_title": User.PRODUCT_MANAGER + }, + { + "email": "poizonorder1@mail.ru", + "password": "2193071Po1", + "job_title": User.ORDER_MANAGER + } +] + + +class Command(BaseCommand): + help = ''' Create test data ''' + + def create_test_users(self): + for user_data in tqdm(users, desc="Creating users"): + user_data['password'] = make_password(user_data['password']) + User.objects.get_or_create(email=user_data['email'], defaults=user_data) + + def handle(self, *args, **kwargs): + self.create_test_users() +