45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
from django.core.management.base import BaseCommand
|
|
from django.contrib.auth import get_user_model
|
|
from lot.models import LotTag
|
|
from user.models import Institution
|
|
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Create a new user"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('institution', type=str, help='institution')
|
|
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']
|
|
institution = Institution.objects.get(name=kwargs['institution'])
|
|
self.create_user(institution, email, password)
|
|
self.create_lot_tags()
|
|
|
|
def create_user(self, institution, email, password):
|
|
self.u = User.objects.create(
|
|
institution=institution,
|
|
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
|
|
)
|