update address

http-api/app/routes/orgs/__init__.py

@@ -1,3 +1,4 @@
+from .address import router as address
 from .policies import router as policies
 
-__all__ = ['policies']
+__all__ = ['policies', 'address']

http-api/app/routes/orgs/address.py

@@ -0,0 +1,51 @@
+from http import HTTPStatus
+
+from aws_lambda_powertools.event_handler.api_gateway import Router
+from layercake.dynamodb import (
+    DynamoDBPersistenceLayer,
+    KeyPair,
+)
+from pydantic import BaseModel
+
+from api_gateway import JSONResponse
+from boto3clients import dynamodb_client
+from config import USER_TABLE
+
+router = Router()
+org_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
+
+
+@router.get(
+    '/<id>/address',
+    compress=True,
+    tags=['Organization'],
+    summary='Get organization address',
+)
+def get_address(id: str):
+    return org_layer.collection.get_item(
+        KeyPair(id, 'metadata#address'),
+    )
+
+
+class Address(BaseModel):
+    postcode: str
+    address1: str
+    address2: str | None = None
+    neighborhood: str
+    city: str
+    state: str
+
+
+@router.post('/<id>/address', compress=True, tags=['Organization'])
+def post_address(id: str, payload: Address):
+    address = payload.model_dump()
+    org_layer.collection.put_item(
+        key=KeyPair(id, 'metadata#address'),
+        cond_expr='attribute_exists(sk)',
+        **address,
+    )
+
+    return JSONResponse(
+        body=payload,
+        status_code=HTTPStatus.OK,
+    )

http-api/app/routes/orgs/policies.py

@@ -1,26 +1,21 @@
 from http import HTTPStatus
 from typing import Literal
 
-from aws_lambda_powertools.event_handler import Response, content_types
 from aws_lambda_powertools.event_handler.api_gateway import Router
-from aws_lambda_powertools.event_handler.exceptions import (
-    BadRequestError,
-)
 from layercake.dynamodb import (
-    DynamoDBCollection,
     DynamoDBPersistenceLayer,
     SortKey,
     TransactKey,
 )
-from pydantic.main import BaseModel
+from pydantic import BaseModel
 
+from api_gateway import JSONResponse
 from boto3clients import dynamodb_client
 from config import USER_TABLE
 from rules.org import update_policies
 
 router = Router()
 org_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
-org_collect = DynamoDBCollection(org_layer, exc_cls=BadRequestError)
 
 
 @router.get(
@@ -30,7 +25,7 @@ org_collect = DynamoDBCollection(org_layer, exc_cls=BadRequestError)
     summary='Get organization policies',
 )
 def get_policies(id: str):
-    return org_collect.get_items(
+    return org_layer.collection.get_items(
         TransactKey(id)
         + SortKey('metadata#billing_policy', remove_prefix='metadata#')
         + SortKey('metadata#payment_policy', remove_prefix='metadata#'),
@@ -40,7 +35,7 @@ def get_policies(id: str):
 
 class BillingPolicy(BaseModel):
     billing_day: int
-    payment_method: Literal['PIX', 'BANK_SLIP', 'MANUAL']
+    payment_method: Literal['BANK_SLIP', 'MANUAL']
 
 
 class PaymentPolicy(BaseModel):
@@ -64,8 +59,7 @@ def put_policies(id: str, payload: Policies):
         persistence_layer=org_layer,
     )
 
-    return Response(
+    return JSONResponse(
         body=payload,
-        content_type=content_types.APPLICATION_JSON,
         status_code=HTTPStatus.OK,
     )