added superuser and last_login fields to BaseUser

Author: dantownsend

piccolo/apps/user/piccolo_migrations/2021-04-30T16-14-15.py

@@ -0,0 +1,48 @@
+from piccolo.apps.migrations.auto import MigrationManager
+from piccolo.columns.column_types import Boolean
+from piccolo.columns.column_types import Timestamp
+from piccolo.columns.indexes import IndexMethod
+
+
+ID = "2021-04-30T16:14:15"
+VERSION = "0.18.2"
+
+
+async def forwards():
+    manager = MigrationManager(migration_id=ID, app_name="user")
+
+    manager.add_column(
+        table_class_name="BaseUser",
+        tablename="piccolo_user",
+        column_name="superuser",
+        column_class_name="Boolean",
+        column_class=Boolean,
+        params={
+            "default": False,
+            "null": False,
+            "primary": False,
+            "key": False,
+            "unique": False,
+            "index": False,
+            "index_method": IndexMethod.btree,
+        },
+    )
+
+    manager.add_column(
+        table_class_name="BaseUser",
+        tablename="piccolo_user",
+        column_name="last_login",
+        column_class_name="Timestamp",
+        column_class=Timestamp,
+        params={
+            "default": None,
+            "null": True,
+            "primary": False,
+            "key": False,
+            "unique": False,
+            "index": False,
+            "index_method": IndexMethod.btree,
+        },
+    )
+
+    return manager

piccolo/apps/user/tables.py

@@ -6,7 +6,7 @@
 import secrets
 import typing as t
 
-from piccolo.columns import Varchar, Boolean, Secret
+from piccolo.columns import Varchar, Boolean, Secret, Timestamp
 from piccolo.columns.readable import Readable
 from piccolo.table import Table
 from piccolo.utils.sync import run_sync
@@ -23,7 +23,19 @@ class BaseUser(Table, tablename="piccolo_user"):
     last_name = Varchar(null=True)
     email = Varchar(length=255, unique=True)
     active = Boolean(default=False)
-    admin = Boolean(default=False)
+    admin = Boolean(
+        default=False, help_text="An admin can log into the Piccolo admin GUI."
+    )
+    superuser = Boolean(
+        default=False,
+        help_text=(
+            "If True, this user can manage other users's passwords in the "
+            "Piccolo admin GUI."
+        ),
+    )
+    last_login = Timestamp(
+        null=True, default=None, help_text="When this user last logged in."
+    )
 
     def __init__(self, **kwargs):
         """

piccolo_conf.py

@@ -0,0 +1,20 @@
+"""
+This piccolo_conf file is just here so migrations can be made for Piccolo's own
+internal apps.
+
+For example:
+
+python -m piccolo.main migration new user --auto
+
+"""
+
+from piccolo.conf.apps import AppRegistry
+from piccolo.engine.postgres import PostgresEngine
+
+
+DB = PostgresEngine(config={})
+
+
+# A list of paths to piccolo apps
+# e.g. ['blog.piccolo_app']
+APP_REGISTRY = AppRegistry(apps=["piccolo.apps.user.piccolo_app"])