38 lines
995 B
Python
38 lines
995 B
Python
from django.core.management.base import BaseCommand
|
|
from django.contrib.auth import get_user_model
|
|
from lot.models import LotTag
|
|
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Create a new user"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('email', type=str, help='email')
|
|
parser.add_argument('password', type=str, help='password')
|
|
|
|
def handle(self, *args, **kwargs):
|
|
email = kwargs['email']
|
|
password = kwargs['password']
|
|
self.create_user(email, password)
|
|
self.create_lot_tags()
|
|
|
|
def create_user(self, email, password):
|
|
self.u = User.objects.create(email=email, password=password)
|
|
self.u.set_password(password)
|
|
self.u.save()
|
|
|
|
def create_lot_tags(self):
|
|
tags = [
|
|
"Entrada",
|
|
"Salida",
|
|
"Temporal"
|
|
]
|
|
for tag in tags:
|
|
LotTag.objects.create(
|
|
name=tag,
|
|
owner=self.u
|
|
)
|