update layercake version
This commit is contained in:
@@ -5,7 +5,6 @@ from aws_lambda_powertools.event_handler.exceptions import (
|
||||
BadRequestError as PowertoolsBadRequestError,
|
||||
)
|
||||
from layercake.dynamodb import (
|
||||
DynamoDBCollection,
|
||||
DynamoDBPersistenceLayer,
|
||||
KeyPair,
|
||||
MissingError,
|
||||
@@ -26,7 +25,6 @@ class BadRequestError(MissingError, PowertoolsBadRequestError): ...
|
||||
|
||||
router = Router()
|
||||
user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
|
||||
user_collect = DynamoDBCollection(user_layer, exc_cls=BadRequestError)
|
||||
|
||||
|
||||
@router.get(
|
||||
@@ -36,7 +34,7 @@ user_collect = DynamoDBCollection(user_layer, exc_cls=BadRequestError)
|
||||
summary='Get user orgs',
|
||||
)
|
||||
def get_orgs(id: str):
|
||||
return user_collect.query(
|
||||
return user_layer.collection.query(
|
||||
KeyPair(id, PrefixKey('orgs')),
|
||||
start_key=router.current_event.get_query_string_value('start_key', None),
|
||||
)
|
||||
@@ -54,7 +52,9 @@ class Unassign(BaseModel):
|
||||
tags=['User'],
|
||||
summary='Delete user org',
|
||||
middlewares=[
|
||||
AuditLogMiddleware('UNASSIGN_ORG', user_collect, ('id', 'name', 'cnpj'))
|
||||
AuditLogMiddleware(
|
||||
'UNASSIGN_ORG', user_layer.collection, ('id', 'name', 'cnpj')
|
||||
)
|
||||
],
|
||||
)
|
||||
def delete_org(id: str, payload: Unassign):
|
||||
|
||||
Reference in New Issue
Block a user