add address

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

@@ -0,0 +1,55 @@
+import re
+from http import HTTPStatus
+from typing import Annotated
+
+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, KeyPair
+from pydantic import BaseModel, ConfigDict, field_validator
+
+from api_gateway import JSONResponse
+from boto3clients import dynamodb_client
+from config import USER_TABLE
+
+router = Router()
+dyn = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
+
+
+@router.get('/<org_id>/address')
+def address(org_id: str):
+    return dyn.collection.get_item(
+        KeyPair(org_id, 'ADDRESS'),
+        raise_on_error=False,
+        default={},
+    )
+
+
+class Address(BaseModel):
+    model_config = ConfigDict(str_strip_whitespace=True)
+
+    postcode: str
+    address1: str
+    address2: str | None = None
+    neighborhood: str
+    city: str
+    state: str
+
+    @field_validator('postcode')
+    @classmethod
+    def _only_numbers(cls, value: str) -> str:
+        return re.sub(r'\D', '', value)
+
+
+@router.post('/<org_id>/address')
+def update(org_id: str, address: Annotated[Address, Body(embed=True)]):
+    dyn.collection.put_item(
+        key=KeyPair(org_id, 'ADDRESS'),
+        updated_at=now(),
+        **address.model_dump(exclude_none=True),
+    )
+
+    return JSONResponse(
+        body=address,
+        status_code=HTTPStatus.OK,
+    )