+ create_test_data command

This commit is contained in:
Phil Zhitnikov 2023-07-06 17:51:14 +04:00
parent e7284e7725
commit 332ffa3a5f

View File

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