From 35bf1a65a2be66651e696a4d70cf675cffbe2c57 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Mon, 7 Oct 2024 16:57:26 +0200 Subject: [PATCH] add admin from add_user command --- reset.sh | 2 +- user/management/commands/add_user.py | 9 ++++++--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/reset.sh b/reset.sh index 7e0d236..8ee970b 100644 --- a/reset.sh +++ b/reset.sh @@ -1,5 +1,5 @@ rm db/* python3 manage.py migrate python3 manage.py add_institution Pangea -python3 manage.py add_user Pangea user@example.org 1234 +python3 manage.py add_user Pangea user@example.org 1234 True python3 manage.py up_snapshots example/snapshots/ user@example.org diff --git a/user/management/commands/add_user.py b/user/management/commands/add_user.py index 7d31ab7..f4cc780 100644 --- a/user/management/commands/add_user.py +++ b/user/management/commands/add_user.py @@ -14,19 +14,22 @@ class Command(BaseCommand): parser.add_argument('institution', type=str, help='institution') parser.add_argument('email', type=str, help='email') parser.add_argument('password', type=str, help='password') + parser.add_argument('is_admin', nargs='?', default=False, type=str, help='is admin') def handle(self, *args, **kwargs): email = kwargs['email'] password = kwargs['password'] + is_admin = kwargs['is_admin'] institution = Institution.objects.get(name=kwargs['institution']) - self.create_user(institution, email, password) + self.create_user(institution, email, password, is_admin) self.create_lot_tags() - def create_user(self, institution, email, password): + def create_user(self, institution, email, password, is_admin): self.u = User.objects.create( institution=institution, email=email, - password=password + password=password, + is_admin=is_admin, ) self.u.set_password(password) self.u.save()