update

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

@@ -4,7 +4,6 @@ from aws_lambda_powertools.event_handler.exceptions import (
 )
 from layercake.dynamodb import (
     ComposeKey,
-    DynamoDBCollection,
     DynamoDBPersistenceLayer,
     MissingError,
     PartitionKey,
@@ -13,29 +12,19 @@ from layercake.dynamodb import (
 from boto3clients import dynamodb_client
 from config import USER_TABLE
 
-from .orgs import router as orgs
-
-__all__ = ['orgs']
-
 
 class BadRequestError(MissingError, PowertoolsBadRequestError): ...
 
 
 router = Router()
 user_layer = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)
-user_collect = DynamoDBCollection(user_layer, exc_cls=BadRequestError)
 
 
-@router.get(
-    '/<id>/logs',
-    compress=True,
-    tags=['User'],
-    summary='Get user logs',
-)
+@router.get('/<id>/logs', compress=True, tags=['User'])
 def get_logs(id: str):
-    return user_collect.query(
-        # Post-migration: uncomment to enable PartitionKey with a composite key (id with `logs` prefix).
-        # PartitionKey(ComposeKey(id, 'logs')),
+    return user_layer.collection.query(
+        # Post-migration: uncomment the following line
+        # PartitionKey(ComposeKey(id, 'LOGS')),
         PartitionKey(ComposeKey(id, 'log', delimiter=':')),
         start_key=router.current_event.get_query_string_value('start_key', None),
     )