update

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

@@ -16,8 +16,7 @@ from pydantic import BaseModel
 from api_gateway import JSONResponse
 from boto3clients import dynamodb_client
 from config import USER_TABLE
-from middlewares.audit_log_middleware import AuditLogMiddleware
-from rules.user import del_org_member
+from rules.user import remove_org_member
 
 
 class BadRequestError(MissingError, PowertoolsBadRequestError): ...
@@ -30,13 +29,13 @@ user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
 @router.get(
     '/<id>/orgs',
     compress=True,
-    tags=['User'],
-    summary='Get user orgs',
 )
 def get_orgs(id: str):
+    start_key = router.current_event.get_query_string_value('start_key', None)
+
     return user_layer.collection.query(
         KeyPair(id, PrefixKey('orgs')),
-        start_key=router.current_event.get_query_string_value('start_key', None),
+        start_key=start_key,
     )
 
 
@@ -46,17 +45,7 @@ class Unassign(BaseModel):
     cnpj: CnpjStr
 
 
-@router.delete(
-    '/<id>/orgs',
-    compress=True,
-    tags=['User'],
-    summary='Delete user org',
-    middlewares=[
-        AuditLogMiddleware(
-            'UNASSIGN_ORG', user_layer.collection, ('id', 'name', 'cnpj')
-        )
-    ],
-)
+@router.delete('/<id>/orgs', compress=True)
 def delete_org(id: str, payload: Unassign):
-    del_org_member(id, org_id=payload.id, persistence_layer=user_layer)
+    remove_org_member(id, org_id=payload.id, persistence_layer=user_layer)
     return JSONResponse(status_code=HTTPStatus.OK, body=payload)