add account CRUD

trauma/api/account/db_requests.py

@@ -1,6 +1,8 @@
 import asyncio
 
 from trauma.api.account.model import AccountModel
+from trauma.api.account.schemas import CreateAccountRequest
+from trauma.api.common.db_requests import check_unique_fields_existence
 from trauma.core.config import settings
 
 
@@ -15,3 +17,14 @@ async def get_all_model_obj(page_size: int, page_index: int) -> tuple[list[Accou
         settings.DB_CLIENT.accounts.count_documents({})
     )
     return objects, total_count
+
+
+async def create_account_obj(data: CreateAccountRequest) -> AccountModel:
+    await check_unique_fields_existence("email", data.email)
+    account = AccountModel(**data.model_dump())
+    await settings.DB_CLIENT.accounts.insert_one(account.to_mongo())
+    return account
+
+
+async def delete_account(account_id: str) -> None:
+    await settings.DB_CLIENT.accounts.delete_one({"id": account_id})

trauma/api/account/schemas.py

@@ -12,3 +12,8 @@ class AccountWrapper(TraumaResponseWrapper[AccountModel]):
 class AllAccountsResponse(BaseModel):
     paging: Paging
     data: list[AccountModel]
+
+
+class CreateAccountRequest(BaseModel):
+    email: str
+    password: str

trauma/api/account/views.py

@@ -3,10 +3,10 @@ from typing import Optional
 from fastapi import Depends, Query
 
 from trauma.api.account import account_router
-from trauma.api.account.db_requests import get_all_model_obj
+from trauma.api.account.db_requests import get_all_model_obj, create_account_obj, delete_account
 from trauma.api.account.dto import AccountType
 from trauma.api.account.model import AccountModel
-from trauma.api.account.schemas import AccountWrapper, AllAccountsResponse
+from trauma.api.account.schemas import AccountWrapper, AllAccountsResponse, CreateAccountRequest
 from trauma.api.common.dto import Paging
 from trauma.core.security import PermissionDependency
 from trauma.core.wrappers import TraumaResponseWrapper
@@ -31,3 +31,20 @@ async def get_account(
         account: AccountModel = Depends(PermissionDependency([AccountType.Admin, AccountType.User]))
 ) -> AccountWrapper:
     return AccountWrapper(data=account)
+
+
+@account_router.post('')
+async def create_account(
+        account_data: CreateAccountRequest,
+        _: AccountModel = Depends(PermissionDependency([AccountType.Admin]))
+) -> AccountWrapper:
+    account = await create_account_obj(account_data)
+    return AccountWrapper(data=account)
+
+@account_router.delete('/{accountId}')
+async def update_account(
+        accountId: str,
+        _: AccountModel = Depends(PermissionDependency([AccountType.Admin]))
+) -> TraumaResponseWrapper:
+    await delete_account(accountId)
+    return TraumaResponseWrapper()

trauma/api/security/db_requests.py

@@ -1,23 +1,11 @@
-import asyncio
-
 from fastapi import HTTPException
 
 from trauma.api.account.model import AccountModel
-from trauma.api.common.db_requests import check_unique_fields_existence
-from trauma.api.security.schemas import RegisterAccountRequest, LoginAccountRequest
+from trauma.api.security.schemas import LoginAccountRequest
 from trauma.core.config import settings
 from trauma.core.security import verify_password
 
 
-async def save_account(data: RegisterAccountRequest) -> AccountModel:
-    await check_unique_fields_existence("email", data.email)
-    account = AccountModel(
-        **data.model_dump()
-    )
-    await settings.DB_CLIENT.accounts.insert_one(account.to_mongo())
-    return account
-
-
 async def authenticate_account(data: LoginAccountRequest) -> AccountModel:
     account = await settings.DB_CLIENT.accounts.find_one(
         {"email": data.email},

trauma/api/security/schemas.py

@@ -5,15 +5,6 @@ from trauma.api.account.model import AccountModel
 from trauma.core.wrappers import TraumaResponseWrapper
 
 
-class RegisterAccountRequest(BaseModel):
-    email: EmailStr
-    password: str
-
-
-class RegisterAccountWrapper(TraumaResponseWrapper[AccountModel]):
-    pass
-
-
 class LoginAccountRequest(BaseModel):
     email: EmailStr
     password: str

trauma/api/security/views.py

@@ -1,20 +1,12 @@
 from trauma.api.account.dto import AccessToken
 from trauma.api.security import security_router
-from trauma.api.security.db_requests import authenticate_account, save_account
-from trauma.api.security.schemas import (RegisterAccountRequest,
-                                         RegisterAccountWrapper,
-                                         LoginAccountResponse,
+from trauma.api.security.db_requests import authenticate_account
+from trauma.api.security.schemas import (LoginAccountResponse,
                                          LoginAccountWrapper,
                                          LoginAccountRequest)
 from trauma.core.security import create_access_token
 
 
-@security_router.post('/register')
-async def register_user(data: RegisterAccountRequest) -> RegisterAccountWrapper:
-    account = await save_account(data)
-    return RegisterAccountWrapper(data=account)
-
-
 @security_router.post('/login')
 async def login(data: LoginAccountRequest) -> LoginAccountWrapper:
     account = await authenticate_account(data)