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 )