Add user registration endpoint and role support

Introduced a new `/register` endpoint to handle user registration, allowing email, password, and role input. Added a `role` field to `AccountModel` and implemented database logic to ensure unique email constraints during registration.

trauma/api/account/model.py

@@ -10,6 +10,7 @@ from trauma.core.database import MongoBaseModel
 class AccountModel(MongoBaseModel):
     email: EmailStr
     password: str | None = Field(exclude=True, default=None)
+    role: str | None = None
     accountType: AccountType = AccountType.User
 
     datetimeInserted: datetime = Field(default_factory=datetime.now)

trauma/api/account/views.py

@@ -41,6 +41,7 @@ async def create_account(
     account = await create_account_obj(account_data)
     return AccountWrapper(data=account)
 
+
 @account_router.delete('/{accountId}')
 async def update_account(
         accountId: str,

trauma/api/security/db_requests.py

@@ -1,7 +1,8 @@
 from fastapi import HTTPException
 
 from trauma.api.account.model import AccountModel
-from trauma.api.security.schemas import LoginAccountRequest
+from trauma.api.common.db_requests import check_unique_fields_existence
+from trauma.api.security.schemas import LoginAccountRequest, RegisterAccountRequest
 from trauma.core.config import settings
 from trauma.core.security import verify_password
 
@@ -18,3 +19,14 @@ async def authenticate_account(data: LoginAccountRequest) -> AccountModel:
     if not verify_password(data.password, account.password):
         raise HTTPException(status_code=401, detail="Invalid email or password.")
     return account
+
+
+async def register_account(data: RegisterAccountRequest) -> AccountModel:
+    await check_unique_fields_existence("email", data.email)
+    account = AccountModel(
+        email=data.email,
+        password=data.password,
+        role=data.role,
+    )
+    await settings.DB_CLIENT.accounts.insert_one(account.to_mongo())
+    return account

trauma/api/security/schemas.py

@@ -17,3 +17,8 @@ class LoginAccountResponse(BaseModel):
 
 class LoginAccountWrapper(TraumaResponseWrapper[LoginAccountResponse]):
     pass
+
+class RegisterAccountRequest(BaseModel):
+    email: EmailStr
+    password: str
+    role: str

trauma/api/security/views.py

@@ -1,10 +1,12 @@
 from trauma.api.account.dto import AccessToken
+from trauma.api.account.model import AccountModel
 from trauma.api.security import security_router
-from trauma.api.security.db_requests import authenticate_account
+from trauma.api.security.db_requests import authenticate_account, register_account
 from trauma.api.security.schemas import (LoginAccountResponse,
                                          LoginAccountWrapper,
-                                         LoginAccountRequest)
+                                         LoginAccountRequest, RegisterAccountRequest)
 from trauma.core.security import create_access_token
+from trauma.core.wrappers import TraumaResponseWrapper
 
 
 @security_router.post('/login')
@@ -16,3 +18,9 @@ async def login(data: LoginAccountRequest) -> LoginAccountWrapper:
         account=account,
     )
     return LoginAccountWrapper(data=response)
+
+
+@security_router.post('/register')
+async def register(data: RegisterAccountRequest) -> TraumaResponseWrapper[AccountModel]:
+    account = await register_account(data)
+    return TraumaResponseWrapper(data=account)