Add a script for creating admin user from the terminal

Author: m-danya

README.md

@@ -24,6 +24,8 @@ echo -e "\nPLANTY_AUTH_SECRET=$(openssl rand -base64 32)" >> .env
 # change other secrets in .env
 
 docker compose up -d
+
+docker exec -it planty-backend-1 uv run python -m planty.scripts.create_admin
 ```
 
 Open `http://localhost` to access the app.

planty/scripts/create_admin.py

@@ -0,0 +1,47 @@
+import asyncio
+import contextlib
+
+import typer
+
+from planty.application.auth import get_user_db, get_user_manager
+from planty.application.schemas import UserCreate
+from planty.infrastructure.database import raw_async_session_maker
+
+app = typer.Typer()
+
+
+get_user_db_context = contextlib.asynccontextmanager(get_user_db)
+get_user_manager_context = contextlib.asynccontextmanager(get_user_manager)
+
+
+async def create_admin_user(email: str, password: str) -> None:
+    async with raw_async_session_maker() as session:
+        async with get_user_db_context(session) as user_db:
+            async with get_user_manager_context(user_db) as user_manager:
+                user = await user_manager.create(
+                    UserCreate(
+                        email=email,
+                        password=password,
+                        is_superuser=True,
+                        is_active=True,
+                        is_verified=True,
+                    )
+                )
+            print(f"User {user} successfully created")
+
+
+@app.command()
+def create_admin(
+    email: str = typer.Option(..., prompt=True),
+    password: str = typer.Option(
+        ...,
+        prompt=True,
+        hide_input=True,
+        confirmation_prompt=True,
+    ),
+):
+    asyncio.run(create_admin_user(email, password))
+
+
+if __name__ == "__main__":
+    app()

pyproject.toml

@@ -19,6 +19,7 @@ dependencies = [
     "pytest-mock>=3.14.0",
     "python-dateutil>=2.9.0.post0",
     "sqlalchemy>=2.0.36",
+    "typer>=0.15.1",
     "types-aiobotocore[s3]>=2.16.1",
     "types-docker>=7.1.0.20241229",
     "types-python-dateutil>=2.9.0.20241206",