modified create user command - can now designate as superuser

Author: dantownsend

piccolo/apps/user/commands/create.py

@@ -33,6 +33,17 @@ def get_is_admin() -> bool:
     return admin == "y"
 
 
+def get_is_superuser() -> bool:
+    while True:
+        superuser = input("Superuser? Enter y or n:\n")
+        if superuser in ("y", "n"):
+            break
+        else:
+            print("Unrecognised option")
+
+    return superuser == "y"
+
+
 def create():
     """
     Create a new user.
@@ -49,13 +60,15 @@ def create():
         sys.exit("The password is too short")
 
     is_admin = get_is_admin()
+    is_superuser = get_is_superuser()
 
     user = BaseUser(
         username=username,
         password=password,
         admin=is_admin,
         email=email,
         active=True,
+        superuser=is_superuser,
     )
     user.save().run_sync()
 

tests/apps/user/commands/test_create.py

@@ -31,6 +31,10 @@ def tearDown(self):
     @patch(
         "piccolo.apps.user.commands.create.get_is_admin", return_value=True,
     )
+    @patch(
+        "piccolo.apps.user.commands.create.get_is_superuser",
+        return_value=True,
+    )
     def test_create(self, *args, **kwargs):
         create()
 
@@ -40,6 +44,7 @@ def test_create(self, *args, **kwargs):
                 (BaseUser.admin == True)  # noqa: E712
                 & (BaseUser.username == "bob123")
                 & (BaseUser.email == "[email protected]")
+                & (BaseUser.superuser == True)
             )
             .run_sync()
         )