diff --git a/NIPTool/commands/load_commands.py b/NIPTool/commands/load_commands.py index 0423ff2b..3508862e 100644 --- a/NIPTool/commands/load_commands.py +++ b/NIPTool/commands/load_commands.py @@ -52,7 +52,7 @@ def load_batch( @load_commands.command(name="user") @click.option("--email", required=True) @click.option("--user-name", required=True) -@click.option("--role", type=click.Choice(["RW"]), default="RW", show_default=True) +@click.option("--role", type=click.Choice(["RW", "R"]), default="RW", show_default=True) @click.pass_obj def load_user(context: dict, email: str, user_name: str, role: str): """Add a user to the database""" diff --git a/NIPTool/models/server/load.py b/NIPTool/models/server/load.py index 45f498c6..1e020d7c 100644 --- a/NIPTool/models/server/load.py +++ b/NIPTool/models/server/load.py @@ -1,6 +1,7 @@ from typing import Optional from pydantic import BaseModel, EmailStr +from typing_extensions import Literal class BatchRequestBody(BaseModel): @@ -12,4 +13,4 @@ class BatchRequestBody(BaseModel): class UserRequestBody(BaseModel): email: EmailStr username: str - role: str + role: Literal["R", "RW"]