update admin name #32

api.saladeaula.digital/app/routes/orgs/__init__.py

@@ -1,6 +1,66 @@
+from typing import Annotated
+from uuid import uuid4
+
+from aws_lambda_powertools.event_handler.api_gateway import Router
+from aws_lambda_powertools.event_handler.openapi.params import Body
+from layercake.dateutils import now
+from layercake.dynamodb import DynamoDBPersistenceLayer
+from layercake.extra_types import CnpjStr, NameStr
+from pydantic import UUID4, BaseModel, EmailStr
+
+from boto3clients import dynamodb_client
+from config import INTERNAL_EMAIL_DOMAIN, USER_TABLE
+from exceptions import ConflictError
+
 from .admins import router as admins
 from .custom_pricing import router as custom_pricing
 from .enrollments.scheduled import router as scheduled
 from .users import router as users
 
+router = Router()
+dyn = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
+
+
 __all__ = ['admins', 'custom_pricing', 'scheduled', 'users']
+
+
+class OrgConflictError(ConflictError): ...
+
+
+class User(BaseModel):
+    id: str | UUID4
+    name: NameStr
+    email: EmailStr
+
+
+@router.post('/s')
+def add_org(
+    name: Annotated[str, Body(embed=True)],
+    cnpj: Annotated[CnpjStr, Body(embed=True)],
+    user: Annotated[User, Body(embed=True)],
+):
+    now_ = now()
+    org_id = str(uuid4())
+
+    with dyn.transact_writer() as transact:
+        transact.put(
+            item={
+                # Post-migration (users): rename `cnpj` to `CNPJ`
+                'id': 'cnpj',
+                'sk': cnpj,
+                'org_id': org_id,
+                'created_at': now_,
+            },
+            cond_expr='attribute_not_exists(sk)',
+            exc_cls=OrgConflictError,
+        )
+        transact.put(
+            item={
+                'id': org_id,
+                'sk': '0',
+                'name': name,
+                'email': f'org+{cnpj}@{INTERNAL_EMAIL_DOMAIN}',
+                'cnpj': cnpj,
+                'created_at': now_,
+            }
+        )

api.saladeaula.digital/app/routes/orgs/users/__init__.py

@@ -3,10 +3,7 @@ from typing import Annotated
 from uuid import uuid4
 
 from aws_lambda_powertools.event_handler.api_gateway import Router
-from aws_lambda_powertools.event_handler.exceptions import (
-    NotFoundError,
-    ServiceError,
-)
+from aws_lambda_powertools.event_handler.exceptions import NotFoundError
 from aws_lambda_powertools.event_handler.openapi.params import Body
 from layercake.dateutils import now, ttl
 from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair, SortKey
@@ -16,6 +13,7 @@ from pydantic import BaseModel, EmailStr, Field
 from api_gateway import JSONResponse
 from boto3clients import dynamodb_client
 from config import INTERNAL_EMAIL_DOMAIN, USER_TABLE
+from exceptions import ConflictError
 
 router = Router()
 dyn = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
@@ -33,11 +31,6 @@ class User(BaseModel):
     email: EmailStr
 
 
-class ConflictError(ServiceError):
-    def __init__(self, msg: str | dict):
-        super().__init__(HTTPStatus.CONFLICT, msg)
-
-
 class CPFConflictError(ConflictError): ...
 
 
@@ -218,7 +211,7 @@ def _add_member(user_id: str, org: Org) -> None:
         transact.update(
             key=KeyPair(user_id, '0'),
             # Post-migration (users): uncomment the following line
-            # update_expr='ADD tenant_id :org_id',
+            # update_expr='ADD org_id :org_id',
             update_expr='ADD tenant_id :org_id',
             expr_attr_values={
                 ':org_id': {org.id},

api.saladeaula.digital/app/routes/users/emails.py

@@ -4,7 +4,6 @@ from uuid import uuid4
 from aws_lambda_powertools.event_handler.api_gateway import Router
 from aws_lambda_powertools.event_handler.exceptions import (
     NotFoundError,
-    ServiceError,
 )
 from aws_lambda_powertools.event_handler.openapi.params import Body, Path, Query
 from layercake.dateutils import now, ttl
@@ -15,16 +14,12 @@ from typing_extensions import Annotated
 from api_gateway import JSONResponse
 from boto3clients import dynamodb_client
 from config import USER_TABLE
+from exceptions import ConflictError
 
 router = Router()
 dyn = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
 
 
-class ConflictError(ServiceError):
-    def __init__(self, msg: str | dict):
-        super().__init__(HTTPStatus.CONFLICT, msg)
-
-
 class UserNotFoundError(NotFoundError): ...