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() +