create-admin

Author: leoguillaume

Makefile

@@ -8,9 +8,9 @@ help:
 quickstart:
 	@python cli.py --quickstart --env-file $(env)
 
-# create-user ----------------------------------------------------------------------------------------------------------------------------------------
-create-user:
-	@python scripts/create_first_user.py
+# create-admon ----------------------------------------------------------------------------------------------------------------------------------------
+create-admin:
+	@python scripts/create_admin.py
 
 # dev ------------------------------------------------------------------------------------------------------------------------------------------------
 dev:
@@ -36,4 +36,4 @@ test-integ:
 %:
 	@:
 
-.PHONY: help quickstart dev lint test-unit test-integ
+.PHONY: help quickstart dev lint test-unit test-integ create-admin

cli.py

@@ -61,7 +61,7 @@ def display_make_help(console: Console) -> None:
     commands.add_column("Description")
     commands.add_row("quickstart [env=.env]", "Start services in docker environment")
     commands.add_row("dev [env=.env]", "Start services in local development mode")
-    commands.add_row("create-user", "Create a first user")
+    commands.add_row("create-admin", "Create a first admin user")
     commands.add_row("lint", "Run linter")
     commands.add_row("test-unit", "Run unit tests")
     commands.add_row("test-integ", "Run integration tests")

scripts/create_first_user.py scripts/create_admin.pyscripts/create_first_user.py renamed to scripts/create_admin.py

@@ -1,15 +1,20 @@
 import argparse
 
 import requests
+from rich import box
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--api_url", type=str, default="http://localhost:8000")
 parser.add_argument("--master_key", type=str, default="changeme")
-parser.add_argument("--first_email", type=str, default="my-first-user")
-parser.add_argument("--first_password", type=str, default="changeme")
+parser.add_argument("--email", type=str, default="admin")
+parser.add_argument("--password", type=str, default="changeme")
 
 
 if __name__ == "__main__":
+    console = Console()
     args = parser.parse_args()
 
     headers = {"Authorization": f"Bearer {args.master_key}"}
@@ -19,7 +24,7 @@
     assert response.status_code == 200, response.text
     routers = response.json()["data"]
 
-    role_name = "my-first-role"
+    role_name = "admin"
     role_permissions = ["admin"]
     role_limits = []
 
@@ -40,7 +45,7 @@
         assert response.status_code == 200, response.text
         roles = response.json()["data"]
         for role in roles:
-            if role["name"] == "my-first-role":
+            if role["name"] == role_name:
                 role_id = role["id"]
                 break
     else:
@@ -51,14 +56,14 @@
     response = requests.post(
         url=f"{args.api_url}/v1/admin/users",
         headers=headers,
-        json={"email": args.first_email, "name": args.first_email, "password": args.first_password, "role": role_id},
+        json={"email": args.email, "name": args.email, "password": args.password, "role": role_id},
     )
     if response.status_code == 409:
         response = requests.get(f"{args.api_url}/v1/admin/users", headers=headers)
         assert response.status_code == 200, response.text
         users = response.json()["data"]
         for user in users:
-            if user["email"] == args.first_email:
+            if user["email"] == args.email:
                 user_id = user["id"]
                 break
         message = "User already exists, new api key created."
@@ -73,18 +78,23 @@
 
     key = response.json()["token"]
 
-    display_limits = "\n                   ".join([f"{router['name']} → unlimited" for router in routers])
-
-    print(f"""
-\033[32;1m✔ {message} \033[0m
-
-\033[32;1mRole:\033[0m              {role_name}
-\033[32;1mRole permissions:\033[0m  {",".join(role_permissions)}
-\033[32;1mRole limits: \033[0m      
-                   {display_limits}
-
-\033[32;1mEmail:\033[0m             {args.first_email}
-\033[32;1mPassword:\033[0m          {args.first_password}
-
-\033[32;1mAPI key:\033[0m           {key}
-""")
+    limits_table = Table(box=box.SIMPLE_HEAD, header_style="bold cyan")
+    limits_table.add_column("Router")
+    limits_table.add_column("Limit")
+    for router in routers:
+        limits_table.add_row(router["name"], "unlimited")
+
+    summary_table = Table(box=box.SIMPLE_HEAD, show_header=False)
+    summary_table.add_column("Field", style="bold green")
+    summary_table.add_column("Value", overflow="fold")
+    summary_table.add_row("Role", role_name)
+    summary_table.add_row("Role permissions", ",".join(role_permissions))
+    summary_table.add_row("Email", args.email)
+    summary_table.add_row("Password", args.password)
+    summary_table.add_row("API key", key)
+
+    console.print(f"✔ {message}", style="bold green")
+    console.print()
+    console.print(Panel(summary_table, title="User information", border_style="bright_blue", padding=(1, 2)))
+    console.print()
+    console.print(Panel(limits_table, title="Role limits", border_style="bright_blue", padding=(1, 2)))