added change_permissions command

Author: dantownsend

piccolo/apps/user/commands/change_permissions.py

@@ -0,0 +1,50 @@
+import typing as t
+
+from piccolo.apps.user.tables import BaseUser
+from piccolo.utils.warnings import colored_string, Level
+
+if t.TYPE_CHECKING:
+    from piccolo.columns import Column
+
+
+async def change_permissions(
+    username: str,
+    admin: t.Optional[bool] = None,
+    superuser: t.Optional[bool] = None,
+    active: t.Optional[bool] = None,
+):
+    """
+    Change a user's permissions.
+
+    :param username:
+        Change the permissions for this user.
+    :param admin:
+        Set `admin` for the user (true / false).
+    :param superuser:
+        Set `superuser` for the user (true / false).
+    :param active:
+        Set `active` for the user (true / false).
+
+    """
+    if not await BaseUser.exists().where(BaseUser.username == username).run():
+        print(
+            colored_string(
+                f"User {username} doesn't exist!", level=Level.medium
+            )
+        )
+        return
+
+    params: t.Dict[Column, bool] = {}
+
+    if admin is not None:
+        params[BaseUser.admin] = admin
+
+    if superuser is not None:
+        params[BaseUser.superuser] = superuser
+
+    if active is not None:
+        params[BaseUser.active] = active
+
+    await BaseUser.update(params).where(BaseUser.username == username).run()
+
+    print(f"Updated permissions for {username}")

piccolo/apps/user/commands/create.py

@@ -44,6 +44,17 @@ def get_is_superuser() -> bool:
     return superuser == "y"
 
 
+def get_is_active() -> bool:
+    while True:
+        active = input("Active? Enter y or n:\n")
+        if active in ("y", "n"):
+            break
+        else:
+            print("Unrecognised option")
+
+    return active == "y"
+
+
 def create():
     """
     Create a new user.
@@ -61,13 +72,14 @@ def create():
 
     is_admin = get_is_admin()
     is_superuser = get_is_superuser()
+    is_active = get_is_active()
 
     user = BaseUser(
         username=username,
         password=password,
         admin=is_admin,
         email=email,
-        active=True,
+        active=is_active,
         superuser=is_superuser,
     )
     user.save().run_sync()

piccolo/apps/user/piccolo_app.py

@@ -2,6 +2,7 @@
 
 from piccolo.conf.apps import AppConfig
 from .commands.change_password import change_password
+from .commands.change_permissions import change_permissions
 from .commands.create import create
 from .tables import BaseUser
 
@@ -16,5 +17,5 @@
     ),
     table_classes=[BaseUser],
     migration_dependencies=[],
-    commands=[create, change_password],
+    commands=[create, change_password, change_permissions],
 )

requirements.txt

@@ -3,5 +3,5 @@ asyncpg>=0.21.0
 black
 colorama>=0.4.0
 Jinja2>=2.11.0
-targ>=0.1.0
+targ>=0.2.0
 inflection>=0.5.1